edmand46
116 lines
3.3 KiB
Go
116 lines
3.3 KiB
Go
package rudder
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"hub.rudder.build/rudder/rudder-module-go/abi"
|
|
)
|
|
|
|
type Ledger struct{}
|
|
|
|
func (Ledger) Balance(parameters abi.LedgerBalanceParameters) (*abi.LedgerBalanceResult, error) {
|
|
return callResult[abi.LedgerBalanceResult](parameters)
|
|
}
|
|
|
|
func (Ledger) Balances(parameters abi.LedgerBalancesParameters) (*abi.LedgerBalancesResult, error) {
|
|
return callResult[abi.LedgerBalancesResult](parameters)
|
|
}
|
|
|
|
func (Ledger) Credit(parameters abi.LedgerCreditParameters) (*abi.LedgerCreditResult, error) {
|
|
return callResult[abi.LedgerCreditResult](parameters)
|
|
}
|
|
|
|
func (Ledger) Debit(parameters abi.LedgerDebitParameters) (*abi.LedgerDebitResult, error) {
|
|
return callResult[abi.LedgerDebitResult](parameters)
|
|
}
|
|
|
|
func (Ledger) History(parameters abi.LedgerHistoryParameters) (*abi.LedgerHistoryResult, error) {
|
|
return callResult[abi.LedgerHistoryResult](parameters)
|
|
}
|
|
|
|
type Inventory struct{}
|
|
|
|
func (Inventory) Get(parameters abi.InventoryGetParameters) (*abi.InventoryGetResult, error) {
|
|
return callResult[abi.InventoryGetResult](parameters)
|
|
}
|
|
|
|
func (Inventory) List(parameters abi.InventoryListParameters) (*abi.InventoryListResult, error) {
|
|
return callResult[abi.InventoryListResult](parameters)
|
|
}
|
|
|
|
func (Inventory) Has(parameters abi.InventoryHasParameters) (*abi.InventoryHasResult, error) {
|
|
return callResult[abi.InventoryHasResult](parameters)
|
|
}
|
|
|
|
func (Inventory) Grant(parameters abi.InventoryGrantParameters) error {
|
|
return call(parameters, nil)
|
|
}
|
|
|
|
func (Inventory) Consume(parameters abi.InventoryConsumeParameters) error {
|
|
return call(parameters, nil)
|
|
}
|
|
|
|
type Storage struct{}
|
|
|
|
func (Storage) Get(parameters abi.StorageGetParameters) (*abi.StorageGetResult, error) {
|
|
return callResult[abi.StorageGetResult](parameters)
|
|
}
|
|
|
|
func (Storage) Set(parameters abi.StorageSetParameters) error {
|
|
return call(parameters, nil)
|
|
}
|
|
|
|
type ProjectStorage struct{}
|
|
|
|
func (ProjectStorage) Get(parameters abi.ProjectStorageGetParameters) (*abi.ProjectStorageGetResult, error) {
|
|
return callResult[abi.ProjectStorageGetResult](parameters)
|
|
}
|
|
|
|
func (ProjectStorage) Set(parameters abi.ProjectStorageSetParameters) error {
|
|
return call(parameters, nil)
|
|
}
|
|
|
|
type Segments struct{}
|
|
|
|
func (Segments) Match(parameters abi.SegmentsMatchParameters) (*abi.SegmentsMatchResult, error) {
|
|
return callResult[abi.SegmentsMatchResult](parameters)
|
|
}
|
|
|
|
func (Segments) Bucket(parameters abi.SegmentsBucketParameters) (*abi.SegmentsBucketResult, error) {
|
|
return callResult[abi.SegmentsBucketResult](parameters)
|
|
}
|
|
|
|
type Events struct{}
|
|
|
|
func (Events) Emit(parameters abi.EventsEmitParameters) error {
|
|
return call(parameters, nil)
|
|
}
|
|
|
|
type Data struct{}
|
|
|
|
func (Data) Exec(parameters abi.DataExecParameters) (*abi.DataExecResult, error) {
|
|
return callResult[abi.DataExecResult](parameters)
|
|
}
|
|
|
|
type Modules struct{}
|
|
|
|
func (Modules) Call(module string, function string, args any) (json.RawMessage, error) {
|
|
body, err := json.Marshal(args)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var result abi.ModulesCallResult
|
|
if err := call(abi.ModulesCallParameters{Module: module, Function: function, Arguments: body}, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return result.Value, nil
|
|
}
|
|
|
|
func callResult[Result any](parameters any) (*Result, error) {
|
|
var result Result
|
|
if err := call(parameters, &result); err != nil {
|
|
return nil, err
|
|
}
|
|
return &result, nil
|
|
}
|