Node.js 連接 postgreSQL數據庫

首先使用npm安裝數據庫連接模塊：

npm install --save pg

連接池創建

然后代碼中引入 pg 模塊，并編寫數據庫配置：

var pg = require('pg');

// 數據庫配置
var config = {  
    user:"postgres",
    database:"ghost",
    password:"123456",
    port:5432,

    // 擴展屬性
    max:20, // 連接池最大連接數
    idleTimeoutMillis:3000, // 連接最大空閑時間 3s
}

pg 模塊中有兩種數據庫連接方式，先講連接池模式，下面是創建連接池：

// 創建連接池
var pool = new pg.Pool(config);

傳入配置后就創建好了連接池。

查詢數據

查詢首先創建好連接，然后調用 api 進行查詢:

// 查詢
pool.connect(function(err, client, done) {  
  if(err) {
    return console.error('數據庫連接出錯', err);
  }
  // 簡單輸出個 Hello World
  client.query('SELECT $1::varchar AS OUT', ["Hello World"], function(err, result) {
    done();// 釋放連接（將其返回給連接池）
    if(err) {
      return console.error('查詢出錯', err);
    }
    console.log(result.rows[0].out); //output: Hello World
  });
});

輸出：

Hello World

參數 done 是一個函數，調用這個函數可以將關閉連接（即將連接還給連接池）。

上面的是需要寫回調的異步查詢，可以使用 ES 7 中 await 和 async （但需安裝最新版本的 pg ，另外，需要使用7.2以上的 nodejs ，最好就是用最新的 nodejs ）優化代碼，如下：

// Async & Await 方式（需 node ^7.2.1，運行時使用 node --harmony-async-await index.js）
var query = async () => {  
  // 同步創建連接
  var connect = await pool.connect()
  try {
    // 同步等待結果
    var res = await connect.query('SELECT $1::varchar AS OUT', ['Hello World By Async&Await'])
    console.log(res.rows[0].out) // 可以通過rows遍歷數據
  } finally {
    connect.release()
  }
}

// 異步進行數據庫處理
query().catch(e => console.error(e.message, e.stack));

在升級了 nodejs 之后，執行代碼的時候，需要加參數 --harmony-async-await

npm --harmony-async-await index.js

當然，都支持到 ES7 了， ES6 的 Promise 方法肯定是支持的，如下：

pool.connect().then(client=>{  
    client.query('SELECT $1::varchar AS OUT', ['Hello World By Promise']).then(res=>{
        client.release()
        console.log(res.rows[0].out)
    }).catch(e => {
        client.release()
        console.error('query error', e.message, e.stack)
    })
})

插入、修改、刪除數據

插入、修改、刪除數據和查詢的差不多

// 在表test中插入、修改、刪除數據，共兩個字段 (name, age)
pool.connect().then(client=>{  
    // insert 數據
    client.query("INSERT INTO test(name, age) VALUES($1::varchar, $2::int)", ["xiaoming","20"]).then(res=>{
        console.log("Insert Success")
        // 如果是自增ID，有返回值的，在res里
        return res;
    }).then(res=>{
        // 查詢xiaoming
        return client.query("Select * FROM test WHERE name = $1", ["xiaoming"]);
    }).then(res=>{
        // 輸出結果，看是否插入成功
        console.log(res.rows[0])
    }).then(res=>{
        // update 數據，將age改為21
        return client.query("UPDATE test SET age=$1 WHERE name=$2", [21, "xiaoming"])
    }).then(res=>{
        // 再查詢一次xiaoming
        return client.query("Select * FROM test WHERE name = $1", ["xiaoming"]);
    }).then(res=>{
        // 再輸出結果，看是否改為了21
        console.log(res.rows[0])
    }).then(res=>{
        // 刪除數據
        client.query("DELETE FROM test WHERE name=$1", ["xiaoming"])
    }).then(res=>{
        // 最后再查詢一次xiaoming
        res = client.query("Select * FROM test WHERE name = $1", ["xiaoming"]);
        // 釋放連接
        client.release()
        return res
    }).then(res=>{
        // 再輸出結果，沒數據 undefined
        console.log(res.rows[0])
    })
})

上面插入、更新里代碼都沒有進行錯誤處理，按道理是要加的，但如果要加 try...catch... 的話，就太麻煩了（畢竟只是示例）.

事件監聽

可以添加 error 事件方法監聽連接池情況

pool.on("error", function(err, client){  
    console.log("error --> ", err)
})

現在連接池的最大空閑時間是3s，也就是3s還沒使用連接，就釋放連接，可將這個時間設置得長一些，比如30s，這就讓我們有足夠的時間關掉數據庫進行測試（與數據庫連接一斷開，這個事件就被觸發了，生產環境中，可以用來寫日志啊、發郵件短信通知什么的。。。）。

另外，還可以監聽 acquire 和 connect 事件，前者在連接被客戶端獲取時觸發，后者在連接生成以及客戶端與數據庫交互時觸發。

pool.on('acquire', function (client) {  
  console.log("acquire Event")
})

pool.on('connect', function () {  
  console.log("connect Event")
})

不使用連接池的客戶端

不使用連接池時，直接創建客戶端即可：

var client = new pg.Client();

連接池只是用來管理（緩存）連接（即客戶端）的，查詢之類的方法跟它沒關系。

代碼地址： https://github.com/zgljl2012/demo-node-postgres

 

 

來自：http://www.zgljl2012.com/2016/12/18/node-js-lian-jie-postgresqlshu-ju-ku/