feat: script to fix db

scripts/fix_pool_ledger_v3/main.go

@@ -0,0 +1,85 @@
+package main
+
+import (
+	"github.com/Gearbox-protocol/go-liquidator/ethclient"
+	"github.com/Gearbox-protocol/sdk-go/core"
+	"github.com/Gearbox-protocol/sdk-go/core/schemas"
+	"github.com/Gearbox-protocol/sdk-go/log"
+	"github.com/Gearbox-protocol/sdk-go/pkg"
+	"github.com/Gearbox-protocol/third-eye/config"
+	"github.com/Gearbox-protocol/third-eye/ds"
+	"github.com/Gearbox-protocol/third-eye/ds/dc_wrapper"
+	"github.com/Gearbox-protocol/third-eye/models/pool/pool_v3"
+	"github.com/Gearbox-protocol/third-eye/repository"
+	"github.com/ethereum/go-ethereum/common"
+)
+
+type Repo struct {
+	ds.DummyRepo
+	events []*schemas.PoolLedger
+	dc     *dc_wrapper.DataCompressorWrapper
+}
+
+func (r *Repo) AddPoolLedger(event *schemas.PoolLedger) {
+	r.events = append(r.events, event)
+}
+func (r *Repo) getPoolLedger() []*schemas.PoolLedger {
+	ans := r.events
+	r.events = nil
+	return ans
+}
+
+func (r *Repo) GetDCWrapper() *dc_wrapper.DataCompressorWrapper {
+	return r.dc
+}
+
+func NewRepo(client core.ClientI) Repo {
+	r := Repo{}
+	r.dc = dc_wrapper.NewDataCompressorWrapper(client)
+	r.dc.AddDataCompressorByVersion(core.NewVersion(300), "0xc0101abAFce0BD3de10aa1F3dd827672B150436E", 18798875)
+	return r
+}
+func main() {
+	cfg := config.NewConfig()
+	db := repository.NewDBClient(cfg)
+	client := ethclient.NewEthClient(cfg)
+	r := NewRepo(client)
+
+	var adapters []*ds.SyncAdapter
+	err := db.Raw(`SELECT * from sync_adapters where type='Pool' and version=300`).Find(&adapters).Error
+	log.CheckFatal(err)
+
+	states := []*schemas.PoolState{}
+	err = db.Raw(`SELECT * from pools where  _version=300`).Find(&states).Error
+	log.CheckFatal(err)
+
+	pools := map[string]*pool_v3.Poolv3{}
+	for _, adapter := range adapters {
+		pool := pool_v3.NewPoolFromAdapter(adapter)
+		pools[pool.GetAddress()] = pool
+	}
+
+	for _, state := range states {
+		pool := pools[state.Address]
+		pool.SetUnderlyingState(state)
+	}
+
+	for _, pool := range pools {
+		txLogs, err := pkg.Node{Client: client}.GetLogs(pool.FirstLogAt, pool.LastSync,
+			pool.GetAllAddrsForLogs(),
+			[][]common.Hash{{
+				core.Topic("Transfer(address,address,uint256)"),
+				core.Topic("Deposit(address,address,uint256,uint256)"),
+				core.Topic("Withdraw(address,address,address,uint256,uint256)"),
+			}})
+		log.CheckFatal(err)
+		for _, txLog := range txLogs {
+			pool.OnLog(txLog)
+		}
+		// err = db.Exec(`DELETE from pool_ledgers where pool=? and event in ('AddLiquidity','RemoveLiquidity')`, pool.GetAddress()).Error
+		// log.CheckFatal(err)
+		// err = db.CreateInBatches(r.getPoolLedger(), 500).Error
+		// log.CheckFatal(err)
+		// pool.UpdatePoolv2Ledger(db)
+	}
+}