do not return duplicate default salt wallet (#48)

AvaProtocol · Dec 7, 2024 · e4c5ab1 · e4c5ab1
1 parent 049bc66
commit e4c5ab1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/core/taskengine/engine.go b/core/taskengine/engine.go
@@ -43,6 +43,8 @@ var (
 	// These macro are define in our aggregator yaml config file under `macros`
 	macroEnvs map[string]string
 	cache     *bigcache.BigCache
+
+	defaultSalt = big.NewInt(0)
 )
 
 // Set a global logger for task engine
@@ -169,8 +171,7 @@ func (n *Engine) MustStart() {
 }
 
 func (n *Engine) GetSmartWallets(owner common.Address) ([]*avsproto.SmartWallet, error) {
-	salt := big.NewInt(0)
-	sender, err := aa.GetSenderAddress(rpcConn, owner, salt)
+	sender, err := aa.GetSenderAddress(rpcConn, owner, defaultSalt)
 	if err != nil {
 		return nil, status.Errorf(codes.Code(avsproto.Error_SmartWalletNotFoundError), SmartAccountCreationError)
 	}
@@ -180,7 +181,7 @@ func (n *Engine) GetSmartWallets(owner common.Address) ([]*avsproto.SmartWallet,
 		&avsproto.SmartWallet{
 			Address: sender.String(),
 			Factory: n.smartWalletConfig.FactoryAddress.String(),
-			Salt:    salt.String(),
+			Salt:    defaultSalt.String(),
 		},
 	}
 
@@ -195,6 +196,10 @@ func (n *Engine) GetSmartWallets(owner common.Address) ([]*avsproto.SmartWallet,
 		w := &model.SmartWallet{}
 		w.FromStorageData(item.Value)
 
+		if w.Salt.Cmp(defaultSalt) == 0 {
+			continue
+		}
+
 		wallets = append(wallets, &avsproto.SmartWallet{
 			Address: w.Address.String(),
 			Factory: w.Factory.String(),