koa+mysql+vue+socket.io全栈开发之数据访问篇

后端搭起大体的框架后，接着涉及到的就是如何将数据持久化的问题，也就是对数据库进行 CURD 操作。

关于数据库方案, mongodb 和 mysql 都使用过，但我选用的是 mysql，原因：

目前为止 mysql 与 mongodb 性能相差不大，尤其是 mysql 8.0 版本，速度非常快，查询数据是 mysql 更快，写数据方面 mongodb 则更胜一筹；
mysql 建立关联数据要更方便些，比如：一对多，多对多的关系；
mysql 作为关系型数据库，数据一致性方面更好，尤其是事务用起来更顺手；
本人对 sql 操作比较得心应手，毕竟大部分项目用得都是 mysql，而 mongodb 在正式些的项目上用的就少了，而且目前关系型数据库也在进化, postgrep 和 mysql 都已经支持 json了。

node-mysql

node-mysql 是用 sql 语句对 mysql 进行操作的库, 并没有使用 Sequelize 这种 orm。因为我对 sql 熟悉，原生开发效率高。

连接池

连接数据库我选用连接池的方式，这种方式能高效的利用数据库连接

//dbPool.js

const mysql = require('mysql');

const dbconfig = require('../config/db');

const log = require('../common/logger');

let pool = null;

/**

 * get the connection pool of database

 * 获取数据库连接池

 */

exports.getPool = function () {

    if (!pool) {

        log.info("creating pool");

        pool = mysql.createPool(dbconfig);

    }

    return pool;

}

数据库配置文件

emoji 格式要用 utf8mb4 格式存储，所以这里连接字符集选用 utf8mb4，当然客户端和数据结果集一样也要设置为 utf8mb4。

module.exports={

	host: "localhost",

	port: "3306",

	user: "root",

	password: "jeff",

	database: "chatdb",

	charset : 'utf8mb4',//utf8mb4才能保存emoji

	multipleStatements: true,// 可同时查询多条语句, 但不能参数化传值

	connectionLimit: 100 //连接数量

};

Dao的编写

基本的代码编写方式如下，每个方法基本都是这么一种流程，获取数据库连接，执行 sql 语句，返回结果，处理异常。

exports.queryInfo = function (params, callback){

    pool.query('select ...', params, function (error, result, fields) {

        if (error) {

          log(error);

          callback(false);

        }

        else callback(result)

    });

}

exportDao

这造成了一大堆重复的样板代码，我们需要封装它，用 JavaScript 高阶函数特性很容易就能实现，同时加上 Promise，调用时就能方便地用 async await 了，还有日志记录功能也加上。

const pool = require("./dbPool").getPool();

const log = require('../common/logger');

/**

 * export named query function

 */

const exportDao = opts => Object.keys(opts).reduce((next, key) => {

    next[key] = (...args) => new Promise((resolve, reject) => {

        if (opts[key]) args.unshift(opts[key]);

        log.info('====== execute sql ======')

        log.info(args);

        pool.query(...args, (err, result, fields) => {// fields is useless

            if (err) reject(err)

            else resolve(result);

        });

    });

    return next;

}, {});

userDao文件为例，使用 exportDao 直接就能把里面的 key-value 对象输出为以key 为方法名的dao方法，挂载到 module.exports 下。

const { exportDao } = require('./common');

//直接就exports里面的key值对应的方法

module.exports = exportDao({

	sql: null,// 有些时候需要直接写sql

	count: 'select count(*) as count from user where ?',

	getUser: 'select * from user where ?',

	insert: 'insert into user set ?',

	update: 'update user set ? where id = ?',

	delete: 'delete from user where ?'

});

/* 最终输出格式

module.exports = {

  sql:() => {},

  count:() => {},

  ...

}*/

transaction

还有事务 transaction 的功能需要用到，来看一下 node-mysql 官方的例子，层层回调