查看: 1004|回复: 0

[HTML代码] Node.js 连接 postgreSQL数据库

发表于 2017-7-23 12:50:24

首先使用npm安装数据库连接模块:

  1. npm install --save pg
复制代码
连接池创建

然后代码中引入 pg 模块,并编写数据库配置:

  1. var pg = require('pg');
  2. // 数据库配置
  3. var config = {
  4. user:"postgres",
  5. database:"ghost",
  6. password:"123456",
  7. port:5432,
  8. // 扩展属性
  9. max:20, // 连接池最大连接数
  10. idleTimeoutMillis:3000, // 连接最大空闲时间 3s
  11. }
复制代码

pg 模块中有两种数据库连接方式,先讲连接池模式,下面是创建连接池:

  1. // 创建连接池
  2. var pool = new pg.Pool(config);
复制代码

传入配置后就创建好了连接池。

查询数据

查询首先创建好连接,然后调用 api 进行查询:

  1. // 查询
  2. pool.connect(function(err, client, done) {
  3. if(err) {
  4. return console.error('数据库连接出错', err);
  5. }
  6. // 简单输出个 Hello World
  7. client.query('SELECT $1::varchar AS OUT', ["Hello World"], function(err, result) {
  8. done();// 释放连接(将其返回给连接池)
  9. if(err) {
  10. return console.error('查询出错', err);
  11. }
  12. console.log(result.rows[0].out); //output: Hello World
  13. });
  14. });
复制代码

输出:

  1. Hello World
复制代码

参数 done 是一个函数,调用这个函数可以将关闭连接(即将连接还给连接池)。

上面的是需要写回调的异步查询,可以使用 ES 7 中 await 和 async (但需安装最新版本的 pg ,另外,需要使用7.2以上的 nodejs ,最好就是用最新的 nodejs )优化代码,如下:

  1. // Async & Await 方式(需 node ^7.2.1,运行时使用 node --harmony-async-await index.js)
  2. var query = async () => {
  3. // 同步创建连接
  4. var connect = await pool.connect()
  5. try {
  6. // 同步等待结果
  7. var res = await connect.query('SELECT $1::varchar AS OUT', ['Hello World By Async&Await'])
  8. console.log(res.rows[0].out) // 可以通过rows遍历数据
  9. } finally {
  10. connect.release()
  11. }
  12. }
  13. // 异步进行数据库处理
  14. query().catch(e => console.error(e.message, e.stack));
复制代码

在升级了 nodejs 之后,执行代码的时候,需要加参数 --harmony-async-await

  1. npm --harmony-async-await index.js
复制代码

当然,都支持到 ES7 了, ES6 的 Promise 方法肯定是支持的,如下:

  1. pool.connect().then(client=>{
  2. client.query('SELECT $1::varchar AS OUT', ['Hello World By Promise']).then(res=>{
  3. client.release()
  4. console.log(res.rows[0].out)
  5. }).catch(e => {
  6. client.release()
  7. console.error('query error', e.message, e.stack)
  8. })
  9. })
复制代码
插入、修改、删除数据

插入、修改、删除数据和查询的差不多

  1. // 在表test中插入、修改、删除数据,共两个字段 (name, age)
  2. pool.connect().then(client=>{
  3. // insert 数据
  4. client.query("INSERT INTO test(name, age) VALUES($1::varchar, $2::int)", ["xiaoming","20"]).then(res=>{
  5. console.log("Insert Success")
  6. // 如果是自增ID,有返回值的,在res里
  7. return res;
  8. }).then(res=>{
  9. // 查询xiaoming
  10. return client.query("Select * FROM test WHERE name = $1", ["xiaoming"]);
  11. }).then(res=>{
  12. // 输出结果,看是否插入成功
  13. console.log(res.rows[0])
  14. }).then(res=>{
  15. // update 数据,将age改为21
  16. return client.query("UPDATE test SET age=$1 WHERE name=$2", [21, "xiaoming"])
  17. }).then(res=>{
  18. // 再查询一次xiaoming
  19. return client.query("Select * FROM test WHERE name = $1", ["xiaoming"]);
  20. }).then(res=>{
  21. // 再输出结果,看是否改为了21
  22. console.log(res.rows[0])
  23. }).then(res=>{
  24. // 删除数据
  25. client.query("DELETE FROM test WHERE name=$1", ["xiaoming"])
  26. }).then(res=>{
  27. // 最后再查询一次xiaoming
  28. res = client.query("Select * FROM test WHERE name = $1", ["xiaoming"]);
  29. // 释放连接
  30. client.release()
  31. return res
  32. }).then(res=>{
  33. // 再输出结果,没数据 undefined
  34. console.log(res.rows[0])
  35. })
  36. })
复制代码

上面插入、更新里代码都没有进行错误处理,按道理是要加的,但如果要加 try...catch... 的话,就太麻烦了(毕竟只是示例).

事件监听

可以添加 error 事件方法监听连接池情况

  1. pool.on("error", function(err, client){
  2. console.log("error --> ", err)
  3. })
复制代码

现在连接池的最大空闲时间是3s,也就是3s还没使用连接,就释放连接,可将这个时间设置得长一些,比如30s,这就让我们有足够的时间关掉数据库进行测试(与数据库连接一断开,这个事件就被触发了,生产环境中,可以用来写日志啊、发邮件短信通知什么的。。。)。

另外,还可以监听 acquire 和 connect 事件,前者在连接被客户端获取时触发,后者在连接生成以及客户端与数据库交互时触发。

  1. pool.on('acquire', function (client) {
  2. console.log("acquire Event")
  3. })
  4. pool.on('connect', function () {
  5. console.log("connect Event")
  6. })
复制代码
不使用连接池的客户端

不使用连接池时,直接创建客户端即可:

  1. var client = new pg.Client();
复制代码

连接池只是用来管理(缓存)连接(即客户端)的,查询之类的方法跟它没关系。

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

来自:http://www.zgljl2012.com/2016/12 ... ostgresqlshu-ju-ku/



回复

使用道具 举报