const { Pool } = require('pg');

// 金山烟草（默认）
const jsPool = new Pool({
  host: process.env.DB_HOST || '121.37.183.152',
  port: parseInt(process.env.DB_PORT || '5432'),
  database: process.env.DB_NAME || 'lp-card-test',
  user: process.env.DB_USER || 'postgres',
  password: process.env.DB_PASSWORD || '123456',
  max: 20,
  idleTimeoutMillis: 30000,
  connectionTimeoutMillis: 5000,
});

// 嘉定烟草
const jdPool = new Pool({
  host: process.env.JD_DB_HOST || 'pgm-uf69f8n00s9mqn91go.pg.rds.aliyuncs.com',
  port: parseInt(process.env.JD_DB_PORT || '5432'),
  database: process.env.JD_DB_NAME || 'gt1_fmzgo',
  user: process.env.JD_DB_USER || 'gt1fmzgo',
  password: process.env.JD_DB_PASSWORD || 'Bm123456@',
  max: 20,
  idleTimeoutMillis: 30000,
  connectionTimeoutMillis: 5000,
});

jsPool.on('error', (err) => console.error('金山数据库连接池异常:', err.message));
jdPool.on('error', (err) => console.error('嘉定数据库连接池异常:', err.message));

function getPool(project) {
  return project === 'jd' ? jdPool : jsPool;
}

module.exports = { getPool, jsPool, jdPool };