每个用户一个连接答案

【问题标题】：One connection per user每个用户一个连接
【发布时间】：2019-05-04 03:25:22
【问题描述】：

我知道这个问题已经被问过了，但似乎还有一些事情需要澄清。 :)

数据库的设计方式是每个用户都有适当的权限来阅读文档，因此连接池需要与不同的用户建立连接，这超出了连接池的概念。由于优化和性能，我需要调用所谓的“用户准备”，包括设置会话变量、计算和缓存缓存中的值等，然后执行查询。

目前，我有两个解决方案。在第一个解决方案中，我首先检查是否已为用户准备好一切，然后执行一个或多个查询。如果没有准备好，那么我需要调用“用户准备”，然后执行一个或多个查询。使用这个解决方案，我会损失很多性能，因为每次我都必须进行检查，所以我决定使用另一个解决方案。

第二种解决方案包括“数据库池”，其中每个池供一个用户使用。只有在第一次连接时useCount === 0（我不使用{direct：true}）我调用“用户准备”（它是设置一些会话变量并准备缓存的存储过程）然后执行sql查询。

我在 initOptions 参数中的 connect 事件中为初始化 pgPromise 所做的用户准备。我使用了 pg-promise-demo，所以我不需要解释其余的代码。

使用数据库池包装的 pgp 初始化代码如下所示：

import * as promise from "bluebird";

import pgPromise from "pg-promise";

import { IDatabase, IMain, IOptions } from "pg-promise";
import { IExtensions, ProductsRepository, UsersRepository, Session, getUserFromJWT } from "../db/repos";
import { dbConfig } from "../server/config";

// pg-promise initialization options:
export const initOptions: IOptions<IExtensions> = {
  promiseLib: promise,

  async connect(client: any, dc: any, useCount: number) {
    if (useCount === 0) {
      try {
        await client.query(pgp.as.format("select prepareUser($1)", [getUserFromJWT(session.JWT)]));
      } catch(error) {
        console.error(error);
      }
    }
  },

  extend(obj: IExtensions, dc: any) {
    obj.users = new UsersRepository(obj);
    obj.products = new ProductsRepository(obj);
  }
};

type DB = IDatabase<IExtensions>&IExtensions;

const pgp: IMain = pgPromise(initOptions);

class DBPool {
  private pool = new Map();

  public get = (ct: any): DB => {
    const checkConfig = {...dbConfig, ...ct};
    const {host, port, database, user} = checkConfig;
    const dbKey = JSON.stringify({host, port, database, user})
    let db: DB = this.pool.get(dbKey) as DB;
    if (!db) {
      // const pgp: IMain = pgPromise(initOptions);
      db = pgp(checkConfig) as DB;
      this.pool.set(dbKey, db);
    }
    return db;
  }
}
export const dbPool = new DBPool();

import diagnostics = require("./diagnostics");
diagnostics.init(initOptions);

而 web api 看起来像：

GET("/api/getuser/:id", (req: Request) => {
  const user = getUserFromJWT(session.JWT);
  const db = dbPool.get({ user });
  return db.users.findById(req.params.id);
});

我对源代码是否正确实例化 pgp 或是否应该在 get 方法内的 if 块中实例化感兴趣（该行已被注释）？

我已经看到 pg-promise 使用从 dbPool.js 导出的 DatabasePool 单例，这类似于我的 DBPool 类，但目的是给出“警告：为同一连接创建重复的数据库对象”。是否可以使用 DatabasePool 单例而不是我的 dbPool 单例？

在我看来dbContext（pgp初始化中的第二个参数）可以解决我的问题，但前提是它可以作为函数转发，而不是作为值或对象。访问数据库对象时我错了还是 dbContext 可以是动态的？

我想知道是否有第三种（更好的）解决方案？或任何其他建议。

【问题讨论】：

标签： node.js postgresql pg-promise

【解决方案1】：

如果您对此警告感到困扰：

警告：为同一连接创建重复的数据库对象

但您的意图是为每个用户维护一个单独的池，您可以通过为连接提供任何唯一参数来表明这一点。例如，您可以在用户名中包含自定义属性：

const cn = {
    database: 'my-db',
    port: 12345,
    user: 'my-login-user',
    password: 'my-login-password'

    ....

    my_dynamic_user: 'john-doe'
}

这足以让库看到您的连接中有一些独特的东西，它与其他连接不匹配，因此不会产生该警告。

这也适用于连接字符串。

请注意，只有当连接总数远远超过用户数时，您想要达到的效果才能正常工作。例如，如果您最多可以使用 100 个连接，最多有 10 个用户。然后您可以分配 10 个池，每个池中最多有 10 个连接。否则，系统的可扩展性将受到影响，因为连接总数是非常有限的资源，通常不会超过 100 个连接，因为它会在同时运行如此多的物理连接的 CPU 上产生过多的负载。这就是为什么共享单个连接池可以更好地扩展。

【讨论】：