Austin/firecrawl
1
0
Fork 0
mirror of https://github.com/mendableai/firecrawl.git synced 2024-11-16 03:32:22 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

add read replica

Browse Source
This commit is contained in:
Gergő Móricz 2024-08-31 05:02:43 +02:00
parent 11232bf4f8
commit afa48e8a6b
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/api/src/services/db/index.ts
View File

@ -1,13 +1,22 @@
import { drizzle, NodePgDatabase } from 'drizzle-orm/node-postgres';
import { withReplicas } from 'drizzle-orm/pg-core';
import { Pool } from "pg";
function createDB() {
if (process.env.DATABASE_URI) {
const pool = new Pool({
const db = drizzle(new Pool({
connectionString: process.env.DATABASE_URI,
})
}))
return drizzle(pool);
if (process.env.DATABASE_REPLICA_URI) {
const replica = drizzle(new Pool({
connectionString: process.env.DATABASE_REPLICA_URI,
}));
return withReplicas(db, [replica])
} else {
return db;
}
} else {
return { no: true };
}