[#787] netmap: Refactor NewEpoch method

Split for user and control methods.

Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
This commit is contained in:
Dmitrii Stepanov 2023-11-13 17:22:31 +03:00
parent c8a62ffedd
commit 8088063195
3 changed files with 20 additions and 9 deletions

View file

@ -45,8 +45,7 @@ func (w *netmapClientWrapper) NetMap() (*netmap.NetMap, error) {
} }
func (w *netmapClientWrapper) NewEpoch(epoch uint64) error { func (w *netmapClientWrapper) NewEpoch(epoch uint64) error {
_, err := w.netmapClient.NewEpoch(epoch, 0, false) return w.netmapClient.NewEpoch(epoch)
return err
} }
func (w *netmapClientWrapper) MorphIsValidScript(script []byte, signers []transaction.Signer) (valid bool, err error) { func (w *netmapClientWrapper) MorphIsValidScript(script []byte, signers []transaction.Signer) (valid bool, err error) {

View file

@ -8,15 +8,27 @@ import (
// NewEpoch updates FrostFS epoch number through // NewEpoch updates FrostFS epoch number through
// Netmap contract call. // Netmap contract call.
// If `force` is true, this call is normally initiated by a control func (c *Client) NewEpoch(epoch uint64) error {
// service command and uses a control notary transaction internally
// to ensure all nodes produce the same transaction with high probability.
// If vub > 0, vub will be used as valid until block value.
func (c *Client) NewEpoch(epoch uint64, vub uint32, force bool) (uint32, error) {
prm := client.InvokePrm{} prm := client.InvokePrm{}
prm.SetMethod(newEpochMethod) prm.SetMethod(newEpochMethod)
prm.SetArgs(epoch) prm.SetArgs(epoch)
prm.SetControlTX(force)
_, err := c.client.Invoke(prm)
if err != nil {
return fmt.Errorf("could not invoke method (%s): %w", newEpochMethod, err)
}
return nil
}
// NewEpochControl updates FrostFS epoch number through
// control notary transaction internally to ensure all
// nodes produce the same transaction with high probability.
// If vub > 0, vub will be used as valid until block value.
func (c *Client) NewEpochControl(epoch uint64, vub uint32) (uint32, error) {
prm := client.InvokePrm{}
prm.SetMethod(newEpochMethod)
prm.SetArgs(epoch)
prm.SetControlTX(true)
prm.SetVUB(vub) prm.SetVUB(vub)
res, err := c.client.Invoke(prm) res, err := c.client.Invoke(prm)

View file

@ -53,7 +53,7 @@ func (s *Server) TickEpoch(_ context.Context, req *control.TickEpochRequest) (*c
return nil, fmt.Errorf("getting current epoch: %w", err) return nil, fmt.Errorf("getting current epoch: %w", err)
} }
vub, err := s.netmapClient.NewEpoch(epoch+1, req.GetBody().GetVub(), true) vub, err := s.netmapClient.NewEpochControl(epoch+1, req.GetBody().GetVub())
if err != nil { if err != nil {
return nil, fmt.Errorf("forcing new epoch: %w", err) return nil, fmt.Errorf("forcing new epoch: %w", err)
} }