cleanup

yihuang · Jun 26, 2024 · 83d2f8e · 83d2f8e
1 parent 1cbd0c9
commit 83d2f8e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/types/mempool/mempool.go b/types/mempool/mempool.go
@@ -19,6 +19,10 @@ func NewMempoolTx(tx sdk.Tx, gasWanted uint64) MempoolTx {
 	}
 }
 
+type GasTx interface {
+	GetGas() uint64
+}
+
 type Mempool interface {
 	// Insert attempts to insert a Tx into the app-side mempool returning
 	// an error upon failure.

diff --git a/types/mempool/priority_nonce.go b/types/mempool/priority_nonce.go
@@ -278,9 +278,7 @@ func (mp *PriorityNonceMempool[C]) InsertWithGasWanted(ctx context.Context, tx s
 
 func (mp *PriorityNonceMempool[C]) Insert(ctx context.Context, tx sdk.Tx) error {
 	var gasLimit uint64
-	if gasTx, ok := tx.(interface {
-		GetGas() uint64
-	}); ok {
+	if gasTx, ok := tx.(GasTx); ok {
 		gasLimit = gasTx.GetGas()
 	}
 

diff --git a/types/mempool/sender_nonce.go b/types/mempool/sender_nonce.go
@@ -157,9 +157,7 @@ func (snm *SenderNonceMempool) InsertWithGasWanted(_ context.Context, tx sdk.Tx,
 
 func (mp *SenderNonceMempool) Insert(ctx context.Context, tx sdk.Tx) error {
 	var gasLimit uint64
-	if gasTx, ok := tx.(interface {
-		GetGas() uint64
-	}); ok {
+	if gasTx, ok := tx.(GasTx); ok {
 		gasLimit = gasTx.GetGas()
 	}