forked from TrueCloudLab/frostfs-node
55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package balance
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
)
|
|
|
|
// LockPrm groups parameters of Lock operation.
|
|
type LockPrm struct {
|
|
id []byte
|
|
user util.Uint160
|
|
lock util.Uint160
|
|
amount int64
|
|
dueEpoch int64
|
|
|
|
client.InvokePrmOptional
|
|
}
|
|
|
|
// SetID sets ID.
|
|
func (l *LockPrm) SetID(id []byte) {
|
|
l.id = id
|
|
}
|
|
|
|
// SetUser set user.
|
|
func (l *LockPrm) SetUser(user util.Uint160) {
|
|
l.user = user
|
|
}
|
|
|
|
// SetLock sets lock.
|
|
func (l *LockPrm) SetLock(lock util.Uint160) {
|
|
l.lock = lock
|
|
}
|
|
|
|
// SetAmount sets amount.
|
|
func (l *LockPrm) SetAmount(amount int64) {
|
|
l.amount = amount
|
|
}
|
|
|
|
// SetDueEpoch sets end of the lock.
|
|
func (l *LockPrm) SetDueEpoch(dueEpoch int64) {
|
|
l.dueEpoch = dueEpoch
|
|
}
|
|
|
|
// Lock locks fund on the user account.
|
|
func (c *Client) Lock(ctx context.Context, p LockPrm) error {
|
|
prm := client.InvokePrm{}
|
|
prm.SetMethod(lockMethod)
|
|
prm.SetArgs(p.id, p.user, p.lock, p.amount, p.dueEpoch)
|
|
prm.InvokePrmOptional = p.InvokePrmOptional
|
|
|
|
_, err := c.client.Invoke(ctx, prm)
|
|
return err
|
|
}
|