fix: temporary fix for testing

Author: qwe2tz

src/db/database.ts

@@ -1,25 +1,32 @@
 import { DataSource } from "typeorm";
-import { CollectedReward } from "@/model/staking/CollectedRewards";
-import { EstimatedReward } from "@/model/staking/EstimatedRewards";
-import { NodeProofRate } from "@/model/staking/NodeProofRate";
+import { CollectedReward } from "../model/staking/CollectedRewards";
+import { NodeProofRate } from "../model/staking/NodeProofRate";
+import { EstimatedReward } from "../model/staking/EstimatedRewards";
+
 
 let AppDataSource: DataSource;
 
-export function initDatabase() {
+export async function initDatabase() {
+  const schema = process.env.DB_SCHEMA || "computed";
+
   AppDataSource = new DataSource({
     type: "postgres",
-    host: process.env.PG_HOST,
+    host: "postgres",
     port: parseInt(process.env.DB_PORT) || 5432,
     username: process.env.DB_USER,
-    password: process.env.DB_PASSWORD,
+    password: process.env.DB_PASS,
     database: process.env.DB_NAME,
     entities: [CollectedReward, EstimatedReward, NodeProofRate],
     // This means we always update the database schema to match the entities
     synchronize: true,
-    schema: process.env.DB_SCHEMA || "computed",
+    schema,
   });
 
-  AppDataSource.initialize().catch((error) => console.log(error));
-  console.log("[✔] Type ORM Data Source initialized. Using schema:", process.env.DB_SCHEMA);
+  await AppDataSource.initialize()
+    .then(() => {
+      console.log("[✔] Type ORM Data Source initialized. Using schema:", schema);
+    })
+    .catch((error) => console.log(error));
+  
   return AppDataSource;
 }

src/db/index.ts

@@ -2,7 +2,6 @@ import { initDatabase } from "./database";
 import { processStakingData } from "./procedures/staking";
 
 export async function processData(blocks: Number[]) {
-  const db = initDatabase();
-
+  const db = await initDatabase();
   await processStakingData(db, blocks);
 }

src/db/procedures/staking.ts

@@ -1,6 +1,6 @@
 import { DataSource } from "typeorm";
-import { CollectedReward } from "@/model/staking/CollectedRewards";
-import { NodeProofRate } from "@/model/staking/NodeProofRate";
+import { CollectedReward } from "../../model/staking/CollectedRewards";
+import { NodeProofRate } from "../../model/staking/NodeProofRate";
 // import { EstimatedReward } from "@/model/staking/EstimatedRewards";
 
 async function _processCollectedReward(AppDataSource: DataSource, blocks: Number[]) {

tsconfig.json

@@ -11,7 +11,7 @@
     "baseUrl": ".",
     "paths": {
       "@/*": ["src/*"],
-      "@model/*": ["src/model/generated/*"]
+      "@model/*": ["src/model/*"]
     }
   },
   "ts-node": {