Shard Base Subgraph

src/mappings/modifyLiquidity.ts

@@ -25,9 +25,15 @@ export function handleModifyLiquidityHelper(
   subgraphConfig: SubgraphConfig = getSubgraphConfig(),
 ): void {
   const poolManagerAddress = subgraphConfig.poolManagerAddress
+  const shard = subgraphConfig.shardNumber
+  const numShards = subgraphConfig.numShards
+  // Consistently hash using murmuhash
+  if (!isPoolInShard(event.params.id.toHexString(), numShards, shard)) {
+    return
+  }
 
   const bundle = Bundle.load('1')!
-  const poolId = event.params.id.toHexString()
+  const poolId = event.params.id.toHexString() + '#' + shard.toString()
   const pool = Pool.load(poolId)
   const poolManager = PoolManager.load(poolManagerAddress)
 

src/mappings/poolManager.ts

@@ -5,6 +5,7 @@ import { PoolManager } from '../types/schema'
 import { Bundle, Pool, Token } from '../types/schema'
 import { getSubgraphConfig, SubgraphConfig } from '../utils/chains'
 import { ADDRESS_ZERO, ONE_BI, ZERO_BD, ZERO_BI } from '../utils/constants'
+import { isPoolInShard } from '../utils/hash'
 import { updatePoolDayData, updatePoolHourData } from '../utils/intervalUpdates'
 import { findNativePerToken, getNativePriceInUSD, sqrtPriceX96ToTokenPrices } from '../utils/pricing'
 import { fetchTokenDecimals, fetchTokenName, fetchTokenSymbol, fetchTokenTotalSupply } from '../utils/token'
@@ -19,6 +20,11 @@ export function handleInitializeHelper(
   event: InitializeEvent,
   subgraphConfig: SubgraphConfig = getSubgraphConfig(),
 ): void {
+  const shard = subgraphConfig.shardNumber
+  const numShards = subgraphConfig.numShards
+  if (shard && numShards && !isPoolInShard(event.params.id.toHexString(), numShards, shard)) {
+    return
+  }
   const poolManagerAddress = subgraphConfig.poolManagerAddress
   const whitelistTokens = subgraphConfig.whitelistTokens
   const tokenOverrides = subgraphConfig.tokenOverrides

src/mappings/swap.ts

@@ -4,6 +4,7 @@ import { Swap as SwapEvent } from '../types/PoolManager/PoolManager'
 import { Bundle, Pool, PoolManager, Swap, Token } from '../types/schema'
 import { getSubgraphConfig, SubgraphConfig } from '../utils/chains'
 import { ONE_BI, ZERO_BD } from '../utils/constants'
+import { isPoolInShard } from '../utils/hash'
 import { convertTokenToDecimal, loadTransaction, safeDiv } from '../utils/index'
 import {
   updatePoolDayData,
@@ -24,6 +25,11 @@ export function handleSwap(event: SwapEvent): void {
 }
 
 export function handleSwapHelper(event: SwapEvent, subgraphConfig: SubgraphConfig = getSubgraphConfig()): void {
+  const shard = subgraphConfig.shardNumber
+  const numShards = subgraphConfig.numShards
+  if (shard && numShards && !isPoolInShard(event.params.id.toHexString(), numShards, shard)) {
+    return
+  }
   const poolManagerAddress = subgraphConfig.poolManagerAddress
   const stablecoinWrappedNativePoolId = subgraphConfig.stablecoinWrappedNativePoolId
   const stablecoinIsToken0 = subgraphConfig.stablecoinIsToken0

src/utils/chains.ts

@@ -66,11 +66,15 @@ export class SubgraphConfig {
 
   // native token details for the chain.
   nativeTokenDetails: NativeTokenDetails
+  numShards?: number
+  shardNumber?: number
 }
 
 export function getSubgraphConfig(): SubgraphConfig {
   // Update this value to the corresponding chain you want to deploy
   const selectedNetwork = dataSource.network()
+  const shardNumber = dataSource?.context()?.getI32('shardNumber')
+  const numShards = dataSource?.context()?.getI32('numShards')
 
   if (selectedNetwork == SEPOLIA_NETWORK_NAME) {
     return {
@@ -170,6 +174,8 @@ export function getSubgraphConfig(): SubgraphConfig {
         name: 'Ethereum',
         decimals: BigInt.fromI32(18),
       },
+      numShards,
+      shardNumber,
     }
   } else if (selectedNetwork == ARBITRUM_ONE_NETWORK_NAME) {
     return {
@@ -238,6 +244,8 @@ export function getSubgraphConfig(): SubgraphConfig {
         name: 'Ethereum',
         decimals: BigInt.fromI32(18),
       },
+      numShards: 5,
+      shardNumber: 0,
     }
   } else if (selectedNetwork == MATIC_NETWORK_NAME) {
     return {

src/utils/hash.ts

@@ -0,0 +1,5 @@
+export function isPoolInShard(poolId: string, numShards: number, shardNumber: number): boolean {
+  const lastByteHex = poolId.slice(-2)
+  const lastByte = parseInt(lastByteHex, 16)
+  return lastByte % numShards == shardNumber
+}