64 lines
1.7 KiB
TypeScript
64 lines
1.7 KiB
TypeScript
import { PostgreSQLClient } from './client';
|
|
import { postgresConfig } from '@stock-bot/config';
|
|
import type { PostgreSQLClientConfig, PostgreSQLConnectionOptions } from './types';
|
|
|
|
/**
|
|
* Factory function to create a PostgreSQL client instance
|
|
*/
|
|
export function createPostgreSQLClient(
|
|
config?: Partial<PostgreSQLClientConfig>,
|
|
options?: PostgreSQLConnectionOptions
|
|
): PostgreSQLClient {
|
|
return new PostgreSQLClient(config, options);
|
|
}
|
|
|
|
/**
|
|
* Create a PostgreSQL client with default configuration
|
|
*/
|
|
export function createDefaultPostgreSQLClient(): PostgreSQLClient {
|
|
const config: Partial<PostgreSQLClientConfig> = {
|
|
host: postgresConfig.POSTGRES_HOST,
|
|
port: postgresConfig.POSTGRES_PORT,
|
|
database: postgresConfig.POSTGRES_DATABASE,
|
|
username: postgresConfig.POSTGRES_USERNAME,
|
|
password: postgresConfig.POSTGRES_PASSWORD
|
|
};
|
|
|
|
return new PostgreSQLClient(config);
|
|
}
|
|
|
|
/**
|
|
* Singleton PostgreSQL client instance
|
|
*/
|
|
let defaultClient: PostgreSQLClient | null = null;
|
|
|
|
/**
|
|
* Get or create the default PostgreSQL client instance
|
|
*/
|
|
export function getPostgreSQLClient(): PostgreSQLClient {
|
|
if (!defaultClient) {
|
|
defaultClient = createDefaultPostgreSQLClient();
|
|
}
|
|
return defaultClient;
|
|
}
|
|
|
|
/**
|
|
* Connect to PostgreSQL using the default client
|
|
*/
|
|
export async function connectPostgreSQL(): Promise<PostgreSQLClient> {
|
|
const client = getPostgreSQLClient();
|
|
if (!client.connected) {
|
|
await client.connect();
|
|
}
|
|
return client;
|
|
}
|
|
|
|
/**
|
|
* Disconnect from PostgreSQL
|
|
*/
|
|
export async function disconnectPostgreSQL(): Promise<void> {
|
|
if (defaultClient) {
|
|
await defaultClient.disconnect();
|
|
defaultClient = null;
|
|
}
|
|
}
|