forked from TrueCloudLab/frostfs-node
102 lines
3 KiB
Go
102 lines
3 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ecdsa"
|
|
"crypto/elliptic"
|
|
"crypto/rand"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/common"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags"
|
|
commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/network"
|
|
tracing "git.frostfs.info/TrueCloudLab/frostfs-observability/tracing/grpc"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
var errInvalidEndpoint = errors.New("provided RPC endpoint is incorrect")
|
|
|
|
// GetSDKClientByFlag returns default frostfs-sdk-go client using the specified flag for the address.
|
|
// On error, outputs to stderr of cmd and exits with non-zero code.
|
|
func GetSDKClientByFlag(cmd *cobra.Command, key *ecdsa.PrivateKey, endpointFlag string) *client.Client {
|
|
cli, err := getSDKClientByFlag(cmd, key, endpointFlag)
|
|
if err != nil {
|
|
commonCmd.ExitOnErr(cmd, "can't create API client: %w", err)
|
|
}
|
|
return cli
|
|
}
|
|
|
|
func getSDKClientByFlag(cmd *cobra.Command, key *ecdsa.PrivateKey, endpointFlag string) (*client.Client, error) {
|
|
var addr network.Address
|
|
|
|
err := addr.FromString(viper.GetString(endpointFlag))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("%v: %w", errInvalidEndpoint, err)
|
|
}
|
|
return GetSDKClient(cmd.Context(), cmd, key, addr)
|
|
}
|
|
|
|
// GetSDKClient returns default frostfs-sdk-go client.
|
|
func GetSDKClient(ctx context.Context, cmd *cobra.Command, key *ecdsa.PrivateKey, addr network.Address) (*client.Client, error) {
|
|
var c client.Client
|
|
|
|
prmInit := client.PrmInit{
|
|
Key: *key,
|
|
}
|
|
|
|
prmDial := client.PrmDial{
|
|
Endpoint: addr.URIAddr(),
|
|
GRPCDialOptions: []grpc.DialOption{
|
|
grpc.WithChainUnaryInterceptor(tracing.NewUnaryClientInteceptor()),
|
|
grpc.WithChainStreamInterceptor(tracing.NewStreamClientInterceptor()),
|
|
},
|
|
}
|
|
if timeout := viper.GetDuration(commonflags.Timeout); timeout > 0 {
|
|
// In CLI we can only set a timeout for the whole operation.
|
|
// By also setting stream timeout we ensure that no operation hands
|
|
// for too long.
|
|
prmDial.DialTimeout = timeout
|
|
prmDial.StreamTimeout = timeout
|
|
|
|
common.PrintVerbose(cmd, "Set request timeout to %s.", timeout)
|
|
}
|
|
|
|
c.Init(prmInit)
|
|
|
|
if err := c.Dial(ctx, prmDial); err != nil {
|
|
return nil, fmt.Errorf("can't init SDK client: %w", err)
|
|
}
|
|
|
|
return &c, nil
|
|
}
|
|
|
|
// GetCurrentEpoch returns current epoch.
|
|
func GetCurrentEpoch(ctx context.Context, cmd *cobra.Command, endpoint string) (uint64, error) {
|
|
var addr network.Address
|
|
|
|
if err := addr.FromString(endpoint); err != nil {
|
|
return 0, fmt.Errorf("can't parse RPC endpoint: %w", err)
|
|
}
|
|
|
|
key, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("can't generate key to sign query: %w", err)
|
|
}
|
|
|
|
c, err := GetSDKClient(ctx, cmd, key, addr)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
ni, err := c.NetworkInfo(ctx, client.PrmNetworkInfo{})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return ni.Info().CurrentEpoch(), nil
|
|
}
|