forked from TrueCloudLab/frostfs-node
Airat Arifullin
9b13a18aac
* Update version within go.mod; * Fix deprecated frostfs-api-go/v2 package and use frostfs-sdk-go/api instead. Signed-off-by: Airat Arifullin <a.arifullin@yadro.com>
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package accounting
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/balance"
|
|
accountingSvc "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/accounting"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/accounting"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
|
)
|
|
|
|
type morphExecutor struct {
|
|
client *balance.Client
|
|
}
|
|
|
|
func NewExecutor(client *balance.Client) accountingSvc.ServiceExecutor {
|
|
return &morphExecutor{
|
|
client: client,
|
|
}
|
|
}
|
|
|
|
func (s *morphExecutor) Balance(_ context.Context, body *accounting.BalanceRequestBody) (*accounting.BalanceResponseBody, error) {
|
|
idV2 := body.GetOwnerID()
|
|
if idV2 == nil {
|
|
return nil, errors.New("missing account")
|
|
}
|
|
|
|
var id user.ID
|
|
|
|
err := id.ReadFromV2(*idV2)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid account: %w", err)
|
|
}
|
|
|
|
amount, err := s.client.BalanceOf(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
balancePrecision, err := s.client.Decimals()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dec := new(accounting.Decimal)
|
|
dec.SetValue(amount.Int64())
|
|
dec.SetPrecision(balancePrecision)
|
|
|
|
res := new(accounting.BalanceResponseBody)
|
|
res.SetBalance(dec)
|
|
|
|
return res, nil
|
|
}
|