forked from TrueCloudLab/frostfs-node
aarifullin
89a68ca836
* Update go.mod with a new version of policy-engine pacakge. * Adapt SwitchRPCGuardedActor to ContractStorage interface. * Fix `frostfs-adm` util. Signed-off-by: Airat Arifullin <a.arifullin@yadro.com>
144 lines
6.1 KiB
Go
144 lines
6.1 KiB
Go
package client
|
|
|
|
import (
|
|
"github.com/google/uuid"
|
|
"github.com/nspcc-dev/neo-go/pkg/config/netmode"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/neorpc/result"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/actor"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/invoker"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
)
|
|
|
|
type actorProvider interface {
|
|
GetActor() *actor.Actor
|
|
GetRPCActor() actor.RPCActor
|
|
}
|
|
|
|
// Client switches an established connection with neo-go if it is broken.
|
|
// This leads to an invalidation of an rpc actor within Client. That means the
|
|
// components that are initilized with the rpc actor may unintentionally use
|
|
// it when it is already invalidated. SwitchRPCGuardedActor is used to prevent
|
|
// this situation, getting the rpc actor from Client.
|
|
type SwitchRPCGuardedActor struct {
|
|
actorProvider actorProvider
|
|
}
|
|
|
|
func NewSwitchRPCGuardedActor(c *Client) *SwitchRPCGuardedActor {
|
|
return &SwitchRPCGuardedActor{
|
|
actorProvider: c,
|
|
}
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) Call(contract util.Uint160, operation string, params ...any) (*result.Invoke, error) {
|
|
return a.actorProvider.GetActor().Call(contract, operation, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) CalculateNetworkFee(tx *transaction.Transaction) (int64, error) {
|
|
return a.actorProvider.GetActor().CalculateNetworkFee(tx)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) CalculateValidUntilBlock() (uint32, error) {
|
|
return a.actorProvider.GetActor().CalculateValidUntilBlock()
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) GetBlockCount() (uint32, error) {
|
|
return a.actorProvider.GetActor().GetBlockCount()
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) GetNetwork() netmode.Magic {
|
|
return a.actorProvider.GetActor().GetNetwork()
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) GetVersion() result.Version {
|
|
return a.actorProvider.GetActor().GetVersion()
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeCall(contract util.Uint160, method string, params ...any) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeCall(contract, method, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeRun(script []byte) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeRun(script)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeTunedCall(contract util.Uint160, method string, attrs []transaction.Attribute, txHook actor.TransactionCheckerModifier, params ...any) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeTunedCall(contract, method, attrs, txHook, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeTunedRun(script []byte, attrs []transaction.Attribute, txHook actor.TransactionCheckerModifier) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeTunedRun(script, attrs, txHook)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeUncheckedRun(script []byte, sysfee int64, attrs []transaction.Attribute, txHook actor.TransactionModifier) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeUncheckedRun(script, sysfee, attrs, txHook)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeUnsignedCall(contract util.Uint160, method string, attrs []transaction.Attribute, params ...any) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeUnsignedCall(contract, method, attrs, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeUnsignedRun(script []byte, attrs []transaction.Attribute) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeUnsignedRun(script, attrs)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) MakeUnsignedUncheckedRun(script []byte, sysFee int64, attrs []transaction.Attribute) (*transaction.Transaction, error) {
|
|
return a.actorProvider.GetActor().MakeUnsignedUncheckedRun(script, sysFee, attrs)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) Send(tx *transaction.Transaction) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().Send(tx)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) SendCall(contract util.Uint160, method string, params ...any) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().SendCall(contract, method, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) SendRun(script []byte) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().SendRun(script)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) SendTunedCall(contract util.Uint160, method string, attrs []transaction.Attribute, txHook actor.TransactionCheckerModifier, params ...any) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().SendTunedCall(contract, method, attrs, txHook, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) SendTunedRun(script []byte, attrs []transaction.Attribute, txHook actor.TransactionCheckerModifier) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().SendTunedRun(script, attrs, txHook)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) SendUncheckedRun(script []byte, sysfee int64, attrs []transaction.Attribute, txHook actor.TransactionModifier) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().SendUncheckedRun(script, sysfee, attrs, txHook)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) Sender() util.Uint160 {
|
|
return a.actorProvider.GetActor().Sender()
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) Sign(tx *transaction.Transaction) error {
|
|
return a.actorProvider.GetActor().Sign(tx)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) SignAndSend(tx *transaction.Transaction) (util.Uint256, uint32, error) {
|
|
return a.actorProvider.GetActor().SignAndSend(tx)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) CallAndExpandIterator(contract util.Uint160, method string, maxItems int, params ...any) (*result.Invoke, error) {
|
|
return a.actorProvider.GetActor().CallAndExpandIterator(contract, method, maxItems, params...)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) TerminateSession(sessionID uuid.UUID) error {
|
|
return a.actorProvider.GetActor().TerminateSession(sessionID)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) TraverseIterator(sessionID uuid.UUID, iterator *result.Iterator, num int) ([]stackitem.Item, error) {
|
|
return a.actorProvider.GetActor().TraverseIterator(sessionID, iterator, num)
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) GetRPCActor() actor.RPCActor {
|
|
return a.actorProvider.GetRPCActor()
|
|
}
|
|
|
|
func (a *SwitchRPCGuardedActor) GetRPCInvoker() invoker.RPCInvoke {
|
|
return a.actorProvider.GetRPCActor()
|
|
}
|