refactored code

zk/txpool/policy.go

@@ -25,16 +25,38 @@ func containsPolicy(policies []byte, policy PolicyName) bool {
 }
 
 // create a method checkpolicy to check an address according to passed policy in the method
-func (p *TxPool) checkPolicy(addr common.Address, policy PolicyName, mode string) (bool, error) {
-	var table string
+func (p *TxPool) checkPolicy(addr common.Address, policy PolicyName) (bool, error) {
+	// Retrieve the mode configuration
+	var mode string
+	err := p.aclDB.View(context.TODO(), func(tx kv.Tx) error {
+		value, err := tx.GetOne(Config, []byte("mode"))
+		if err != nil {
+			return err
+		}
+		if value == nil || string(value) == "disabled" {
+			mode = "disabled"
+			return nil
+		}
+
+		mode = string(value)
+		return nil
+	})
+	if err != nil {
+		return false, err
+	}
+
+	if mode == "disabled" {
+		return true, nil
+	}
+
+	// Determine the appropriate table based on the mode
+	table := Blacklist
 	if mode == "allowlist" {
 		table = Whitelist
-	} else {
-		table = Blacklist
 	}
 
 	var policyBytes []byte
-	err := p.aclDB.View(context.TODO(), func(tx kv.Tx) error {
+	err = p.aclDB.View(context.TODO(), func(tx kv.Tx) error {
 		value, err := tx.GetOne(table, addr.Bytes())
 		if err != nil {
 			return err

zk/txpool/pool.go

@@ -713,37 +713,25 @@ func (p *TxPool) validateTx(txn *types.TxSlot, isLocal bool, stateCache kvcache.
 		return InsufficientFunds
 	}
 
-	var mode string
-	p.aclDB.View(context.TODO(), func(tx kv.Tx) error {
-		value, err := tx.GetOne(Config, []byte("mode"))
+	switch resolvePolicy(txn) {
+	case SendTx:
+		var allow bool
+		allow, err := p.checkPolicy(from, SendTx)
 		if err != nil {
 			panic(err)
 		}
-		mode = string(value)
-		return nil
-	})
-
-	if mode != "disabled" {
-		switch resolvePolicy(txn) {
-		case SendTx:
-			var allow bool
-			allow, err := p.checkPolicy(from, SendTx, mode)
-			if err != nil {
-				panic(err)
-			}
-			if !allow {
-				return SenderDisallowedSendTx
-			}
-		case Deploy:
-			var allow bool
-			// check that sender may deploy contracts
-			allow, err := p.checkPolicy(from, Deploy, mode)
-			if err != nil {
-				panic(err)
-			}
-			if !allow {
-				return SenderDisallowedDeploy
-			}
+		if !allow {
+			return SenderDisallowedSendTx
+		}
+	case Deploy:
+		var allow bool
+		// check that sender may deploy contracts
+		allow, err := p.checkPolicy(from, Deploy)
+		if err != nil {
+			panic(err)
+		}
+		if !allow {
+			return SenderDisallowedDeploy
 		}
 	}