Improve ErrorOutOfGas handling in predicate execution

x/logic/keeper/grpc_query_ask_test.go

@@ -17,8 +17,10 @@ import (
 
 	"github.com/cosmos/cosmos-sdk/baseapp"
 	"github.com/cosmos/cosmos-sdk/testutil"
+	sdk "github.com/cosmos/cosmos-sdk/types"
 	moduletestutil "github.com/cosmos/cosmos-sdk/types/module/testutil"
 	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
+	bankypes "github.com/cosmos/cosmos-sdk/x/bank/types"
 	govtypes "github.com/cosmos/cosmos-sdk/x/gov/types"
 
 	"github.com/axone-protocol/axoned/v8/x/logic"
@@ -137,16 +139,18 @@ func TestGRPCAsk(t *testing.T) {
 				maxGas:        1000,
 				expectedError: "out of gas: logic <ReadPerByte> (1018/1000): limit exceeded",
 			},
+			{
+				query:         "bank_balances(X, Y).",
+				maxGas:        3000,
+				expectedError: "error executing query: out of gas in location: panic: {ValuePerByte}: out of gas: invalid argument",
+			},
 			{
 				query:  "block_height(X).",
 				maxGas: 3000,
 				predicateCosts: map[string]uint64{
 					"block_height/1": 10000,
 				},
-				expectedAnswer: &types.Answer{
-					Variables: []string{"X"},
-					Results:   []types.Result{{Error: "error(resource_error(gas(block_height/1,12353,3000)),block_height/1)"}},
-				},
+				expectedError: "error executing query: out of gas in location: block_height/1: out of gas: invalid argument",
 			},
 			{
 				program: "father(bob, 'élodie').",
@@ -325,6 +329,11 @@ foo(a4).
 					bankKeeper := logictestutil.NewMockBankKeeper(ctrl)
 					fsProvider := logictestutil.NewMockFS(ctrl)
 
+					bankKeeper.EXPECT().GetAccountsBalances(gomock.Any()).Do(func(ctx gocontext.Context) []bankypes.Balance {
+						sdk.UnwrapSDKContext(ctx).GasMeter().ConsumeGas(2000, "ValuePerByte")
+						return nil
+					}).AnyTimes()
+
 					logicKeeper := keeper.NewKeeper(
 						encCfg.Codec,
 						key,

x/logic/keeper/interpreter.go

@@ -16,6 +16,7 @@
 	storetypes "cosmossdk.io/store/types"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
 
 	"github.com/axone-protocol/axoned/v8/x/logic/fs/filtered"
 	"github.com/axone-protocol/axoned/v8/x/logic/interpreter"
@@ -107,12 +108,12 @@
 
 			defer func() {
 				if r := recover(); r != nil {
-					if gasError, ok := r.(storetypes.ErrorOutOfGas); ok {
-						err = engine.ResourceError(prolog2.ResourceGas(gasError.Descriptor, gasMeter.GasConsumed(), gasMeter.Limit()), env)
-						return
+					switch rType := r.(type) {
+					case storetypes.ErrorOutOfGas:
+						err = errorsmod.Wrapf(sdkerrors.ErrOutOfGas, "out of gas in location: %v", rType.Descriptor)
+					default:
+						panic(r)
 					}
-
-					panic(r)
 				}
 			}()
 

x/logic/prolog/error.go

@@ -2,8 +2,6 @@ package prolog
 
 import (
 	"github.com/ichiban/prolog/engine"
-
-	"cosmossdk.io/store/types"
 )
 
 var (
@@ -82,8 +80,6 @@ var (
 	// The module resource is the representation of the module with which the interaction is made.
 	// The module resource is denoted as a compound with the name of the module.
 	AtomResourceModule = engine.NewAtom("resource_module")
-	// AtomResourceGas is the atom denoting the "gas" resource.
-	AtomResourceGas = engine.NewAtom("gas")
 )
 
 // ResourceContext returns a term representing the context resource.
@@ -96,12 +92,6 @@ func ResourceModule(module string) engine.Term {
 	return AtomResourceModule.Apply(engine.NewAtom(module))
 }
 
-// ResourceGas returns a term representing the gas resource with the given descriptor, consumed and limit at the
-// given context.
-func ResourceGas(descriptor string, consumed types.Gas, limit types.Gas) engine.Term {
-	return AtomResourceGas.Apply(engine.NewAtom(descriptor), engine.Integer(int64(consumed)), engine.Integer(int64(limit)))
-}
-
 var (
 	AtomOperationInput   = engine.NewAtom("input")
 	AtomOperationExecute = engine.NewAtom("execute")

x/logic/util/prolog.go

@@ -2,14 +2,19 @@
 
 import (
 	"context"
+	"errors"
 	"strings"
 
 	"github.com/ichiban/prolog"
 	"github.com/ichiban/prolog/engine"
 	"github.com/samber/lo"
 
+	errorsmod "cosmossdk.io/errors"
 	sdkmath "cosmossdk.io/math"
 
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
+
 	"github.com/axone-protocol/axoned/v8/x/logic/types"
 )
 
@@ -48,7 +53,14 @@
 	if callErr != nil {
 		if sdkmath.NewUint(uint64(len(results))).LT(solutionsLimit) {
 			// error is not part of the look-ahead and should be included in the solutions
-			results = append(results, types.Result{Error: callErr.Error()})
+			switch {
+			case errors.Is(callErr, sdkerrors.ErrOutOfGas):
+				return nil, callErr
+			case sdk.UnwrapSDKContext(ctx).GasMeter().IsOutOfGas():
+				return nil, errorsmod.Wrapf(sdkerrors.ErrOutOfGas, "out of gas in location: %v", callErr.Error())
+			default:
+				results = append(results, types.Result{Error: callErr.Error()})
+			}
 		} else {
 			// error is part of the look-ahead, so let's consider that there's one more solution
 			count = count.Incr()