refactor: avoid breaking change due to cosmos#16415 included in v0.50 (…

…cosmos#16430)
InjectiveLabs · Aug 15, 2023 · 576d860 · 576d860
1 parent 7c5f95c
commit 576d860
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/baseapp/circuit.go b/baseapp/circuit.go
@@ -1,10 +1,8 @@
 package baseapp
 
-import (
-	sdk "github.com/cosmos/cosmos-sdk/types"
-)
+import "context"
 
 // CircuitBreaker is an interface that defines the methods for a circuit breaker.
 type CircuitBreaker interface {
-	IsAllowed(ctx sdk.Context, typeURL string) bool
+	IsAllowed(ctx context.Context, typeURL string) (bool, error)
 }
diff --git a/baseapp/msg_service_router.go b/baseapp/msg_service_router.go
@@ -124,7 +124,13 @@ func (msr *MsgServiceRouter) RegisterService(sd *grpc.ServiceDesc, handler inter
 
 			if msr.circuitBreaker != nil {
 				msgURL := sdk.MsgTypeURL(req)
-				if !msr.circuitBreaker.IsAllowed(ctx, msgURL) {
+
+				isAllowed, err := msr.circuitBreaker.IsAllowed(ctx, msgURL)
+				if err != nil {
+					return nil, err
+				}
+
+				if !isAllowed {
 					return nil, fmt.Errorf("circuit breaker disables execution of this message: %s", msgURL)
 				}
 			}