Update eden subgraph build and watch subgraph contracts (#54)

* Update eden subgraph build

* Watch subgraph contracts in the job-runner

* Change eden subgraph startBlocks to original values

packages/eden-watcher/src/indexer.ts

@@ -261,8 +261,10 @@ export class Indexer implements IndexerInterface {
     const checkpoint = await this.getLatestIPLDBlock(contractAddress, 'checkpoint');
 
     // There should be an initial checkpoint at least.
-    // Assumption: There should be no events for the contract at the starting block.
-    assert(checkpoint, 'Initial checkpoint doesn\'t exist');
+    // Return if initial checkpoint doesn't exist.
+    if (!checkpoint) {
+      return;
+    }
 
     // Check if the latest checkpoint is in the same block.
     assert(checkpoint.block.blockHash !== block.blockHash, 'Checkpoint already created for the block hash.');
@@ -289,7 +291,7 @@ export class Indexer implements IndexerInterface {
         const checkpointBlock = await this.getLatestIPLDBlock(contract.address, 'checkpoint');
 
         if (!checkpointBlock) {
-          if (blockNumber === contract.startingBlock) {
+          if (blockNumber >= contract.startingBlock) {
             // Call initial checkpoint hook.
             await createInitialCheckpoint(this, contract.address, blockHash);
           }

packages/eden-watcher/src/job-runner.ts

@@ -146,6 +146,9 @@ export const main = async (): Promise<any> => {
   graphWatcher.setIndexer(indexer);
   await graphWatcher.init();
 
+  // Watching all the contracts in the subgraph.
+  await graphWatcher.addContracts();
+
   const jobQueueConfig = config.jobQueue;
   assert(jobQueueConfig, 'Missing job queue config');
 

packages/graph-node/src/database.ts

@@ -42,27 +42,35 @@ export class Database {
 
   async getEntity<Entity> (entity: (new () => Entity) | string, id: string, blockHash: string): Promise<Entity | undefined> {
     const queryRunner = this._conn.createQueryRunner();
-    const repo = queryRunner.manager.getRepository(entity);
-    const whereOptions: { [key: string]: any } = { id };
 
-    if (blockHash) {
-      whereOptions.blockHash = blockHash;
-    }
+    try {
+      const repo = queryRunner.manager.getRepository(entity);
 
-    const findOptions = {
-      where: whereOptions,
-      order: {
-        blockNumber: 'DESC'
+      const whereOptions: { [key: string]: any } = { id };
+
+      if (blockHash) {
+        whereOptions.blockHash = blockHash;
       }
-    };
 
-    let entityData = await repo.findOne(findOptions as FindOneOptions<any>);
+      const findOptions = {
+        where: whereOptions,
+        order: {
+          blockNumber: 'DESC'
+        }
+      };
 
-    if (!entityData && findOptions.where.blockHash) {
-      entityData = await this._baseDatabase.getPrevEntityVersion(queryRunner, repo, findOptions);
+      let entityData = await repo.findOne(findOptions as FindOneOptions<any>);
+
+      if (!entityData && findOptions.where.blockHash) {
+        entityData = await this._baseDatabase.getPrevEntityVersion(queryRunner, repo, findOptions);
+      }
+
+      return entityData;
+    } catch (error) {
+      console.log(error);
+    } finally {
+      await queryRunner.release();
     }
-
-    return entityData;
   }
 
   async saveEntity (entity: string, data: any): Promise<void> {
@@ -129,7 +137,7 @@ export class Database {
 
       // Get blockNumber as _blockNumber and blockHash as _blockHash from the entityInstance (wasm).
       if (['_blockNumber', '_blockHash'].includes(propertyName)) {
-        return fromEntityValue(instanceExports, entityInstance, propertyName);
+        return fromEntityValue(instanceExports, entityInstance, propertyName.slice(1));
       }
 
       return fromEntityValue(instanceExports, entityInstance, propertyName);

packages/graph-node/src/utils.ts

@@ -303,6 +303,9 @@ export const createBlock = async (instanceExports: any, blockData: Block): Promi
 
   const authorPtr = await Address.zero();
 
+  const sizePtr = await __newString('0');
+  const size = await BigInt.fromString(sizePtr);
+
   // Missing fields from watcher in block data:
   // author
   // size
@@ -320,7 +323,7 @@ export const createBlock = async (instanceExports: any, blockData: Block): Promi
     blockTimestamp,
     difficulty,
     totalDifficulty,
-    null
+    size
   );
 };
 

packages/graph-node/src/watcher.ts

@@ -95,6 +95,16 @@ export class GraphWatcher {
     }, {});
   }
 
+  async addContracts () {
+    assert(this._indexer?.watchContract);
+
+    // Watching the contract(s).
+    for (const dataSource of this._dataSources) {
+      const { source: { address, startBlock }, name } = dataSource;
+      await this._indexer.watchContract(address, name, true, startBlock);
+    }
+  }
+
   async handleEvent (eventData: any) {
     const { contract, event, eventSignature, block, tx, eventIndex } = eventData;
 
@@ -147,6 +157,8 @@ export class GraphWatcher {
   async handleBlock (blockHash: string) {
     const blockData = await getFullBlock(this._postgraphileClient, blockHash);
 
+    this._context.event.block = blockData;
+
     // Call block handler(s) for each contract.
     for (const dataSource of this._dataSources) {
       // Check if block handler(s) are configured.

packages/util/src/types.ts

@@ -73,6 +73,7 @@ export interface IndexerInterface {
   isWatchedContract?: (address: string) => Promise<ContractInterface | undefined>;
   cacheContract?: (contract: ContractInterface) => void;
   createDiffStaged?: (contractAddress: string, blockHash: string, data: any) => Promise<void>
+  watchContract?: (address: string, kind: string, checkpoint: boolean, startingBlock?: number) => Promise<boolean>
 }
 
 export interface EventWatcherInterface {