node.js中對 mysql 進行增刪改查等操作和async，await處理

要對mysql進行操作，我們需要安裝一個mysql的庫。

一、安裝mysql庫

npm install mysql --save

　　

二、對mysql進行簡單查詢操作

const mysql = require('mysql');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

//連接數據庫
conn.connect(function (err) {
    if (err) {
        throw err;
    }
    console.log('連接成功');
});

//查詢數據庫
conn.query('select * from tb_user', function (err, data, field) {
    if (err) {
        throw err;
    }
    //data表示結果集數據，是一個數組
    console.log(data);
    data.forEach(function (value) {
        console.log(value.id, value.user_name, value.addr);
    });
    //表字段的詳細信息
    console.log(field);
});

//關閉數據庫連接
conn.end();

　　

二、對mysql進行增刪改操作

const mysql = require('mysql');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

//連接數據庫
conn.connect(function (err) {
    if (err) {
        throw err;
    }
    console.log('連接成功');
});

//插入數據，query()方法可以對sql語句進行參數綁定，用?號作為占位符。
conn.query('insert into tb_user values(null, ?, ?)', ['xxx', 'xxx'], function (err, data) {
    if (err) {
        throw err;
    }
    if (data && data.affectedRows) {
        console.log('插入數據成功，id為', data.insertId);
    }
});

//修改數據
conn.query('update tb_user set user_name = ? where id = ?', ['ggg', 7], function (err, data) {
    if (err) {
        throw err;
    }
    if (data && data.affectedRows) {
        console.log('修改數據成功');
    }
});

//刪除數據
conn.query('delete from tb_user where id = ?', [5], function (err, data) {
    if (err) {
        throw err;
    }
    if (data && data.affectedRows) {
        console.log('刪除數據成功');
    }
});

//關閉數據庫連接
conn.end();

　　

三、使用mysql連接池來優化對數據庫的操作

頻繁的連接和斷開mysql是比較消耗資源的，我們可以創建一個連接池，復用連接池中的連接，提高效率。

const mysql = require('mysql');

//創建數據庫連接池
let pool = mysql.createPool({
    //連接數量，默認是10
    connectionLimit: 20,
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

//pool.query()方法可以自動的幫我們在連接池中獲取可用連接
pool.query('select * from tb_user', function (err, data) {
    if (err) {
        throw err;
    }
    data.forEach(function (value) {
        console.log(value.id, value.user_name, value.addr);
    });
});

//當然我們也可以手動獲取可用連接
pool.getConnection(function (err, conn) {
    if (err) {
        throw err;
    }
    conn.query('select * from `order`', function (err, data) {
        if (err) {
            throw err;
        }
        data.forEach(function (value) {
            console.log(value.id, value.order_id, value.user_id);
        });

        //連接用完之后，需要釋放，重新放回連接池中。
        //注意這里並沒有銷毀該連接，該連接仍然可用，但需要重新獲取
        conn.release();
    });
});

//從連接池中獲取連接時，將觸發該事件
pool.on('acquire', function (conn) {
    console.log('獲取連接', conn.threadId);
});

//在連接池中建立新連接時，將觸發該事件
pool.on('connection', function (conn) {
    console.log('建立新連接', conn.threadId);
});

//等待可用連接時，將觸發該事件
pool.on('enqueue', function () {
    console.log('等待可用連接');
});

//當連接釋放回池中時，觸發該事件
pool.on('release', function (conn) {
    console.log('連接被釋放回池中', conn.threadId);
});

//結束池中所有的連接，不然node.js的事件循環會一直保持
setTimeout(function () {
    pool.end(function (err) {
        console.log('關閉連接池');
        console.log(err);
    });
}, 3000);

　　

四、按流的方式進行查詢

const mysql = require('mysql');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

let query = conn.query('select * from tb_user');
//Query類繼承自Sequence，而Sequence繼承自EventEmitter
//所以Query類的實例是可以監聽事件

//發生錯誤時
query.on('error', function (err) {
    console.log(err);
});

//獲取查詢字段信息
query.on('fields', function (fields) {
    console.log(fields);
});

//獲取查詢結果
query.on('result', function (result) {
    //暫停獲取結果
    conn.pause();
    //跟流的pause()和resume()很類似，控制獲取數據的頻率。
    setTimeout(function () {
        console.log(result);
        //恢復獲取結果
        conn.resume();
    }, 1000);
});

//查詢結束
query.on('end', function () {
    console.log('查詢結束');
});

conn.end();

通過query.stream()方法返回一個可讀流來獲取數據

const mysql = require('mysql');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

//從一個查詢中獲取一個可讀流
let qs = conn.query('select * from tb_user').stream({highWaterMark: 2});

let result = [];
qs.on('data', function (data) {
    result.push(data);
});

qs.on('end', function () {
    console.log('查詢結束');
    console.log(result);
});

conn.end();

　　

五、mysql的事務處理

const mysql = require('mysql');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

//連接數據庫
conn.connect(function (err) {
    if (err) {
        throw err;
    }
    console.log('連接成功');
});

//開啟一個事務
conn.beginTransaction(function (err) {
    if (err) {
        throw err;
    }
    conn.query('update account set money = money - 50 where name = ?', ['A'], function (err, data) {
        if (err) {
            //如果有錯誤則回滾
            return conn.rollback(function () {
                throw err;
            });
        }
        conn.query('update account set money = money + 50 where name = ?', ['B'], function (err, data) {
            if (err) {
                //如果有錯誤則回滾
                return conn.rollback(function () {
                    throw err;
                });
            }
            //提交事務
            conn.commit(function (err) {
                if (err) {
                    //如果有錯誤則回滾
                    return conn.rollback(function () {
                        throw err;
                    });
                }
                console.log('處理成功');
                conn.end();
            });
        });
    });
});

　　

六、解決mysql嵌套回調的問題

有些時候我們的操作需要上一個操作的結果，這樣會導致比較深的嵌套問題，為了解決可以使用async和await來解決，而async和await又是基於promise的。

const mysql = require('mysql');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

function query(conn, sql, params = []) {
    if (!conn) {
        return;
    }
    return new Promise(function (resolve, reject) {
        conn.query(sql, params, function (err, data) {
            if (err) {
                reject(err);
            } else {
                resolve(data);
            }
        });
    });
}

(async function () {
    let result = await query(conn, 'select * from tb_user');
    console.log(result);
    let row = await query(conn, 'select * from tb_user where id = ?', [result[0].id]);
    console.log(row);
    conn.end();
})();

當然我們還可以使用 util.promiseify() 進行包裝。

const mysql = require('mysql');
const util = require('util');

//創建數據庫連接
let conn = mysql.createConnection({
    //主機地址
    host: '127.0.0.1',
    //用戶名
    user: 'root',
    //密碼
    password: '123456',
    //數據庫
    database: 'test',
    //端口
    port: 3306,
    //字符集
    charset: 'utf8'
});

//注意通過util.promisify進行包裝的函數，必須滿足
//1、函數的最后一個參數是回調函數
//2、回調函數的參數為(err, result)，前者是錯誤，后者是正常結果
//注意這里不要重新創建一個變量，不然會報錯。
conn.query = util.promisify(conn.query);

(async function () {
    let result = await conn.query('select * from tb_user');
    console.log(result);
    let row = await conn.query('select * from tb_user where id = ?', [result[0].id]);
    console.log(row);
    conn.end();
})();