address comments

dydxprotocol · Jul 18, 2024 · 76333e7 · 76333e7
1 parent 8f76dea
commit 76333e7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/...postgres/src/db/migrations/migration_files/20240717160024_create_leaderboard_pnl_table.ts b/...postgres/src/db/migrations/migration_files/20240717160024_create_leaderboard_pnl_table.ts
@@ -2,7 +2,7 @@ import * as Knex from 'knex';
 
 export async function up(knex: Knex): Promise<void> {
   return knex.schema.createTable('leaderboard_pnl', (table) => {
-    table.string('address').notNullable();
+    table.string('address').notNullable().references('address').inTable('wallets');
     table.enum(
       'timeSpan',
       [

diff --git a/indexer/packages/postgres/src/models/leaderboard-pnl-model.ts b/indexer/packages/postgres/src/models/leaderboard-pnl-model.ts
@@ -1,3 +1,7 @@
+import path from 'path';
+
+import { Model } from 'objection';
+
 import { NumericPattern } from '../lib/validators';
 import UpsertQueryBuilder from '../query-builders/upsert';
 import BaseModel from './base-model';
@@ -12,7 +16,16 @@ export default class LeaderboardPnlModel extends BaseModel {
     return ['address', 'timeSpan'];
   }
 
-  static relationMappings = {};
+  static relationMappings = {
+    wallets: {
+      relation: Model.BelongsToOneRelation,
+      modelClass: path.join(__dirname, 'wallet-model'),
+      join: {
+        from: 'leaderboard_pnl.address',
+        to: 'wallets.address',
+      },
+    },
+  };
 
   static get jsonSchema() {
     return {