Create knex config file

Author: mzyy94

server/index.js

@@ -3,24 +3,17 @@
 // require modules
 const Koa = require("koa")
 const Router = require("koa-router")
+const Knex = require("knex")
 const rp = require("request-promise")
 const crypto = require("crypto")
 const uuid = require("uuid")
-const knex = require("knex")({
-    client: "sqlite3",
-    connection: {
-        filename: "./development.db",
-        timezone: "UTC"
-    },
-    acquireConnectionTimeout: 1000,
-    useNullAsDefault: true
-})
 
 // define constant values
 const app = new Koa()
 const router = new Router({prefix: "/api/v1"})
 const secretKey = crypto.randomBytes(32).hexSlice()
-const {client_id, client_secret} = process.env
+const {client_id, client_secret, NODE_ENV} = process.env
+const knex = Knex(require("./knexfile.js")[NODE_ENV || "development"])
 const endpoint = "https://qiita.com/api/v2"
 
 

server/init_db.js

@@ -1,13 +1,5 @@
-const uuid = require("uuid")
-const knex = require("knex")({
-    client: "sqlite3",
-    connection: {
-        filename: "./development.db",
-        timezone: "UTC"
-    },
-    acquireConnectionTimeout: 1000,
-    useNullAsDefault: true
-})
+const Knex = require("knex")
+const knex = Knex(require("./knexfile.js")[process.env.NODE_ENV || "development"])
 
 // Create table
 knex.schema.createTableIfNotExists("users", (table) => {
@@ -27,11 +19,3 @@ knex.schema.createTableIfNotExists("item_dislike", (table) => {
     table.primary("by_whom", "id")
 })
 .then(() => console.log("Item_dislike table initialize done."))
-
-// Create test account
-const token = uuid.v1()
-const suffix = new Date().getMilliseconds()
-knex("users").insert({id: `testuser${suffix}`, token: token, revoked: false })
-.then(() => console.log(`New user testuser${suffix} created.
-Request with "Authorization: Bearer ${token}" http header.
-`))

server/knexfile.js

@@ -0,0 +1,45 @@
+module.exports = {
+
+  development: {
+    client: "sqlite3",
+    connection: {
+      filename: "./development.db",
+      timezone: "UTC"
+    },
+    acquireConnectionTimeout: 1000,
+    useNullAsDefault: true
+  },
+
+  staging: {
+    client: 'postgresql',
+    connection: {
+      database: 'my_db',
+      user:     'username',
+      password: 'password'
+    },
+    pool: {
+      min: 2,
+      max: 10
+    },
+    migrations: {
+      tableName: 'knex_migrations'
+    }
+  },
+
+  production: {
+    client: 'postgresql',
+    connection: {
+      database: 'my_db',
+      user:     'username',
+      password: 'password'
+    },
+    pool: {
+      min: 2,
+      max: 10
+    },
+    migrations: {
+      tableName: 'knex_migrations'
+    }
+  }
+
+};