package innerring import ( "context" "encoding/hex" "math/big" auditAPI "github.com/nspcc-dev/neofs-api-go/pkg/audit" containerAPI "github.com/nspcc-dev/neofs-api-go/pkg/container" netmapAPI "github.com/nspcc-dev/neofs-api-go/pkg/netmap" "github.com/nspcc-dev/neofs-api-go/pkg/object" "github.com/nspcc-dev/neofs-api-go/pkg/owner" "github.com/nspcc-dev/neofs-api-go/pkg/storagegroup" crypto "github.com/nspcc-dev/neofs-crypto" "github.com/nspcc-dev/neofs-node/pkg/core/container" "github.com/nspcc-dev/neofs-node/pkg/core/netmap" "github.com/nspcc-dev/neofs-node/pkg/innerring/processors/settlement/audit" "github.com/nspcc-dev/neofs-node/pkg/innerring/processors/settlement/basic" "github.com/nspcc-dev/neofs-node/pkg/innerring/processors/settlement/common" auditClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/audit/wrapper" balanceClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/balance/wrapper" "github.com/nspcc-dev/neofs-node/pkg/morph/client/container/wrapper" containerClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/container/wrapper" "github.com/nspcc-dev/neofs-node/pkg/util/logger" "github.com/pkg/errors" "go.uber.org/zap" ) type globalConfig interface { BasicIncomeRate() (uint64, error) } type settlementDeps struct { log *logger.Logger cnrSrc container.Source auditClient *auditClient.ClientWrapper nmSrc netmap.Source clientCache *ClientCache balanceClient *balanceClient.Wrapper } type auditSettlementDeps struct { *settlementDeps } type basicIncomeSettlementDeps struct { *settlementDeps cnrClient *containerClient.Wrapper cfg globalConfig } type basicSettlementConstructor struct { dep *basicIncomeSettlementDeps } type auditSettlementCalculator audit.Calculator type containerWrapper containerAPI.Container type nodeInfoWrapper struct { ni *netmapAPI.Node } type sgWrapper storagegroup.StorageGroup func (s *sgWrapper) Size() uint64 { return (*storagegroup.StorageGroup)(s).ValidationDataSize() } func (n nodeInfoWrapper) PublicKey() []byte { return n.ni.PublicKey() } func (n nodeInfoWrapper) Price() *big.Int { return big.NewInt(int64(n.ni.Price)) } func (c *containerWrapper) Owner() *owner.ID { return (*containerAPI.Container)(c).OwnerID() } func (s settlementDeps) AuditResultsForEpoch(epoch uint64) ([]*auditAPI.Result, error) { idList, err := s.auditClient.ListAuditResultIDByEpoch(epoch) if err != nil { return nil, errors.Wrap(err, "could not list audit results in sidechain") } res := make([]*auditAPI.Result, 0, len(idList)) for i := range idList { r, err := s.auditClient.GetAuditResult(idList[i]) if err != nil { return nil, errors.Wrap(err, "could not get audit result") } res = append(res, r) } return res, nil } func (s settlementDeps) ContainerInfo(cid *containerAPI.ID) (common.ContainerInfo, error) { cnr, err := s.cnrSrc.Get(cid) if err != nil { return nil, errors.Wrap(err, "could not get container from storage") } return (*containerWrapper)(cnr), nil } func (s settlementDeps) buildContainer(e uint64, cid *containerAPI.ID) (netmapAPI.ContainerNodes, *netmapAPI.Netmap, error) { var ( nm *netmapAPI.Netmap err error ) if e > 0 { nm, err = s.nmSrc.GetNetMapByEpoch(e) } else { nm, err = netmap.GetLatestNetworkMap(s.nmSrc) } if err != nil { return nil, nil, errors.Wrap(err, "could not get network map from storage") } cnr, err := s.cnrSrc.Get(cid) if err != nil { return nil, nil, errors.Wrap(err, "could not get container from sidechain") } cn, err := nm.GetContainerNodes( cnr.PlacementPolicy(), cid.ToV2().GetValue(), // may be replace pivot calculation to neofs-api-go ) if err != nil { return nil, nil, errors.Wrap(err, "could not calculate container nodes") } return cn, nm, nil } func (s settlementDeps) ContainerNodes(e uint64, cid *containerAPI.ID) ([]common.NodeInfo, error) { cn, _, err := s.buildContainer(e, cid) if err != nil { return nil, err } ns := cn.Flatten() res := make([]common.NodeInfo, 0, len(ns)) for i := range ns { res = append(res, &nodeInfoWrapper{ ni: ns[i], }) } return res, nil } func (s settlementDeps) SGInfo(addr *object.Address) (audit.SGInfo, error) { cn, nm, err := s.buildContainer(0, addr.ContainerID()) if err != nil { return nil, err } sg, err := s.clientCache.getSG(context.Background(), addr, nm, cn) if err != nil { return nil, err } return (*sgWrapper)(sg), nil } func (s settlementDeps) ResolveKey(ni common.NodeInfo) (*owner.ID, error) { w, err := owner.NEO3WalletFromPublicKey(crypto.UnmarshalPublicKey(ni.PublicKey())) if err != nil { return nil, err } id := owner.NewID() id.SetNeo3Wallet(w) return id, nil } var ( transferAuditDetails = []byte("settlement-audit") basicIncomeAuditDetails = []byte("settlement-basic-income") ) func (s settlementDeps) transfer(sender, recipient *owner.ID, amount *big.Int, details []byte) { log := s.log.With( zap.Stringer("sender", sender), zap.Stringer("recipient", recipient), zap.Stringer("amount (GASe-12)", amount), ) if !amount.IsInt64() { s.log.Error("amount can not be represented as an int64") return } if err := s.balanceClient.TransferX(balanceClient.TransferPrm{ Amount: amount.Int64(), From: sender, To: recipient, Details: details, }); err != nil { log.Error("could not send transfer transaction for audit", zap.String("error", err.Error()), ) return } log.Debug("transfer transaction for audit was successfully sent") } func (a auditSettlementDeps) Transfer(sender, recipient *owner.ID, amount *big.Int) { a.transfer(sender, recipient, amount, transferAuditDetails) } func (b basicIncomeSettlementDeps) Transfer(sender, recipient *owner.ID, amount *big.Int) { b.transfer(sender, recipient, amount, basicIncomeAuditDetails) } func (b basicIncomeSettlementDeps) BasicRate() (uint64, error) { return b.cfg.BasicIncomeRate() } func (b basicIncomeSettlementDeps) Estimations(epoch uint64) ([]*wrapper.Estimations, error) { estimationIDs, err := b.cnrClient.ListLoadEstimationsByEpoch(epoch) if err != nil { return nil, err } result := make([]*wrapper.Estimations, 0, len(estimationIDs)) for i := range estimationIDs { estimation, err := b.cnrClient.GetUsedSpaceEstimations(estimationIDs[i]) if err != nil { b.log.Warn("can't get used space estimation", zap.String("estimation_id", hex.EncodeToString(estimationIDs[i])), zap.String("error", err.Error())) continue } result = append(result, estimation) } return result, nil } func (b basicIncomeSettlementDeps) Balance(id *owner.ID) (*big.Int, error) { return b.balanceClient.BalanceOf(id) } func (s *auditSettlementCalculator) ProcessAuditSettlements(epoch uint64) { (*audit.Calculator)(s).Calculate(&audit.CalculatePrm{ Epoch: epoch, }) } func (b *basicSettlementConstructor) CreateContext(epoch uint64) (*basic.IncomeSettlementContext, error) { return basic.NewIncomeSettlementContext(&basic.IncomeSettlementContextPrms{ Log: b.dep.log, Epoch: epoch, Rate: b.dep, Estimations: b.dep, Balances: b.dep, Container: b.dep, Placement: b.dep, Exchange: b.dep, Accounts: b.dep, }) }