diff --git a/packages/server/src/ChildProcess.ts b/packages/server/src/ChildProcess.ts index c14a4977..a93906c0 100644 --- a/packages/server/src/ChildProcess.ts +++ b/packages/server/src/ChildProcess.ts @@ -142,19 +142,19 @@ export class ChildProcess { * @returns {DataSource} */ async function initDB() { - let childAppDataSource; - let homePath; + 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'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break case 'mysql': childAppDataSource = new DataSource({ type: 'mysql', @@ -164,11 +164,11 @@ async function initDB() { password: process.env.DATABASE_PASSWORD, database: process.env.DATABASE_NAME, charset: 'utf8mb4', - synchronize: (process.env.OVERRIDE_DATABASE == 'true'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break case 'postgres': childAppDataSource = new DataSource({ type: 'postgres', @@ -177,21 +177,21 @@ async function initDB() { username: process.env.DATABASE_USER, password: process.env.DATABASE_PASSWORD, database: process.env.DATABASE_NAME, - synchronize: (process.env.OVERRIDE_DATABASE == 'true'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + 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'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break } return await childAppDataSource.initialize() diff --git a/packages/server/src/DataSource.ts b/packages/server/src/DataSource.ts index b2ae5cd9..958074cc 100644 --- a/packages/server/src/DataSource.ts +++ b/packages/server/src/DataSource.ts @@ -9,18 +9,18 @@ import { getUserHome } from './utils' let appDataSource: DataSource export const init = async (): Promise => { - let homePath; + let homePath switch (process.env.DATABASE_TYPE) { case 'sqlite': homePath = process.env.DATABASE_PATH ?? path.join(getUserHome(), '.flowise') appDataSource = new DataSource({ type: 'sqlite', database: path.resolve(homePath, 'database.sqlite'), - synchronize: (process.env.OVERRIDE_DATABASE == 'true'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break case 'mysql': appDataSource = new DataSource({ type: 'mysql', @@ -30,11 +30,11 @@ export const init = async (): Promise => { password: process.env.DATABASE_PASSWORD, database: process.env.DATABASE_NAME, charset: 'utf8mb4', - synchronize: (process.env.OVERRIDE_DATABASE == 'true'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break case 'postgres': appDataSource = new DataSource({ type: 'postgres', @@ -43,21 +43,21 @@ export const init = async (): Promise => { username: process.env.DATABASE_USER, password: process.env.DATABASE_PASSWORD, database: process.env.DATABASE_NAME, - synchronize: (process.env.OVERRIDE_DATABASE == 'true'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break default: homePath = process.env.DATABASE_PATH ?? path.join(getUserHome(), '.flowise') appDataSource = new DataSource({ type: 'sqlite', database: path.resolve(homePath, 'database.sqlite'), - synchronize: (process.env.OVERRIDE_DATABASE == 'true'), + synchronize: process.env.OVERRIDE_DATABASE == 'true', entities: [ChatFlow, ChatMessage, Tool], migrations: [] }) - break; + break } }