67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
|
package container
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
||
|
|
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
||
|
|
"github.com/nspcc-dev/neofs-node/pkg/morph/event"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
// StartEstimation structure of container.StartEstimation notification from
|
||
|
|
// morph chain.
|
||
|
|
type StartEstimation struct {
|
||
|
|
epoch uint64
|
||
|
|
}
|
||
|
|
|
||
|
|
// StopEstimation structure of container.StopEstimation notification from
|
||
|
|
// morph chain.
|
||
|
|
type StopEstimation struct {
|
||
|
|
epoch uint64
|
||
|
|
}
|
||
|
|
|
||
|
|
// MorphEvent implements Neo:Morph Event interface.
|
||
|
|
func (StartEstimation) MorphEvent() {}
|
||
|
|
|
||
|
|
// MorphEvent implements Neo:Morph Event interface.
|
||
|
|
func (StopEstimation) MorphEvent() {}
|
||
|
|
|
||
|
|
// Epoch returns epoch value for which to start container size estimation.
|
||
|
|
func (s StartEstimation) Epoch() uint64 { return s.epoch }
|
||
|
|
|
||
|
|
// Epoch returns epoch value for which to stop container size estimation.
|
||
|
|
func (s StopEstimation) Epoch() uint64 { return s.epoch }
|
||
|
|
|
||
|
|
// ParseStartEstimation from notification into container event structure.
|
||
|
|
func ParseStartEstimation(params []stackitem.Item) (event.Event, error) {
|
||
|
|
epoch, err := parseEstimation(params)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return StartEstimation{epoch: epoch}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ParseStopEstimation from notification into container event structure.
|
||
|
|
func ParseStopEstimation(params []stackitem.Item) (event.Event, error) {
|
||
|
|
epoch, err := parseEstimation(params)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return StopEstimation{epoch: epoch}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseEstimation(params []stackitem.Item) (uint64, error) {
|
||
|
|
if ln := len(params); ln != 1 {
|
||
|
|
return 0, event.WrongNumberOfParameters(1, ln)
|
||
|
|
}
|
||
|
|
|
||
|
|
// parse container
|
||
|
|
epoch, err := client.IntFromStackItem(params[0])
|
||
|
|
if err != nil {
|
||
|
|
return 0, errors.Wrap(err, "could not get estimation epoch")
|
||
|
|
}
|
||
|
|
|
||
|
|
return uint64(epoch), nil
|
||
|
|
}
|