study-univ3-on-hardhat/scripts/lp/lpMaintenance.ts

import { web3 } from "hardhat";
import History from "../interface/history";
import contracts from "../../config/contracts";
import logger from "../../utils/logger";
import {BigNumber} from "ethers";
import {replaceAll} from "hardhat/internal/util/strings";
import history from "../interface/history";
const ierc20abi = require('../../abi/IERC20_ABI.json')

// 初始化410 v2工具箱
const v2ToolBy410Abi = require('../../abi/410_V2_TOOLS.json')
const v2ToolBy410 = new web3.eth.Contract(v2ToolBy410Abi, contracts.V2_TOOLS_BY_410)

export class LpMaintenance {
  tokenInstance: any = {}
  maxMemoryOfByte: number = 0

  async handleToken(pool: any, zero: boolean) {
    const token = {
      'address': zero ? pool.token0 : pool.token1,
      'symbol': zero ? pool.symbol0 : pool.symbol1,
      'decimals': zero ? pool.decimals0 : pool.decimals1,
      'name': 'xxxxxxxx'
    }

    let tokenObj = this.tokenInstance[token.address]
    if (!tokenObj) {
      tokenObj = new web3.eth.Contract(ierc20abi, token.address)
      this.tokenInstance[token.address] = tokenObj
    }

    token.name = await tokenObj.methods.name().call()
    token.name = token.name.replace(/[^A-Za-z0-9 ]+/g, '').substring(0, 37)

    return token
  }

  async saveToken(lp: any, type='token') {
    try {
      const token = await this.handleToken(lp, true)

      await history.appendOrUpdate(type, token.address, token)
    } catch (e) {}

    try {
      const token = await this.handleToken(lp, false)

      await history.appendOrUpdate(type, token.address, token)
    } catch (e) {}
  }

  async saveLp(lp: any, type='0') {
    await History.appendOrUpdate(type, lp.LP, lp)
  }

  async checkLpType(lp: any) {
    const ethTokenList = [
      '0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2', // WETH
      '0xaf3ccfd9b59b36628cc2f659a09d6440795b2520', // ETH
      '0x7bf88d2c0e32de92cdaf2d43ccdc23e8edfd5990'  // WETHW
    ]

    const filterTypeList = [
      { type: 'topLp', limit: 10 * 1e18 },
      { type: 'normalLp', limit: 1 * 1e18 },
      { type: 'lp', limit: 0 },
    ]

    if (ethTokenList.indexOf(lp.token0.toLowerCase()) == -1
      && ethTokenList.indexOf(lp.token1.toLowerCase()) == -1) {
      return 'normalLp'
    }

    for (const filterType of filterTypeList) {
      if (ethTokenList.indexOf(lp.token0.toLowerCase()) != -1) {
        if (parseInt(lp.r0) >= filterType.limit) return filterType.type
      } else if (ethTokenList.indexOf(lp.token1.toLowerCase()) != -1) {
        if (parseInt(lp.r1) >= filterType.limit) return filterType.type
      }
    }
  }

  async handleLp(lp: any, oldType: any) {
    // 过滤Lp
    const lpType = await this.checkLpType(lp)
    if (lpType != oldType) {
      // 保存变更之后的的Token
      // await this.saveToken(lp, lpType)
      // 保存变更后的Lp
      await this.saveLp(lp, lpType)
      logger.debug(`lp:${lp.LP},${oldType}->${lpType}.`)
    }
  }

  showMemory (mainInfo: string) {
    const memoryUsage = process.memoryUsage()
    if (this.maxMemoryOfByte < memoryUsage.rss) {
      this.maxMemoryOfByte = memoryUsage.rss
    }
    logger.debug(`${mainInfo}  ${this.format(memoryUsage.rss)}/${this.format(this.maxMemoryOfByte)}`)
  }

  format (bytes: any) {
    return (bytes / 1024 / 1024).toFixed(2) + ' MB';
  }

  generateAddressListByLpList (lpList: any) {
    const lpAddressList: any = []

    for (const lp of lpList) {
      lpAddressList.push(lp.dataObj.LP)
    }

    return lpAddressList
  }

  updateLocalLpListR0R1(lpList: any, r0s: any, r1s: any) {
    for (let index = 0; index < lpList.length; index++) {
      lpList[index].dataObj.r0 = r0s[index]
      lpList[index].dataObj.r1 = r1s[index]
    }
  }

  async run() {
    logger.debug('LP maintenance start.')
    while (true) {
      this.showMemory('pull db lp data...')
      const lpPullRst = await history.findByHashOrBlockOrDataVague('topLp', '', '', 0, 20000)
      if (!lpPullRst.state) continue
      const topLpList = lpPullRst.data
      logger.debug(`lp length:${topLpList.length}, generate address list...`)
      const lpList: any = [].concat(topLpList)
      const lpAddressList: any = this.generateAddressListByLpList(lpList)

      // 集中拉取r0，r1并更新本地的
      const size = 2000
      logger.debug(`update lp r0,r1 by 410 tools...`)
      for (let from = 0; from < topLpList.length; from += size) {
        logger.debug(`get lp info from ${from} to ${from + size}...`)
        const lpR0R1Info: any = await v2ToolBy410.methods.getPairSBalance(lpAddressList.slice(from, from + size)).call()
        const r0s = lpR0R1Info.amounts0
        const r1s = lpR0R1Info.amounts1

        this.updateLocalLpListR0R1(lpList.slice(from, from + size), r0s, r1s)
      }

      // 将lp类型有变动的全部更新
      for (const lpDbObj of lpList) {
        await this.handleLp(lpDbObj.dataObj, lpDbObj.block)
      }
    }
  }
}

async function main() {
  await new LpMaintenance().run()
}

main().catch((error) => {
  console.error(error);
  process.exitCode = 1;
})