使用Node.js连接池与PostgreSQL的方法

Question

我正在使用Node.js作为我的REST API主机。总体功能非常基本，即进行REST调用，应查询PostgreSQL数据库并返回数据。我遇到的问题是代码组织方面存在问题。

我有一个server.js文件，其中包含基本初始化过程。现在，我应该把PostgreSQL连接放在哪里，并在需要时调用它来查询数据库。当前的结构如下所示。因此，我认为我应该将PostgreSQL连接代码移动到server.js中？正确的结构应该是什么？

server.js

"use strict";

 var express = require('express'),
  app = express(),
  port = process.env.PORT || 4001,
  bodyParser = require('body-parser');

  app.use(bodyParser.urlencoded({ extended: true }));
  app.use(bodyParser.json());
  app.use(function (req, res, next) {

    // Website you wish to allow to connect
    res.setHeader('Access-Control-Allow-Origin', '*')
    // Request methods you wish to allow
    res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, PUT, PATCH, DELETE');
    // Request headers you wish to allow
    res.setHeader('Access-Control-Allow-Headers', 'X-Requested-With,content-type');
    // Set to true if you need the website to include cookies in the requests sent
    // to the API (e.g. in case you use sessions)
    res.setHeader('Access-Control-Allow-Credentials', true);
    // Pass to next layer of middleware
    next();
});
var routes = require('./api/routes/pushRoutes');
routes(app);

app.listen(port);
console.log('push notification server started on: ' + port);

pushController.js

'use strict';

exports.send_notification = function(req, res) {
    console.log("start of send_notification ");

     if (req.method == 'POST') {
        var jsonString = '';

        req.on('data', function (data) {
            jsonString += data;
        });

        req.on('end', function () {
            console.log(JSON.parse(jsonString));
            var json = JSON.parse(jsonString);
            var timeNow = json.message;

            //get device token from db
            var deviceToken = '';
            var pg = require('pg')
            var format = require('pg-format')
            var PGUSER = 'deploy'
            var PGDATABASE = 'oscpushserver'
            var config = {
              user: PGUSER, // name of the user account
              database: PGDATABASE, // name of the database
              max: 10, // max number of clients in the pool
              idleTimeoutMillis: 30000 
            }

            var pool = new pg.Pool(config)
            var myClient

            pool.connect(function (err, client, done) {
              if (err) console.log(err)
              app.listen(3000, function () {
                console.log('listening on 3000')
              })
              myClient = client
              var ageQuery = format('SELECT * from push_table WHERE seq_id = 1')
              myClient.query(ageQuery, function (err, result) {
                if (err) {
                  console.log(err)
                }
                console.log("row data:" + result.rows[0])
                console.log("device id from db:" + result.rows[0].device_id)
                deviceToken =result.rows[0].device_id;


              })
            });


            res.json(JSON.parse(jsonString));
        });
    }

};

- Vik

---

看起来没问题，如果您想要更短的代码，可以尝试使用 https://node-postgres.com/。 - Mμ.

在我的情况下，pool.connect 内的应用程序未被解析，因为它是在 server.js 中定义的。 - Vik

哦，抱歉我的错，你已经在使用node-postgres了。我可以问一下为什么你把app放在pool里面吗？你已经让app在server.js上监听了。 - Mμ.

我从两个不同的位置复制粘贴了代码，而且我对Node.js非常陌生，所以不知道该如何解决它。 - Vik

pg-promise-demo 提供了最佳的代码结构，可与 pg-promise 一起使用，用于数据库。 - vitaly-t

Answer 1

我假设您想创建一个只从数据库获取数据的REST API。

server.js

大部分看起来都不错，只是你没有正确使用路由。要使用你定义的路由，需要使用app.use:

app.use('/', routes);

database.js 我会创建一个新的文件，将数据库凭据放在一个地方，并创建一个新的连接池。

var pg = require('pg')
var PGUSER = 'deploy'
var PGDATABASE = 'oscpushserver'
var config = {
  user: PGUSER, // name of the user account
  database: PGDATABASE, // name of the database
  max: 10, // max number of clients in the pool
  idleTimeoutMillis: 30000 
}

var pool = new pg.Pool(config);

module.exports = pool;

pushController

我假设你想在server.js中引用这个文件。所以要确保路径指向'.../pushController'而不是'./api/routes/pushRoutes'。

const express = require('express');
const router = express.Router();
const format = require('pg-format');
const pool = require('../path/to/database.js');

router.get(function(req, res, next) {
  pool.connect(function (err, client, done) {
    if (err) throw new Error(err);
    var ageQuery = format('SELECT * from push_table WHERE seq_id = 1')
    client.query(ageQuery, function (err, result) {
      if (err) throw new Error(err);
      res.json(result.rows[0]); 
    })
  }); 
});

module.exports = router;

现在，对API服务器的GET请求应该从您的PostgreSQL数据库返回一个JSON对象。

有很多关于如何使用Node和PostgreSQL创建REST API的在线示例。其中之一是这个：https://github.com/mjhea0/node-postgres-todo。

- Mμ.

---

我应该如何在我的路由器中跨多个路由文件共享连接池？在./routes目录中，我为每个端点都有一个单独的文件，并且需要在应用程序中的一个位置实例化连接池并将其暴露给所有路由。在Koa中，我可以将连接池实例放在ctx.pool上，那么在Express中是否可能呢？ - chovy