farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 13:00:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

Multiple Database Support

Browse Source
This commit is contained in:
Atish Amte
2023-07-18 23:52:16 +05:30
parent 10563bd6b8
commit c5eff32cf0
8 changed files with 128 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/server/src/ChildProcess.ts
+52 -8
View File
@@ -142,14 +142,58 @@ export class ChildProcess {
* @returns {DataSource}
*/
async function initDB() {
const homePath = process.env.DATABASE_PATH ?? path.join(getUserHome(), '.flowise')
const childAppDataSource = new DataSource({
type: 'sqlite',
database: path.resolve(homePath, 'database.sqlite'),
synchronize: true,
entities: [ChatFlow, ChatMessage, Tool],
migrations: []
})
let childAppDataSource;
let homePath;
switch (process.env.DATABASE_TYPE) {
case 'sqlite':
homePath = process.env.DATABASE_PATH ?? path.join(getUserHome(), '.flowise')
childAppDataSource = new DataSource({
type: 'sqlite',
database: path.resolve(homePath, 'database.sqlite'),
synchronize: (process.env.OVERRIDE_DATABASE == 'true'),
entities: [ChatFlow, ChatMessage, Tool],
migrations: []
})
break;
case 'mysql':
childAppDataSource = new DataSource({
type: 'mysql',
host: process.env.DATABASE_HOST,
port: parseInt(process.env.DATABASE_PORT || '3306'),
username: process.env.DATABASE_USER,
password: process.env.DATABASE_PASSWORD,
database: process.env.DATABASE_NAME,
charset: 'utf8mb4',
synchronize: (process.env.OVERRIDE_DATABASE == 'true'),
entities: [ChatFlow, ChatMessage, Tool],
migrations: []
})
break;
case 'postgres':
childAppDataSource = new DataSource({
type: 'postgres',
host: process.env.DATABASE_HOST,
port: parseInt(process.env.DATABASE_PORT || '5432'),
username: process.env.DATABASE_USER,
password: process.env.DATABASE_PASSWORD,
database: process.env.DATABASE_NAME,
synchronize: (process.env.OVERRIDE_DATABASE == 'true'),
entities: [ChatFlow, ChatMessage, Tool],
migrations: []
})
break;
default:
homePath = process.env.DATABASE_PATH ?? path.join(getUserHome(), '.flowise')
childAppDataSource = new DataSource({
type: 'sqlite',
database: path.resolve(homePath, 'database.sqlite'),
synchronize: (process.env.OVERRIDE_DATABASE == 'true'),
entities: [ChatFlow, ChatMessage, Tool],
migrations: []
})
break;
}
return await childAppDataSource.initialize()
}