diff --git a/src/db.ts b/src/db.ts index 5d9deea..4e77a9d 100644 --- a/src/db.ts +++ b/src/db.ts @@ -3,6 +3,7 @@ import { DataSource } from "typeorm"; import { User } from "./entities/User"; import { Review } from "./entities/Review"; import { Saved } from "./entities/Saved"; +import { Hazard } from "./entities/Hazard"; let db: DataSource | null = null; @@ -12,7 +13,7 @@ export function getDb(forceSync = false): DataSource { type: "postgres", url: process.env.DATABASE_URL, synchronize: process.argv.includes("sync") || forceSync, - entities: [User, Review, Saved], + entities: [User, Review, Saved, Hazard], }); } return db;