ionian/src/database.ts

import { existsSync } from 'fs';
import Knex from 'knex';
import {Knex as KnexConfig} from './config';

export async function connectDB(){
    const env = process.env.NODE_ENV || 'development';
    const config = KnexConfig.config;
    if(env != "production" && env != "development"){
        throw new Error("process unknown value in NODE_ENV: must be either \"development\" or \"production\"");
    }
    const init_need = existsSync(config[env].connection.filename);
    const knex = Knex(config[env]);
    let tries = 0;
    for(;;){
        try{
            console.log("try to connect db");
            await knex.raw('select 1 + 1;');
            console.log("connect success");
        }
        catch(err){
            if(tries < 3){
                tries++;
                console.error(`connection fail ${err} retry...`);
                continue;
            }
            else{
                throw err;
            }
        }
        break;
    }
    if(init_need){
        const migrate = await import("../migrations/initial");
        await migrate.up(knex);
    }
    return knex;
}