101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package runtime
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"cosmossdk.io/core/gas"
|
|
storetypes "cosmossdk.io/store/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
var _ gas.Service = GasService{}
|
|
|
|
type GasService struct{}
|
|
|
|
func (g GasService) GetGasMeter(ctx context.Context) gas.Meter {
|
|
return CoreGasmeter{gm: sdk.UnwrapSDKContext(ctx).GasMeter()}
|
|
}
|
|
|
|
func (g GasService) GetBlockGasMeter(ctx context.Context) gas.Meter {
|
|
return CoreGasmeter{gm: sdk.UnwrapSDKContext(ctx).BlockGasMeter()}
|
|
}
|
|
|
|
func (g GasService) WithGasMeter(ctx context.Context, meter gas.Meter) context.Context {
|
|
return sdk.UnwrapSDKContext(ctx).WithGasMeter(SDKGasMeter{gm: meter})
|
|
}
|
|
|
|
func (g GasService) WithBlockGasMeter(ctx context.Context, meter gas.Meter) context.Context {
|
|
return sdk.UnwrapSDKContext(ctx).WithGasMeter(SDKGasMeter{gm: meter})
|
|
}
|
|
|
|
// ______________________________________________________________________________________________
|
|
// Gas Meter Wrappers
|
|
// ______________________________________________________________________________________________
|
|
|
|
// SDKGasMeter is a wrapper around the SDK's GasMeter that implements the GasMeter interface.
|
|
type SDKGasMeter struct {
|
|
gm gas.Meter
|
|
}
|
|
|
|
func (gm SDKGasMeter) GasConsumed() storetypes.Gas {
|
|
return gm.gm.Remaining()
|
|
}
|
|
|
|
func (gm SDKGasMeter) GasConsumedToLimit() storetypes.Gas {
|
|
if gm.IsPastLimit() {
|
|
return gm.gm.Limit()
|
|
}
|
|
return gm.gm.Remaining()
|
|
}
|
|
|
|
func (gm SDKGasMeter) GasRemaining() storetypes.Gas {
|
|
return gm.gm.Remaining()
|
|
}
|
|
|
|
func (gm SDKGasMeter) Limit() storetypes.Gas {
|
|
return gm.gm.Limit()
|
|
}
|
|
|
|
func (gm SDKGasMeter) ConsumeGas(amount storetypes.Gas, descriptor string) {
|
|
gm.gm.Consume(amount, descriptor)
|
|
}
|
|
|
|
func (gm SDKGasMeter) RefundGas(amount storetypes.Gas, descriptor string) {
|
|
gm.gm.Refund(amount, descriptor)
|
|
}
|
|
|
|
func (gm SDKGasMeter) IsPastLimit() bool {
|
|
return gm.gm.Remaining() <= gm.gm.Limit()
|
|
}
|
|
|
|
func (gm SDKGasMeter) IsOutOfGas() bool {
|
|
return gm.gm.Remaining() >= gm.gm.Limit()
|
|
}
|
|
|
|
func (gm SDKGasMeter) String() string {
|
|
return fmt.Sprintf("BasicGasMeter:\n limit: %d\n consumed: %d", gm.gm.Limit(), gm.gm.Remaining())
|
|
}
|
|
|
|
// CoreGasmeter is a wrapper around the SDK's GasMeter that implements the GasMeter interface.
|
|
type CoreGasmeter struct {
|
|
gm storetypes.GasMeter
|
|
}
|
|
|
|
func (cgm CoreGasmeter) Consume(amount gas.Gas, descriptor string) {
|
|
cgm.gm.ConsumeGas(amount, descriptor)
|
|
}
|
|
|
|
func (cgm CoreGasmeter) Refund(amount gas.Gas, descriptor string) {
|
|
cgm.gm.RefundGas(amount, descriptor)
|
|
}
|
|
|
|
func (cgm CoreGasmeter) Remaining() gas.Gas {
|
|
return cgm.gm.GasRemaining()
|
|
}
|
|
|
|
func (cgm CoreGasmeter) Limit() gas.Gas {
|
|
return cgm.gm.Limit()
|
|
}
|