forked from TrueCloudLab/frostfs-node
Dmitrii Stepanov
d433b49265
`fmt.Errorf can be replaced with errors.New` and `fmt.Sprintf can be replaced with string addition` Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package policy
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"text/tabwriter"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/helper"
|
|
commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/invoker"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/policy"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const (
|
|
execFeeParam = "ExecFeeFactor"
|
|
storagePriceParam = "StoragePrice"
|
|
setFeeParam = "FeePerByte"
|
|
)
|
|
|
|
var errInvalidParameterFormat = errors.New("invalid parameter format, must be Parameter=Value")
|
|
|
|
func SetPolicyCmd(cmd *cobra.Command, args []string) error {
|
|
wCtx, err := helper.NewInitializeContext(cmd, viper.GetViper())
|
|
if err != nil {
|
|
return fmt.Errorf("can't to initialize context: %w", err)
|
|
}
|
|
|
|
bw := io.NewBufBinWriter()
|
|
for i := range args {
|
|
k, v, found := strings.Cut(args[i], "=")
|
|
if !found {
|
|
return errInvalidParameterFormat
|
|
}
|
|
|
|
switch k {
|
|
case execFeeParam, storagePriceParam, setFeeParam:
|
|
default:
|
|
return fmt.Errorf("parameter must be one of %s, %s and %s", execFeeParam, storagePriceParam, setFeeParam)
|
|
}
|
|
|
|
value, err := strconv.ParseUint(v, 10, 32)
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse parameter value '%s': %w", args[1], err)
|
|
}
|
|
|
|
emit.AppCall(bw.BinWriter, policy.Hash, "set"+k, callflag.All, int64(value))
|
|
}
|
|
|
|
if err := wCtx.SendCommitteeTx(bw.Bytes(), false); err != nil {
|
|
return err
|
|
}
|
|
|
|
return wCtx.AwaitTx()
|
|
}
|
|
|
|
func dumpPolicyCmd(cmd *cobra.Command, _ []string) error {
|
|
c, err := helper.GetN3Client(viper.GetViper())
|
|
commonCmd.ExitOnErr(cmd, "can't create N3 client:", err)
|
|
|
|
inv := invoker.New(c, nil)
|
|
policyContract := policy.NewReader(inv)
|
|
|
|
execFee, err := policyContract.GetExecFeeFactor()
|
|
commonCmd.ExitOnErr(cmd, "can't get execution fee factor:", err)
|
|
|
|
feePerByte, err := policyContract.GetFeePerByte()
|
|
commonCmd.ExitOnErr(cmd, "can't get fee per byte:", err)
|
|
|
|
storagePrice, err := policyContract.GetStoragePrice()
|
|
commonCmd.ExitOnErr(cmd, "can't get storage price:", err)
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
tw := tabwriter.NewWriter(buf, 0, 2, 2, ' ', 0)
|
|
|
|
_, _ = tw.Write([]byte(fmt.Sprintf("Execution Fee Factor:\t%d (int)\n", execFee)))
|
|
_, _ = tw.Write([]byte(fmt.Sprintf("Fee Per Byte:\t%d (int)\n", feePerByte)))
|
|
_, _ = tw.Write([]byte(fmt.Sprintf("Storage Price:\t%d (int)\n", storagePrice)))
|
|
|
|
_ = tw.Flush()
|
|
cmd.Print(buf.String())
|
|
|
|
return nil
|
|
}
|