Jelajahi Sumber

refactor: 封装数据库初始化函数

modood 7 tahun lalu
induk
melakukan
501d67085b
1 mengubah file dengan 11 tambahan dan 4 penghapusan
  1. 11 4
      lib/sqlite.js

+ 11 - 4
lib/sqlite.js

@@ -8,6 +8,7 @@ const sequelizeCursorPagination = require('sequelize-cursor-pagination')
 const sequelize = new Sequelize('gopkg', undefined, undefined, {
   dialect: 'sqlite',
   storage: path.resolve(__dirname, '../dist/data.sqlite'),
+  logging: false,
   define: {
     timestamps: false,
     freezeTableName: true
@@ -52,8 +53,14 @@ Village.belongsTo(Street)
 
 /* Connect database */
 
-sequelize.sync()
-  .then(() => console.log('db connected!'))
-  .error(err => console.log(err))
+async function init () {
+  try {
+    await sequelize.sync()
+    console.log('db connected!')
+  } catch (err) {
+    console.log(err)
+    process.exit(-1)
+  }
+}
 
-module.exports = { Province, City, Area, Street, Village }
+module.exports = { init, Province, City, Area, Street, Village }