chore: improve log for shared handler saddle (#40)

KyberNetwork · Jun 22, 2023 · 57d1709 · 57d1709
1 parent 964aab8
commit 57d1709
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/pkg/source/handler.go b/pkg/source/handler.go
@@ -2,6 +2,7 @@ package source
 
 import (
 	"fmt"
+
 	"github.com/KyberNetwork/ethrpc"
 	"github.com/KyberNetwork/kyberswap-dex-lib/pkg/source/balancer"
 	"github.com/KyberNetwork/kyberswap-dex-lib/pkg/source/biswap"
@@ -386,7 +387,9 @@ func NewPoolTrackerHandler(
 	case oneswap.DexTypeOneSwap:
 		return oneswap.NewPoolTracker(ethrpcClient), nil
 	case saddle.DexTypeSaddle:
-		return saddle.NewPoolTracker(ethrpcClient), nil
+		var cfg saddle.Config
+		cfg.DexID = scanDexCfg.Id
+		return saddle.NewPoolTracker(&cfg, ethrpcClient), nil
 	case nerve.DexTypeNerve:
 		var cfg nerve.Config
 		err := PropertiesToStruct(scanDexCfg.Properties, &cfg)

diff --git a/pkg/source/saddle/pool_tracker.go b/pkg/source/saddle/pool_tracker.go
@@ -13,17 +13,19 @@ import (
 )
 
 type PoolTracker struct {
+	config       *Config
 	ethrpcClient *ethrpc.Client
 }
 
-func NewPoolTracker(ethrpcClient *ethrpc.Client) *PoolTracker {
+func NewPoolTracker(cfg *Config, ethrpcClient *ethrpc.Client) *PoolTracker {
 	return &PoolTracker{
+		config:       cfg,
 		ethrpcClient: ethrpcClient,
 	}
 }
 
 func (d *PoolTracker) GetNewPoolState(ctx context.Context, p entity.Pool) (entity.Pool, error) {
-	logger.Infof("[Saddle] Start getting new state of pool: %v", p.Address)
+	logger.Infof("[%s] Start getting new state of pool: %v", d.config.DexID, p.Address)
 
 	var (
 		lpSupply    *big.Int
@@ -92,7 +94,7 @@ func (d *PoolTracker) GetNewPoolState(ctx context.Context, p entity.Pool) (entit
 	p.Reserves = reserves
 	p.Timestamp = time.Now().Unix()
 
-	logger.Infof("[Saddle] Finish updating state of pool: %v", p.Address)
+	logger.Infof("[%s] Finish updating state of pool: %v", d.config.DexID, p.Address)
 
 	return p, nil
 }
diff --git a/pkg/source/saddle/pools_list_updater.go b/pkg/source/saddle/pools_list_updater.go
@@ -63,7 +63,7 @@ func (d *PoolsListUpdater) initPools(ctx context.Context) ([]entity.Pool, error)
 		return nil, err
 	}
 
-	logger.Infof("got %v from pool path config file", len(poolItems))
+	logger.Infof("[%s] got %v from pool path config file", d.config.DexID, len(poolItems))
 
 	pools, err := d.processBatch(ctx, poolItems)
 	if err != nil {