83 lines
1.9 KiB
TypeScript
83 lines
1.9 KiB
TypeScript
import pg from 'pg'
|
|
|
|
const { Pool } = pg
|
|
|
|
// 환경 설정
|
|
const isDev = process.env.NODE_ENV !== 'production'
|
|
|
|
// PostgreSQL 연결 풀 (싱글톤)
|
|
let pool: pg.Pool | null = null
|
|
|
|
/**
|
|
* PostgreSQL 연결 풀 가져오기
|
|
*/
|
|
export function getPool(): pg.Pool {
|
|
if (!pool) {
|
|
pool = new Pool({
|
|
host: process.env.DB_HOST || 'localhost',
|
|
port: parseInt(process.env.DB_PORT || '5432'),
|
|
database: process.env.DB_NAME || 'osolit_monitor',
|
|
user: process.env.DB_USER || 'postgres',
|
|
password: process.env.DB_PASSWORD || '',
|
|
max: 10,
|
|
idleTimeoutMillis: 30000,
|
|
connectionTimeoutMillis: 2000,
|
|
})
|
|
|
|
pool.on('error', (err) => {
|
|
console.error('[DB] Unexpected pool error:', err)
|
|
})
|
|
|
|
console.log(`[DB] PostgreSQL pool created (${isDev ? 'development' : 'production'})`)
|
|
}
|
|
return pool
|
|
}
|
|
|
|
/**
|
|
* 단일 쿼리 실행 (자동 커넥션 반환)
|
|
*/
|
|
export async function query<T = any>(sql: string, params?: any[]): Promise<T[]> {
|
|
const pool = getPool()
|
|
const result = await pool.query(sql, params)
|
|
return result.rows as T[]
|
|
}
|
|
|
|
/**
|
|
* 단일 행 조회
|
|
*/
|
|
export async function queryOne<T = any>(sql: string, params?: any[]): Promise<T | null> {
|
|
const rows = await query<T>(sql, params)
|
|
return rows[0] || null
|
|
}
|
|
|
|
/**
|
|
* INSERT/UPDATE/DELETE 실행
|
|
*/
|
|
export async function execute(sql: string, params?: any[]): Promise<number> {
|
|
const pool = getPool()
|
|
const result = await pool.query(sql, params)
|
|
return result.rowCount || 0
|
|
}
|
|
|
|
/**
|
|
* 스케줄러 자동시작 여부
|
|
*/
|
|
export function shouldAutoStartScheduler(): boolean {
|
|
const envValue = process.env.AUTO_START_SCHEDULER
|
|
if (envValue !== undefined) {
|
|
return envValue === 'true'
|
|
}
|
|
return !isDev
|
|
}
|
|
|
|
/**
|
|
* 연결 풀 종료
|
|
*/
|
|
export async function closeDb(): Promise<void> {
|
|
if (pool) {
|
|
await pool.end()
|
|
pool = null
|
|
console.log('[DB] PostgreSQL pool closed')
|
|
}
|
|
}
|