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/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.