首先使用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