running prettier for cleanup
This commit is contained in:
parent
fe7733aeb5
commit
d85cd58acd
151 changed files with 29158 additions and 27966 deletions
|
|
@ -1,339 +1,348 @@
|
|||
import { Pool, PoolClient, QueryResult as PgQueryResult, QueryResultRow } from 'pg';
|
||||
import { postgresConfig } from '@stock-bot/config';
|
||||
import { getLogger } from '@stock-bot/logger';
|
||||
import type {
|
||||
PostgreSQLClientConfig,
|
||||
PostgreSQLConnectionOptions,
|
||||
QueryResult,
|
||||
TransactionCallback
|
||||
} from './types';
|
||||
import { PostgreSQLHealthMonitor } from './health';
|
||||
import { PostgreSQLQueryBuilder } from './query-builder';
|
||||
import { PostgreSQLTransactionManager } from './transactions';
|
||||
|
||||
/**
|
||||
* PostgreSQL Client for Stock Bot
|
||||
*
|
||||
* Provides type-safe access to PostgreSQL with connection pooling,
|
||||
* health monitoring, and transaction support.
|
||||
*/
|
||||
export class PostgreSQLClient {
|
||||
private pool: Pool | null = null;
|
||||
private readonly config: PostgreSQLClientConfig;
|
||||
private readonly options: PostgreSQLConnectionOptions;
|
||||
private readonly logger: ReturnType<typeof getLogger>;
|
||||
private readonly healthMonitor: PostgreSQLHealthMonitor;
|
||||
private readonly transactionManager: PostgreSQLTransactionManager;
|
||||
private isConnected = false;
|
||||
|
||||
constructor(
|
||||
config?: Partial<PostgreSQLClientConfig>,
|
||||
options?: PostgreSQLConnectionOptions
|
||||
) {
|
||||
this.config = this.buildConfig(config);
|
||||
this.options = {
|
||||
retryAttempts: 3,
|
||||
retryDelay: 1000,
|
||||
healthCheckInterval: 30000,
|
||||
...options
|
||||
};
|
||||
|
||||
this.logger = getLogger('postgres-client');
|
||||
this.healthMonitor = new PostgreSQLHealthMonitor(this);
|
||||
this.transactionManager = new PostgreSQLTransactionManager(this);
|
||||
}
|
||||
|
||||
/**
|
||||
* Connect to PostgreSQL
|
||||
*/
|
||||
async connect(): Promise<void> {
|
||||
if (this.isConnected && this.pool) {
|
||||
return;
|
||||
}
|
||||
|
||||
let lastError: Error | null = null;
|
||||
|
||||
for (let attempt = 1; attempt <= this.options.retryAttempts!; attempt++) {
|
||||
try {
|
||||
this.logger.info(`Connecting to PostgreSQL (attempt ${attempt}/${this.options.retryAttempts})...`);
|
||||
|
||||
this.pool = new Pool(this.buildPoolConfig());
|
||||
|
||||
// Test the connection
|
||||
const client = await this.pool.connect();
|
||||
await client.query('SELECT 1');
|
||||
client.release();
|
||||
|
||||
this.isConnected = true;
|
||||
this.logger.info('Successfully connected to PostgreSQL');
|
||||
|
||||
// Start health monitoring
|
||||
this.healthMonitor.start();
|
||||
|
||||
// Setup error handlers
|
||||
this.setupErrorHandlers();
|
||||
|
||||
return;
|
||||
} catch (error) {
|
||||
lastError = error as Error;
|
||||
this.logger.error(`PostgreSQL connection attempt ${attempt} failed:`, error);
|
||||
|
||||
if (this.pool) {
|
||||
await this.pool.end();
|
||||
this.pool = null;
|
||||
}
|
||||
|
||||
if (attempt < this.options.retryAttempts!) {
|
||||
await this.delay(this.options.retryDelay! * attempt);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
throw new Error(`Failed to connect to PostgreSQL after ${this.options.retryAttempts} attempts: ${lastError?.message}`);
|
||||
}
|
||||
|
||||
/**
|
||||
* Disconnect from PostgreSQL
|
||||
*/
|
||||
async disconnect(): Promise<void> {
|
||||
if (!this.pool) {
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
this.healthMonitor.stop();
|
||||
await this.pool.end();
|
||||
this.isConnected = false;
|
||||
this.pool = null;
|
||||
this.logger.info('Disconnected from PostgreSQL');
|
||||
} catch (error) {
|
||||
this.logger.error('Error disconnecting from PostgreSQL:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute a query
|
||||
*/
|
||||
async query<T extends QueryResultRow = any>(text: string, params?: any[]): Promise<QueryResult<T>> {
|
||||
if (!this.pool) {
|
||||
throw new Error('PostgreSQL client not connected');
|
||||
}
|
||||
|
||||
const startTime = Date.now();
|
||||
|
||||
try {
|
||||
const result = await this.pool.query<T>(text, params);
|
||||
const executionTime = Date.now() - startTime;
|
||||
|
||||
this.logger.debug(`Query executed in ${executionTime}ms`, {
|
||||
query: text.substring(0, 100),
|
||||
params: params?.length
|
||||
});
|
||||
|
||||
return {
|
||||
...result,
|
||||
executionTime
|
||||
} as QueryResult<T>;
|
||||
} catch (error) {
|
||||
const executionTime = Date.now() - startTime;
|
||||
this.logger.error(`Query failed after ${executionTime}ms:`, {
|
||||
error,
|
||||
query: text,
|
||||
params
|
||||
});
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute multiple queries in a transaction
|
||||
*/
|
||||
async transaction<T>(callback: TransactionCallback<T>): Promise<T> {
|
||||
return await this.transactionManager.execute(callback);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a query builder instance
|
||||
*/
|
||||
queryBuilder(): PostgreSQLQueryBuilder {
|
||||
return new PostgreSQLQueryBuilder(this);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with SELECT
|
||||
*/
|
||||
select(columns: string | string[] = '*'): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().select(columns);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with INSERT
|
||||
*/
|
||||
insert(table: string): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().insert(table);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with UPDATE
|
||||
*/
|
||||
update(table: string): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().update(table);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with DELETE
|
||||
*/
|
||||
delete(table: string): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().delete(table);
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute a stored procedure or function
|
||||
*/
|
||||
async callFunction<T extends QueryResultRow = any>(functionName: string, params?: any[]): Promise<QueryResult<T>> {
|
||||
const placeholders = params ? params.map((_, i) => `$${i + 1}`).join(', ') : '';
|
||||
const query = `SELECT * FROM ${functionName}(${placeholders})`;
|
||||
return await this.query<T>(query, params);
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if a table exists
|
||||
*/
|
||||
async tableExists(tableName: string, schemaName: string = 'public'): Promise<boolean> {
|
||||
const result = await this.query(
|
||||
`SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_schema = $1 AND table_name = $2
|
||||
)`,
|
||||
[schemaName, tableName]
|
||||
);
|
||||
return result.rows[0].exists;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get table schema information
|
||||
*/
|
||||
async getTableSchema(tableName: string, schemaName: string = 'public'): Promise<any[]> {
|
||||
const result = await this.query(
|
||||
`SELECT
|
||||
column_name,
|
||||
data_type,
|
||||
is_nullable,
|
||||
column_default,
|
||||
character_maximum_length
|
||||
FROM information_schema.columns
|
||||
WHERE table_schema = $1 AND table_name = $2
|
||||
ORDER BY ordinal_position`,
|
||||
[schemaName, tableName]
|
||||
);
|
||||
return result.rows;
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute EXPLAIN for query analysis
|
||||
*/
|
||||
async explain(query: string, params?: any[]): Promise<any[]> {
|
||||
const explainQuery = `EXPLAIN (ANALYZE, BUFFERS, FORMAT JSON) ${query}`;
|
||||
const result = await this.query(explainQuery, params);
|
||||
return result.rows[0]['QUERY PLAN'];
|
||||
}
|
||||
|
||||
/**
|
||||
* Get database statistics
|
||||
*/
|
||||
async getStats(): Promise<any> {
|
||||
const result = await this.query(`
|
||||
SELECT
|
||||
(SELECT count(*) FROM pg_stat_activity WHERE state = 'active') as active_connections,
|
||||
(SELECT count(*) FROM pg_stat_activity WHERE state = 'idle') as idle_connections,
|
||||
(SELECT setting FROM pg_settings WHERE name = 'max_connections') as max_connections,
|
||||
pg_size_pretty(pg_database_size(current_database())) as database_size
|
||||
`);
|
||||
return result.rows[0];
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if client is connected
|
||||
*/
|
||||
get connected(): boolean {
|
||||
return this.isConnected && !!this.pool;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the underlying connection pool
|
||||
*/
|
||||
get connectionPool(): Pool | null {
|
||||
return this.pool;
|
||||
}
|
||||
|
||||
private buildConfig(config?: Partial<PostgreSQLClientConfig>): PostgreSQLClientConfig {
|
||||
return {
|
||||
host: config?.host || postgresConfig.POSTGRES_HOST,
|
||||
port: config?.port || postgresConfig.POSTGRES_PORT,
|
||||
database: config?.database || postgresConfig.POSTGRES_DATABASE,
|
||||
username: config?.username || postgresConfig.POSTGRES_USERNAME,
|
||||
password: config?.password || postgresConfig.POSTGRES_PASSWORD,
|
||||
poolSettings: {
|
||||
min: postgresConfig.POSTGRES_POOL_MIN,
|
||||
max: postgresConfig.POSTGRES_POOL_MAX,
|
||||
idleTimeoutMillis: postgresConfig.POSTGRES_POOL_IDLE_TIMEOUT,
|
||||
...config?.poolSettings
|
||||
},
|
||||
ssl: {
|
||||
enabled: postgresConfig.POSTGRES_SSL,
|
||||
rejectUnauthorized: postgresConfig.POSTGRES_SSL_REJECT_UNAUTHORIZED,
|
||||
...config?.ssl
|
||||
},
|
||||
timeouts: {
|
||||
query: postgresConfig.POSTGRES_QUERY_TIMEOUT,
|
||||
connection: postgresConfig.POSTGRES_CONNECTION_TIMEOUT,
|
||||
statement: postgresConfig.POSTGRES_STATEMENT_TIMEOUT,
|
||||
lock: postgresConfig.POSTGRES_LOCK_TIMEOUT,
|
||||
idleInTransaction: postgresConfig.POSTGRES_IDLE_IN_TRANSACTION_SESSION_TIMEOUT,
|
||||
...config?.timeouts
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
private buildPoolConfig(): any {
|
||||
return {
|
||||
host: this.config.host,
|
||||
port: this.config.port,
|
||||
database: this.config.database,
|
||||
user: this.config.username,
|
||||
password: this.config.password,
|
||||
min: this.config.poolSettings?.min,
|
||||
max: this.config.poolSettings?.max,
|
||||
idleTimeoutMillis: this.config.poolSettings?.idleTimeoutMillis,
|
||||
connectionTimeoutMillis: this.config.timeouts?.connection,
|
||||
query_timeout: this.config.timeouts?.query,
|
||||
statement_timeout: this.config.timeouts?.statement,
|
||||
lock_timeout: this.config.timeouts?.lock,
|
||||
idle_in_transaction_session_timeout: this.config.timeouts?.idleInTransaction,
|
||||
ssl: this.config.ssl?.enabled ? {
|
||||
rejectUnauthorized: this.config.ssl.rejectUnauthorized
|
||||
} : false
|
||||
};
|
||||
}
|
||||
|
||||
private setupErrorHandlers(): void {
|
||||
if (!this.pool) return;
|
||||
|
||||
this.pool.on('error', (error) => {
|
||||
this.logger.error('PostgreSQL pool error:', error);
|
||||
});
|
||||
|
||||
this.pool.on('connect', () => {
|
||||
this.logger.debug('New PostgreSQL client connected');
|
||||
});
|
||||
|
||||
this.pool.on('remove', () => {
|
||||
this.logger.debug('PostgreSQL client removed from pool');
|
||||
});
|
||||
}
|
||||
|
||||
private delay(ms: number): Promise<void> {
|
||||
return new Promise(resolve => setTimeout(resolve, ms));
|
||||
}
|
||||
}
|
||||
import { QueryResult as PgQueryResult, Pool, PoolClient, QueryResultRow } from 'pg';
|
||||
import { postgresConfig } from '@stock-bot/config';
|
||||
import { getLogger } from '@stock-bot/logger';
|
||||
import { PostgreSQLHealthMonitor } from './health';
|
||||
import { PostgreSQLQueryBuilder } from './query-builder';
|
||||
import { PostgreSQLTransactionManager } from './transactions';
|
||||
import type {
|
||||
PostgreSQLClientConfig,
|
||||
PostgreSQLConnectionOptions,
|
||||
QueryResult,
|
||||
TransactionCallback,
|
||||
} from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Client for Stock Bot
|
||||
*
|
||||
* Provides type-safe access to PostgreSQL with connection pooling,
|
||||
* health monitoring, and transaction support.
|
||||
*/
|
||||
export class PostgreSQLClient {
|
||||
private pool: Pool | null = null;
|
||||
private readonly config: PostgreSQLClientConfig;
|
||||
private readonly options: PostgreSQLConnectionOptions;
|
||||
private readonly logger: ReturnType<typeof getLogger>;
|
||||
private readonly healthMonitor: PostgreSQLHealthMonitor;
|
||||
private readonly transactionManager: PostgreSQLTransactionManager;
|
||||
private isConnected = false;
|
||||
|
||||
constructor(config?: Partial<PostgreSQLClientConfig>, options?: PostgreSQLConnectionOptions) {
|
||||
this.config = this.buildConfig(config);
|
||||
this.options = {
|
||||
retryAttempts: 3,
|
||||
retryDelay: 1000,
|
||||
healthCheckInterval: 30000,
|
||||
...options,
|
||||
};
|
||||
|
||||
this.logger = getLogger('postgres-client');
|
||||
this.healthMonitor = new PostgreSQLHealthMonitor(this);
|
||||
this.transactionManager = new PostgreSQLTransactionManager(this);
|
||||
}
|
||||
|
||||
/**
|
||||
* Connect to PostgreSQL
|
||||
*/
|
||||
async connect(): Promise<void> {
|
||||
if (this.isConnected && this.pool) {
|
||||
return;
|
||||
}
|
||||
|
||||
let lastError: Error | null = null;
|
||||
|
||||
for (let attempt = 1; attempt <= this.options.retryAttempts!; attempt++) {
|
||||
try {
|
||||
this.logger.info(
|
||||
`Connecting to PostgreSQL (attempt ${attempt}/${this.options.retryAttempts})...`
|
||||
);
|
||||
|
||||
this.pool = new Pool(this.buildPoolConfig());
|
||||
|
||||
// Test the connection
|
||||
const client = await this.pool.connect();
|
||||
await client.query('SELECT 1');
|
||||
client.release();
|
||||
|
||||
this.isConnected = true;
|
||||
this.logger.info('Successfully connected to PostgreSQL');
|
||||
|
||||
// Start health monitoring
|
||||
this.healthMonitor.start();
|
||||
|
||||
// Setup error handlers
|
||||
this.setupErrorHandlers();
|
||||
|
||||
return;
|
||||
} catch (error) {
|
||||
lastError = error as Error;
|
||||
this.logger.error(`PostgreSQL connection attempt ${attempt} failed:`, error);
|
||||
|
||||
if (this.pool) {
|
||||
await this.pool.end();
|
||||
this.pool = null;
|
||||
}
|
||||
|
||||
if (attempt < this.options.retryAttempts!) {
|
||||
await this.delay(this.options.retryDelay! * attempt);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
throw new Error(
|
||||
`Failed to connect to PostgreSQL after ${this.options.retryAttempts} attempts: ${lastError?.message}`
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Disconnect from PostgreSQL
|
||||
*/
|
||||
async disconnect(): Promise<void> {
|
||||
if (!this.pool) {
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
this.healthMonitor.stop();
|
||||
await this.pool.end();
|
||||
this.isConnected = false;
|
||||
this.pool = null;
|
||||
this.logger.info('Disconnected from PostgreSQL');
|
||||
} catch (error) {
|
||||
this.logger.error('Error disconnecting from PostgreSQL:', error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute a query
|
||||
*/
|
||||
async query<T extends QueryResultRow = any>(
|
||||
text: string,
|
||||
params?: any[]
|
||||
): Promise<QueryResult<T>> {
|
||||
if (!this.pool) {
|
||||
throw new Error('PostgreSQL client not connected');
|
||||
}
|
||||
|
||||
const startTime = Date.now();
|
||||
|
||||
try {
|
||||
const result = await this.pool.query<T>(text, params);
|
||||
const executionTime = Date.now() - startTime;
|
||||
|
||||
this.logger.debug(`Query executed in ${executionTime}ms`, {
|
||||
query: text.substring(0, 100),
|
||||
params: params?.length,
|
||||
});
|
||||
|
||||
return {
|
||||
...result,
|
||||
executionTime,
|
||||
} as QueryResult<T>;
|
||||
} catch (error) {
|
||||
const executionTime = Date.now() - startTime;
|
||||
this.logger.error(`Query failed after ${executionTime}ms:`, {
|
||||
error,
|
||||
query: text,
|
||||
params,
|
||||
});
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute multiple queries in a transaction
|
||||
*/
|
||||
async transaction<T>(callback: TransactionCallback<T>): Promise<T> {
|
||||
return await this.transactionManager.execute(callback);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a query builder instance
|
||||
*/
|
||||
queryBuilder(): PostgreSQLQueryBuilder {
|
||||
return new PostgreSQLQueryBuilder(this);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with SELECT
|
||||
*/
|
||||
select(columns: string | string[] = '*'): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().select(columns);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with INSERT
|
||||
*/
|
||||
insert(table: string): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().insert(table);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with UPDATE
|
||||
*/
|
||||
update(table: string): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().update(table);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new query builder with DELETE
|
||||
*/
|
||||
delete(table: string): PostgreSQLQueryBuilder {
|
||||
return this.queryBuilder().delete(table);
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute a stored procedure or function
|
||||
*/
|
||||
async callFunction<T extends QueryResultRow = any>(
|
||||
functionName: string,
|
||||
params?: any[]
|
||||
): Promise<QueryResult<T>> {
|
||||
const placeholders = params ? params.map((_, i) => `$${i + 1}`).join(', ') : '';
|
||||
const query = `SELECT * FROM ${functionName}(${placeholders})`;
|
||||
return await this.query<T>(query, params);
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if a table exists
|
||||
*/
|
||||
async tableExists(tableName: string, schemaName: string = 'public'): Promise<boolean> {
|
||||
const result = await this.query(
|
||||
`SELECT EXISTS (
|
||||
SELECT FROM information_schema.tables
|
||||
WHERE table_schema = $1 AND table_name = $2
|
||||
)`,
|
||||
[schemaName, tableName]
|
||||
);
|
||||
return result.rows[0].exists;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get table schema information
|
||||
*/
|
||||
async getTableSchema(tableName: string, schemaName: string = 'public'): Promise<any[]> {
|
||||
const result = await this.query(
|
||||
`SELECT
|
||||
column_name,
|
||||
data_type,
|
||||
is_nullable,
|
||||
column_default,
|
||||
character_maximum_length
|
||||
FROM information_schema.columns
|
||||
WHERE table_schema = $1 AND table_name = $2
|
||||
ORDER BY ordinal_position`,
|
||||
[schemaName, tableName]
|
||||
);
|
||||
return result.rows;
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute EXPLAIN for query analysis
|
||||
*/
|
||||
async explain(query: string, params?: any[]): Promise<any[]> {
|
||||
const explainQuery = `EXPLAIN (ANALYZE, BUFFERS, FORMAT JSON) ${query}`;
|
||||
const result = await this.query(explainQuery, params);
|
||||
return result.rows[0]['QUERY PLAN'];
|
||||
}
|
||||
|
||||
/**
|
||||
* Get database statistics
|
||||
*/
|
||||
async getStats(): Promise<any> {
|
||||
const result = await this.query(`
|
||||
SELECT
|
||||
(SELECT count(*) FROM pg_stat_activity WHERE state = 'active') as active_connections,
|
||||
(SELECT count(*) FROM pg_stat_activity WHERE state = 'idle') as idle_connections,
|
||||
(SELECT setting FROM pg_settings WHERE name = 'max_connections') as max_connections,
|
||||
pg_size_pretty(pg_database_size(current_database())) as database_size
|
||||
`);
|
||||
return result.rows[0];
|
||||
}
|
||||
|
||||
/**
|
||||
* Check if client is connected
|
||||
*/
|
||||
get connected(): boolean {
|
||||
return this.isConnected && !!this.pool;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the underlying connection pool
|
||||
*/
|
||||
get connectionPool(): Pool | null {
|
||||
return this.pool;
|
||||
}
|
||||
|
||||
private buildConfig(config?: Partial<PostgreSQLClientConfig>): PostgreSQLClientConfig {
|
||||
return {
|
||||
host: config?.host || postgresConfig.POSTGRES_HOST,
|
||||
port: config?.port || postgresConfig.POSTGRES_PORT,
|
||||
database: config?.database || postgresConfig.POSTGRES_DATABASE,
|
||||
username: config?.username || postgresConfig.POSTGRES_USERNAME,
|
||||
password: config?.password || postgresConfig.POSTGRES_PASSWORD,
|
||||
poolSettings: {
|
||||
min: postgresConfig.POSTGRES_POOL_MIN,
|
||||
max: postgresConfig.POSTGRES_POOL_MAX,
|
||||
idleTimeoutMillis: postgresConfig.POSTGRES_POOL_IDLE_TIMEOUT,
|
||||
...config?.poolSettings,
|
||||
},
|
||||
ssl: {
|
||||
enabled: postgresConfig.POSTGRES_SSL,
|
||||
rejectUnauthorized: postgresConfig.POSTGRES_SSL_REJECT_UNAUTHORIZED,
|
||||
...config?.ssl,
|
||||
},
|
||||
timeouts: {
|
||||
query: postgresConfig.POSTGRES_QUERY_TIMEOUT,
|
||||
connection: postgresConfig.POSTGRES_CONNECTION_TIMEOUT,
|
||||
statement: postgresConfig.POSTGRES_STATEMENT_TIMEOUT,
|
||||
lock: postgresConfig.POSTGRES_LOCK_TIMEOUT,
|
||||
idleInTransaction: postgresConfig.POSTGRES_IDLE_IN_TRANSACTION_SESSION_TIMEOUT,
|
||||
...config?.timeouts,
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
private buildPoolConfig(): any {
|
||||
return {
|
||||
host: this.config.host,
|
||||
port: this.config.port,
|
||||
database: this.config.database,
|
||||
user: this.config.username,
|
||||
password: this.config.password,
|
||||
min: this.config.poolSettings?.min,
|
||||
max: this.config.poolSettings?.max,
|
||||
idleTimeoutMillis: this.config.poolSettings?.idleTimeoutMillis,
|
||||
connectionTimeoutMillis: this.config.timeouts?.connection,
|
||||
query_timeout: this.config.timeouts?.query,
|
||||
statement_timeout: this.config.timeouts?.statement,
|
||||
lock_timeout: this.config.timeouts?.lock,
|
||||
idle_in_transaction_session_timeout: this.config.timeouts?.idleInTransaction,
|
||||
ssl: this.config.ssl?.enabled
|
||||
? {
|
||||
rejectUnauthorized: this.config.ssl.rejectUnauthorized,
|
||||
}
|
||||
: false,
|
||||
};
|
||||
}
|
||||
|
||||
private setupErrorHandlers(): void {
|
||||
if (!this.pool) return;
|
||||
|
||||
this.pool.on('error', error => {
|
||||
this.logger.error('PostgreSQL pool error:', error);
|
||||
});
|
||||
|
||||
this.pool.on('connect', () => {
|
||||
this.logger.debug('New PostgreSQL client connected');
|
||||
});
|
||||
|
||||
this.pool.on('remove', () => {
|
||||
this.logger.debug('PostgreSQL client removed from pool');
|
||||
});
|
||||
}
|
||||
|
||||
private delay(ms: number): Promise<void> {
|
||||
return new Promise(resolve => setTimeout(resolve, ms));
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,64 +1,64 @@
|
|||
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;
|
||||
}
|
||||
}
|
||||
import { postgresConfig } from '@stock-bot/config';
|
||||
import { PostgreSQLClient } from './client';
|
||||
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;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,142 +1,142 @@
|
|||
import { getLogger } from '@stock-bot/logger';
|
||||
import type { PostgreSQLClient } from './client';
|
||||
import type { PostgreSQLHealthCheck, PostgreSQLHealthStatus, PostgreSQLMetrics } from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Health Monitor
|
||||
*
|
||||
* Monitors PostgreSQL connection health and provides metrics
|
||||
*/
|
||||
export class PostgreSQLHealthMonitor {
|
||||
private readonly client: PostgreSQLClient;
|
||||
private readonly logger: ReturnType<typeof getLogger>;
|
||||
private healthCheckInterval: NodeJS.Timeout | null = null;
|
||||
private metrics: PostgreSQLMetrics;
|
||||
private lastHealthCheck: PostgreSQLHealthCheck | null = null;
|
||||
|
||||
constructor(client: PostgreSQLClient) {
|
||||
this.client = client;
|
||||
this.logger = getLogger('postgres-health-monitor');
|
||||
this.metrics = {
|
||||
queriesPerSecond: 0,
|
||||
averageQueryTime: 0,
|
||||
errorRate: 0,
|
||||
connectionPoolUtilization: 0,
|
||||
slowQueries: 0
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Start health monitoring
|
||||
*/
|
||||
start(intervalMs: number = 30000): void {
|
||||
if (this.healthCheckInterval) {
|
||||
this.stop();
|
||||
}
|
||||
|
||||
this.logger.info(`Starting PostgreSQL health monitoring (interval: ${intervalMs}ms)`);
|
||||
|
||||
this.healthCheckInterval = setInterval(async () => {
|
||||
try {
|
||||
await this.performHealthCheck();
|
||||
} catch (error) {
|
||||
this.logger.error('Health check failed:', error);
|
||||
}
|
||||
}, intervalMs);
|
||||
|
||||
// Perform initial health check
|
||||
this.performHealthCheck().catch(error => {
|
||||
this.logger.error('Initial health check failed:', error);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Stop health monitoring
|
||||
*/
|
||||
stop(): void {
|
||||
if (this.healthCheckInterval) {
|
||||
clearInterval(this.healthCheckInterval);
|
||||
this.healthCheckInterval = null;
|
||||
this.logger.info('Stopped PostgreSQL health monitoring');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get current health status
|
||||
*/
|
||||
async getHealth(): Promise<PostgreSQLHealthCheck> {
|
||||
if (!this.lastHealthCheck) {
|
||||
await this.performHealthCheck();
|
||||
}
|
||||
return this.lastHealthCheck!;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get current metrics
|
||||
*/
|
||||
getMetrics(): PostgreSQLMetrics {
|
||||
return { ...this.metrics };
|
||||
}
|
||||
|
||||
/**
|
||||
* Perform a health check
|
||||
*/
|
||||
private async performHealthCheck(): Promise<void> {
|
||||
const startTime = Date.now();
|
||||
const errors: string[] = [];
|
||||
let status: PostgreSQLHealthStatus = 'healthy';
|
||||
|
||||
try {
|
||||
if (!this.client.connected) {
|
||||
errors.push('PostgreSQL client not connected');
|
||||
status = 'unhealthy';
|
||||
} else {
|
||||
// Test basic connectivity
|
||||
await this.client.query('SELECT 1');
|
||||
|
||||
// Get connection stats
|
||||
const stats = await this.client.getStats();
|
||||
|
||||
// Check connection pool utilization
|
||||
const utilization = parseInt(stats.active_connections) / parseInt(stats.max_connections);
|
||||
if (utilization > 0.8) {
|
||||
errors.push('High connection pool utilization');
|
||||
status = status === 'healthy' ? 'degraded' : status;
|
||||
}
|
||||
|
||||
// Check for high latency
|
||||
const latency = Date.now() - startTime;
|
||||
if (latency > 1000) {
|
||||
errors.push(`High latency: ${latency}ms`);
|
||||
status = status === 'healthy' ? 'degraded' : status;
|
||||
}
|
||||
|
||||
this.metrics.connectionPoolUtilization = utilization;
|
||||
}
|
||||
} catch (error) {
|
||||
errors.push(`Health check failed: ${(error as Error).message}`);
|
||||
status = 'unhealthy';
|
||||
}
|
||||
|
||||
const latency = Date.now() - startTime;
|
||||
|
||||
this.lastHealthCheck = {
|
||||
status,
|
||||
timestamp: new Date(),
|
||||
latency,
|
||||
connections: {
|
||||
active: 1,
|
||||
idle: 9,
|
||||
total: 10
|
||||
},
|
||||
errors: errors.length > 0 ? errors : undefined
|
||||
};
|
||||
|
||||
// Log health status changes
|
||||
if (status !== 'healthy') {
|
||||
this.logger.warn(`PostgreSQL health status: ${status}`, { errors, latency });
|
||||
} else {
|
||||
this.logger.debug(`PostgreSQL health check passed (${latency}ms)`);
|
||||
}
|
||||
}
|
||||
}
|
||||
import { getLogger } from '@stock-bot/logger';
|
||||
import type { PostgreSQLClient } from './client';
|
||||
import type { PostgreSQLHealthCheck, PostgreSQLHealthStatus, PostgreSQLMetrics } from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Health Monitor
|
||||
*
|
||||
* Monitors PostgreSQL connection health and provides metrics
|
||||
*/
|
||||
export class PostgreSQLHealthMonitor {
|
||||
private readonly client: PostgreSQLClient;
|
||||
private readonly logger: ReturnType<typeof getLogger>;
|
||||
private healthCheckInterval: NodeJS.Timeout | null = null;
|
||||
private metrics: PostgreSQLMetrics;
|
||||
private lastHealthCheck: PostgreSQLHealthCheck | null = null;
|
||||
|
||||
constructor(client: PostgreSQLClient) {
|
||||
this.client = client;
|
||||
this.logger = getLogger('postgres-health-monitor');
|
||||
this.metrics = {
|
||||
queriesPerSecond: 0,
|
||||
averageQueryTime: 0,
|
||||
errorRate: 0,
|
||||
connectionPoolUtilization: 0,
|
||||
slowQueries: 0,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Start health monitoring
|
||||
*/
|
||||
start(intervalMs: number = 30000): void {
|
||||
if (this.healthCheckInterval) {
|
||||
this.stop();
|
||||
}
|
||||
|
||||
this.logger.info(`Starting PostgreSQL health monitoring (interval: ${intervalMs}ms)`);
|
||||
|
||||
this.healthCheckInterval = setInterval(async () => {
|
||||
try {
|
||||
await this.performHealthCheck();
|
||||
} catch (error) {
|
||||
this.logger.error('Health check failed:', error);
|
||||
}
|
||||
}, intervalMs);
|
||||
|
||||
// Perform initial health check
|
||||
this.performHealthCheck().catch(error => {
|
||||
this.logger.error('Initial health check failed:', error);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Stop health monitoring
|
||||
*/
|
||||
stop(): void {
|
||||
if (this.healthCheckInterval) {
|
||||
clearInterval(this.healthCheckInterval);
|
||||
this.healthCheckInterval = null;
|
||||
this.logger.info('Stopped PostgreSQL health monitoring');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get current health status
|
||||
*/
|
||||
async getHealth(): Promise<PostgreSQLHealthCheck> {
|
||||
if (!this.lastHealthCheck) {
|
||||
await this.performHealthCheck();
|
||||
}
|
||||
return this.lastHealthCheck!;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get current metrics
|
||||
*/
|
||||
getMetrics(): PostgreSQLMetrics {
|
||||
return { ...this.metrics };
|
||||
}
|
||||
|
||||
/**
|
||||
* Perform a health check
|
||||
*/
|
||||
private async performHealthCheck(): Promise<void> {
|
||||
const startTime = Date.now();
|
||||
const errors: string[] = [];
|
||||
let status: PostgreSQLHealthStatus = 'healthy';
|
||||
|
||||
try {
|
||||
if (!this.client.connected) {
|
||||
errors.push('PostgreSQL client not connected');
|
||||
status = 'unhealthy';
|
||||
} else {
|
||||
// Test basic connectivity
|
||||
await this.client.query('SELECT 1');
|
||||
|
||||
// Get connection stats
|
||||
const stats = await this.client.getStats();
|
||||
|
||||
// Check connection pool utilization
|
||||
const utilization = parseInt(stats.active_connections) / parseInt(stats.max_connections);
|
||||
if (utilization > 0.8) {
|
||||
errors.push('High connection pool utilization');
|
||||
status = status === 'healthy' ? 'degraded' : status;
|
||||
}
|
||||
|
||||
// Check for high latency
|
||||
const latency = Date.now() - startTime;
|
||||
if (latency > 1000) {
|
||||
errors.push(`High latency: ${latency}ms`);
|
||||
status = status === 'healthy' ? 'degraded' : status;
|
||||
}
|
||||
|
||||
this.metrics.connectionPoolUtilization = utilization;
|
||||
}
|
||||
} catch (error) {
|
||||
errors.push(`Health check failed: ${(error as Error).message}`);
|
||||
status = 'unhealthy';
|
||||
}
|
||||
|
||||
const latency = Date.now() - startTime;
|
||||
|
||||
this.lastHealthCheck = {
|
||||
status,
|
||||
timestamp: new Date(),
|
||||
latency,
|
||||
connections: {
|
||||
active: 1,
|
||||
idle: 9,
|
||||
total: 10,
|
||||
},
|
||||
errors: errors.length > 0 ? errors : undefined,
|
||||
};
|
||||
|
||||
// Log health status changes
|
||||
if (status !== 'healthy') {
|
||||
this.logger.warn(`PostgreSQL health status: ${status}`, { errors, latency });
|
||||
} else {
|
||||
this.logger.debug(`PostgreSQL health check passed (${latency}ms)`);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,34 +1,34 @@
|
|||
/**
|
||||
* PostgreSQL Client Library for Stock Bot
|
||||
*
|
||||
* Provides type-safe PostgreSQL access for operational data,
|
||||
* transactions, and relational queries.
|
||||
*/
|
||||
|
||||
export { PostgreSQLClient } from './client';
|
||||
export { PostgreSQLHealthMonitor } from './health';
|
||||
export { PostgreSQLTransactionManager } from './transactions';
|
||||
export { PostgreSQLQueryBuilder } from './query-builder';
|
||||
// export { PostgreSQLMigrationManager } from './migrations'; // TODO: Implement migrations
|
||||
|
||||
// Types
|
||||
export type {
|
||||
PostgreSQLClientConfig,
|
||||
PostgreSQLConnectionOptions,
|
||||
PostgreSQLHealthStatus,
|
||||
PostgreSQLMetrics,
|
||||
QueryResult,
|
||||
TransactionCallback,
|
||||
SchemaNames,
|
||||
TableNames,
|
||||
Trade,
|
||||
Order,
|
||||
Position,
|
||||
Portfolio,
|
||||
Strategy,
|
||||
RiskLimit,
|
||||
AuditLog
|
||||
} from './types';
|
||||
|
||||
// Utils
|
||||
export { createPostgreSQLClient, getPostgreSQLClient } from './factory';
|
||||
/**
|
||||
* PostgreSQL Client Library for Stock Bot
|
||||
*
|
||||
* Provides type-safe PostgreSQL access for operational data,
|
||||
* transactions, and relational queries.
|
||||
*/
|
||||
|
||||
export { PostgreSQLClient } from './client';
|
||||
export { PostgreSQLHealthMonitor } from './health';
|
||||
export { PostgreSQLTransactionManager } from './transactions';
|
||||
export { PostgreSQLQueryBuilder } from './query-builder';
|
||||
// export { PostgreSQLMigrationManager } from './migrations'; // TODO: Implement migrations
|
||||
|
||||
// Types
|
||||
export type {
|
||||
PostgreSQLClientConfig,
|
||||
PostgreSQLConnectionOptions,
|
||||
PostgreSQLHealthStatus,
|
||||
PostgreSQLMetrics,
|
||||
QueryResult,
|
||||
TransactionCallback,
|
||||
SchemaNames,
|
||||
TableNames,
|
||||
Trade,
|
||||
Order,
|
||||
Position,
|
||||
Portfolio,
|
||||
Strategy,
|
||||
RiskLimit,
|
||||
AuditLog,
|
||||
} from './types';
|
||||
|
||||
// Utils
|
||||
export { createPostgreSQLClient, getPostgreSQLClient } from './factory';
|
||||
|
|
|
|||
|
|
@ -1,268 +1,270 @@
|
|||
import type { QueryResultRow } from 'pg';
|
||||
import type { PostgreSQLClient } from './client';
|
||||
import type { WhereCondition, JoinCondition, OrderByCondition, QueryResult } from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Query Builder
|
||||
*
|
||||
* Provides a fluent interface for building SQL queries
|
||||
*/
|
||||
export class PostgreSQLQueryBuilder {
|
||||
private queryType: 'SELECT' | 'INSERT' | 'UPDATE' | 'DELETE' | null = null;
|
||||
private selectColumns: string[] = [];
|
||||
private fromTable: string = '';
|
||||
private joins: JoinCondition[] = [];
|
||||
private whereConditions: WhereCondition[] = [];
|
||||
private groupByColumns: string[] = [];
|
||||
private havingConditions: WhereCondition[] = [];
|
||||
private orderByConditions: OrderByCondition[] = [];
|
||||
private limitCount: number | null = null;
|
||||
private offsetCount: number | null = null;
|
||||
private insertValues: Record<string, any> = {};
|
||||
private updateValues: Record<string, any> = {};
|
||||
|
||||
private readonly client: PostgreSQLClient;
|
||||
|
||||
constructor(client: PostgreSQLClient) {
|
||||
this.client = client;
|
||||
}
|
||||
|
||||
/**
|
||||
* SELECT statement
|
||||
*/
|
||||
select(columns: string | string[] = '*'): this {
|
||||
this.queryType = 'SELECT';
|
||||
this.selectColumns = Array.isArray(columns) ? columns : [columns];
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* FROM clause
|
||||
*/
|
||||
from(table: string): this {
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* JOIN clause
|
||||
*/
|
||||
join(table: string, on: string, type: 'INNER' | 'LEFT' | 'RIGHT' | 'FULL' = 'INNER'): this {
|
||||
this.joins.push({ type, table, on });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* WHERE clause
|
||||
*/
|
||||
where(column: string, operator: string, value?: any): this {
|
||||
this.whereConditions.push({ column, operator: operator as any, value });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* GROUP BY clause
|
||||
*/
|
||||
groupBy(columns: string | string[]): this {
|
||||
this.groupByColumns = Array.isArray(columns) ? columns : [columns];
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* ORDER BY clause
|
||||
*/
|
||||
orderBy(column: string, direction: 'ASC' | 'DESC' = 'ASC'): this {
|
||||
this.orderByConditions.push({ column, direction });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* LIMIT clause
|
||||
*/
|
||||
limit(count: number): this {
|
||||
this.limitCount = count;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* OFFSET clause
|
||||
*/
|
||||
offset(count: number): this {
|
||||
this.offsetCount = count;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* INSERT statement
|
||||
*/
|
||||
insert(table: string): this {
|
||||
this.queryType = 'INSERT';
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* VALUES for INSERT
|
||||
*/
|
||||
values(data: Record<string, any>): this {
|
||||
this.insertValues = data;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* UPDATE statement
|
||||
*/
|
||||
update(table: string): this {
|
||||
this.queryType = 'UPDATE';
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* SET for UPDATE
|
||||
*/
|
||||
set(data: Record<string, any>): this {
|
||||
this.updateValues = data;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* DELETE statement
|
||||
*/
|
||||
delete(table: string): this {
|
||||
this.queryType = 'DELETE';
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Build and execute the query
|
||||
*/
|
||||
async execute<T extends QueryResultRow = any>(): Promise<QueryResult<T>> {
|
||||
const { sql, params } = this.build();
|
||||
return await this.client.query<T>(sql, params);
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the SQL query
|
||||
*/
|
||||
build(): { sql: string; params: any[] } {
|
||||
const params: any[] = [];
|
||||
let sql = '';
|
||||
|
||||
switch (this.queryType) {
|
||||
case 'SELECT':
|
||||
sql = this.buildSelectQuery(params);
|
||||
break;
|
||||
case 'INSERT':
|
||||
sql = this.buildInsertQuery(params);
|
||||
break;
|
||||
case 'UPDATE':
|
||||
sql = this.buildUpdateQuery(params);
|
||||
break;
|
||||
case 'DELETE':
|
||||
sql = this.buildDeleteQuery(params);
|
||||
break;
|
||||
default:
|
||||
throw new Error('Query type not specified');
|
||||
}
|
||||
|
||||
return { sql, params };
|
||||
}
|
||||
|
||||
private buildSelectQuery(params: any[]): string {
|
||||
let sql = `SELECT ${this.selectColumns.join(', ')}`;
|
||||
|
||||
if (this.fromTable) {
|
||||
sql += ` FROM ${this.fromTable}`;
|
||||
}
|
||||
|
||||
// Add JOINs
|
||||
for (const join of this.joins) {
|
||||
sql += ` ${join.type} JOIN ${join.table} ON ${join.on}`;
|
||||
}
|
||||
|
||||
// Add WHERE
|
||||
if (this.whereConditions.length > 0) {
|
||||
sql += ' WHERE ' + this.buildWhereClause(this.whereConditions, params);
|
||||
}
|
||||
|
||||
// Add GROUP BY
|
||||
if (this.groupByColumns.length > 0) {
|
||||
sql += ` GROUP BY ${this.groupByColumns.join(', ')}`;
|
||||
}
|
||||
|
||||
// Add HAVING
|
||||
if (this.havingConditions.length > 0) {
|
||||
sql += ' HAVING ' + this.buildWhereClause(this.havingConditions, params);
|
||||
}
|
||||
|
||||
// Add ORDER BY
|
||||
if (this.orderByConditions.length > 0) {
|
||||
const orderBy = this.orderByConditions
|
||||
.map(order => `${order.column} ${order.direction}`)
|
||||
.join(', ');
|
||||
sql += ` ORDER BY ${orderBy}`;
|
||||
}
|
||||
|
||||
// Add LIMIT
|
||||
if (this.limitCount !== null) {
|
||||
sql += ` LIMIT $${params.length + 1}`;
|
||||
params.push(this.limitCount);
|
||||
}
|
||||
|
||||
// Add OFFSET
|
||||
if (this.offsetCount !== null) {
|
||||
sql += ` OFFSET $${params.length + 1}`;
|
||||
params.push(this.offsetCount);
|
||||
}
|
||||
|
||||
return sql;
|
||||
}
|
||||
|
||||
private buildInsertQuery(params: any[]): string {
|
||||
const columns = Object.keys(this.insertValues);
|
||||
const placeholders = columns.map((_, i) => `$${params.length + i + 1}`);
|
||||
|
||||
params.push(...Object.values(this.insertValues));
|
||||
|
||||
return `INSERT INTO ${this.fromTable} (${columns.join(', ')}) VALUES (${placeholders.join(', ')})`;
|
||||
}
|
||||
|
||||
private buildUpdateQuery(params: any[]): string {
|
||||
const sets = Object.keys(this.updateValues).map((key, i) => {
|
||||
return `${key} = $${params.length + i + 1}`;
|
||||
});
|
||||
|
||||
params.push(...Object.values(this.updateValues));
|
||||
|
||||
let sql = `UPDATE ${this.fromTable} SET ${sets.join(', ')}`;
|
||||
|
||||
if (this.whereConditions.length > 0) {
|
||||
sql += ' WHERE ' + this.buildWhereClause(this.whereConditions, params);
|
||||
}
|
||||
|
||||
return sql;
|
||||
}
|
||||
|
||||
private buildDeleteQuery(params: any[]): string {
|
||||
let sql = `DELETE FROM ${this.fromTable}`;
|
||||
|
||||
if (this.whereConditions.length > 0) {
|
||||
sql += ' WHERE ' + this.buildWhereClause(this.whereConditions, params);
|
||||
}
|
||||
|
||||
return sql;
|
||||
}
|
||||
|
||||
private buildWhereClause(conditions: WhereCondition[], params: any[]): string {
|
||||
return conditions.map(condition => {
|
||||
if (condition.operator === 'IS NULL' || condition.operator === 'IS NOT NULL') {
|
||||
return `${condition.column} ${condition.operator}`;
|
||||
} else {
|
||||
params.push(condition.value);
|
||||
return `${condition.column} ${condition.operator} $${params.length}`;
|
||||
}
|
||||
}).join(' AND ');
|
||||
}
|
||||
}
|
||||
import type { QueryResultRow } from 'pg';
|
||||
import type { PostgreSQLClient } from './client';
|
||||
import type { JoinCondition, OrderByCondition, QueryResult, WhereCondition } from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Query Builder
|
||||
*
|
||||
* Provides a fluent interface for building SQL queries
|
||||
*/
|
||||
export class PostgreSQLQueryBuilder {
|
||||
private queryType: 'SELECT' | 'INSERT' | 'UPDATE' | 'DELETE' | null = null;
|
||||
private selectColumns: string[] = [];
|
||||
private fromTable: string = '';
|
||||
private joins: JoinCondition[] = [];
|
||||
private whereConditions: WhereCondition[] = [];
|
||||
private groupByColumns: string[] = [];
|
||||
private havingConditions: WhereCondition[] = [];
|
||||
private orderByConditions: OrderByCondition[] = [];
|
||||
private limitCount: number | null = null;
|
||||
private offsetCount: number | null = null;
|
||||
private insertValues: Record<string, any> = {};
|
||||
private updateValues: Record<string, any> = {};
|
||||
|
||||
private readonly client: PostgreSQLClient;
|
||||
|
||||
constructor(client: PostgreSQLClient) {
|
||||
this.client = client;
|
||||
}
|
||||
|
||||
/**
|
||||
* SELECT statement
|
||||
*/
|
||||
select(columns: string | string[] = '*'): this {
|
||||
this.queryType = 'SELECT';
|
||||
this.selectColumns = Array.isArray(columns) ? columns : [columns];
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* FROM clause
|
||||
*/
|
||||
from(table: string): this {
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* JOIN clause
|
||||
*/
|
||||
join(table: string, on: string, type: 'INNER' | 'LEFT' | 'RIGHT' | 'FULL' = 'INNER'): this {
|
||||
this.joins.push({ type, table, on });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* WHERE clause
|
||||
*/
|
||||
where(column: string, operator: string, value?: any): this {
|
||||
this.whereConditions.push({ column, operator: operator as any, value });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* GROUP BY clause
|
||||
*/
|
||||
groupBy(columns: string | string[]): this {
|
||||
this.groupByColumns = Array.isArray(columns) ? columns : [columns];
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* ORDER BY clause
|
||||
*/
|
||||
orderBy(column: string, direction: 'ASC' | 'DESC' = 'ASC'): this {
|
||||
this.orderByConditions.push({ column, direction });
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* LIMIT clause
|
||||
*/
|
||||
limit(count: number): this {
|
||||
this.limitCount = count;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* OFFSET clause
|
||||
*/
|
||||
offset(count: number): this {
|
||||
this.offsetCount = count;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* INSERT statement
|
||||
*/
|
||||
insert(table: string): this {
|
||||
this.queryType = 'INSERT';
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* VALUES for INSERT
|
||||
*/
|
||||
values(data: Record<string, any>): this {
|
||||
this.insertValues = data;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* UPDATE statement
|
||||
*/
|
||||
update(table: string): this {
|
||||
this.queryType = 'UPDATE';
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* SET for UPDATE
|
||||
*/
|
||||
set(data: Record<string, any>): this {
|
||||
this.updateValues = data;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* DELETE statement
|
||||
*/
|
||||
delete(table: string): this {
|
||||
this.queryType = 'DELETE';
|
||||
this.fromTable = table;
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Build and execute the query
|
||||
*/
|
||||
async execute<T extends QueryResultRow = any>(): Promise<QueryResult<T>> {
|
||||
const { sql, params } = this.build();
|
||||
return await this.client.query<T>(sql, params);
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the SQL query
|
||||
*/
|
||||
build(): { sql: string; params: any[] } {
|
||||
const params: any[] = [];
|
||||
let sql = '';
|
||||
|
||||
switch (this.queryType) {
|
||||
case 'SELECT':
|
||||
sql = this.buildSelectQuery(params);
|
||||
break;
|
||||
case 'INSERT':
|
||||
sql = this.buildInsertQuery(params);
|
||||
break;
|
||||
case 'UPDATE':
|
||||
sql = this.buildUpdateQuery(params);
|
||||
break;
|
||||
case 'DELETE':
|
||||
sql = this.buildDeleteQuery(params);
|
||||
break;
|
||||
default:
|
||||
throw new Error('Query type not specified');
|
||||
}
|
||||
|
||||
return { sql, params };
|
||||
}
|
||||
|
||||
private buildSelectQuery(params: any[]): string {
|
||||
let sql = `SELECT ${this.selectColumns.join(', ')}`;
|
||||
|
||||
if (this.fromTable) {
|
||||
sql += ` FROM ${this.fromTable}`;
|
||||
}
|
||||
|
||||
// Add JOINs
|
||||
for (const join of this.joins) {
|
||||
sql += ` ${join.type} JOIN ${join.table} ON ${join.on}`;
|
||||
}
|
||||
|
||||
// Add WHERE
|
||||
if (this.whereConditions.length > 0) {
|
||||
sql += ' WHERE ' + this.buildWhereClause(this.whereConditions, params);
|
||||
}
|
||||
|
||||
// Add GROUP BY
|
||||
if (this.groupByColumns.length > 0) {
|
||||
sql += ` GROUP BY ${this.groupByColumns.join(', ')}`;
|
||||
}
|
||||
|
||||
// Add HAVING
|
||||
if (this.havingConditions.length > 0) {
|
||||
sql += ' HAVING ' + this.buildWhereClause(this.havingConditions, params);
|
||||
}
|
||||
|
||||
// Add ORDER BY
|
||||
if (this.orderByConditions.length > 0) {
|
||||
const orderBy = this.orderByConditions
|
||||
.map(order => `${order.column} ${order.direction}`)
|
||||
.join(', ');
|
||||
sql += ` ORDER BY ${orderBy}`;
|
||||
}
|
||||
|
||||
// Add LIMIT
|
||||
if (this.limitCount !== null) {
|
||||
sql += ` LIMIT $${params.length + 1}`;
|
||||
params.push(this.limitCount);
|
||||
}
|
||||
|
||||
// Add OFFSET
|
||||
if (this.offsetCount !== null) {
|
||||
sql += ` OFFSET $${params.length + 1}`;
|
||||
params.push(this.offsetCount);
|
||||
}
|
||||
|
||||
return sql;
|
||||
}
|
||||
|
||||
private buildInsertQuery(params: any[]): string {
|
||||
const columns = Object.keys(this.insertValues);
|
||||
const placeholders = columns.map((_, i) => `$${params.length + i + 1}`);
|
||||
|
||||
params.push(...Object.values(this.insertValues));
|
||||
|
||||
return `INSERT INTO ${this.fromTable} (${columns.join(', ')}) VALUES (${placeholders.join(', ')})`;
|
||||
}
|
||||
|
||||
private buildUpdateQuery(params: any[]): string {
|
||||
const sets = Object.keys(this.updateValues).map((key, i) => {
|
||||
return `${key} = $${params.length + i + 1}`;
|
||||
});
|
||||
|
||||
params.push(...Object.values(this.updateValues));
|
||||
|
||||
let sql = `UPDATE ${this.fromTable} SET ${sets.join(', ')}`;
|
||||
|
||||
if (this.whereConditions.length > 0) {
|
||||
sql += ' WHERE ' + this.buildWhereClause(this.whereConditions, params);
|
||||
}
|
||||
|
||||
return sql;
|
||||
}
|
||||
|
||||
private buildDeleteQuery(params: any[]): string {
|
||||
let sql = `DELETE FROM ${this.fromTable}`;
|
||||
|
||||
if (this.whereConditions.length > 0) {
|
||||
sql += ' WHERE ' + this.buildWhereClause(this.whereConditions, params);
|
||||
}
|
||||
|
||||
return sql;
|
||||
}
|
||||
|
||||
private buildWhereClause(conditions: WhereCondition[], params: any[]): string {
|
||||
return conditions
|
||||
.map(condition => {
|
||||
if (condition.operator === 'IS NULL' || condition.operator === 'IS NOT NULL') {
|
||||
return `${condition.column} ${condition.operator}`;
|
||||
} else {
|
||||
params.push(condition.value);
|
||||
return `${condition.column} ${condition.operator} $${params.length}`;
|
||||
}
|
||||
})
|
||||
.join(' AND ');
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,57 +1,56 @@
|
|||
import { PoolClient } from 'pg';
|
||||
import { getLogger } from '@stock-bot/logger';
|
||||
import type { PostgreSQLClient } from './client';
|
||||
import type { TransactionCallback } from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Transaction Manager
|
||||
*
|
||||
* Provides transaction support for multi-statement operations
|
||||
*/
|
||||
export class PostgreSQLTransactionManager {
|
||||
private readonly client: PostgreSQLClient;
|
||||
private readonly logger: ReturnType<typeof getLogger>;
|
||||
|
||||
constructor(client: PostgreSQLClient) {
|
||||
this.client = client;
|
||||
this.logger = getLogger('postgres-transaction-manager');
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute operations within a transaction
|
||||
*/
|
||||
async execute<T>(callback: TransactionCallback<T>): Promise<T> {
|
||||
const pool = this.client.connectionPool;
|
||||
if (!pool) {
|
||||
throw new Error('PostgreSQL client not connected');
|
||||
}
|
||||
|
||||
const client = await pool.connect();
|
||||
|
||||
try {
|
||||
this.logger.debug('Starting PostgreSQL transaction');
|
||||
|
||||
await client.query('BEGIN');
|
||||
|
||||
const result = await callback(client);
|
||||
|
||||
await client.query('COMMIT');
|
||||
|
||||
this.logger.debug('PostgreSQL transaction committed successfully');
|
||||
return result;
|
||||
|
||||
} catch (error) {
|
||||
this.logger.error('PostgreSQL transaction failed, rolling back:', error);
|
||||
|
||||
try {
|
||||
await client.query('ROLLBACK');
|
||||
} catch (rollbackError) {
|
||||
this.logger.error('Failed to rollback transaction:', rollbackError);
|
||||
}
|
||||
|
||||
throw error;
|
||||
} finally {
|
||||
client.release();
|
||||
}
|
||||
}
|
||||
}
|
||||
import { PoolClient } from 'pg';
|
||||
import { getLogger } from '@stock-bot/logger';
|
||||
import type { PostgreSQLClient } from './client';
|
||||
import type { TransactionCallback } from './types';
|
||||
|
||||
/**
|
||||
* PostgreSQL Transaction Manager
|
||||
*
|
||||
* Provides transaction support for multi-statement operations
|
||||
*/
|
||||
export class PostgreSQLTransactionManager {
|
||||
private readonly client: PostgreSQLClient;
|
||||
private readonly logger: ReturnType<typeof getLogger>;
|
||||
|
||||
constructor(client: PostgreSQLClient) {
|
||||
this.client = client;
|
||||
this.logger = getLogger('postgres-transaction-manager');
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute operations within a transaction
|
||||
*/
|
||||
async execute<T>(callback: TransactionCallback<T>): Promise<T> {
|
||||
const pool = this.client.connectionPool;
|
||||
if (!pool) {
|
||||
throw new Error('PostgreSQL client not connected');
|
||||
}
|
||||
|
||||
const client = await pool.connect();
|
||||
|
||||
try {
|
||||
this.logger.debug('Starting PostgreSQL transaction');
|
||||
|
||||
await client.query('BEGIN');
|
||||
|
||||
const result = await callback(client);
|
||||
|
||||
await client.query('COMMIT');
|
||||
|
||||
this.logger.debug('PostgreSQL transaction committed successfully');
|
||||
return result;
|
||||
} catch (error) {
|
||||
this.logger.error('PostgreSQL transaction failed, rolling back:', error);
|
||||
|
||||
try {
|
||||
await client.query('ROLLBACK');
|
||||
} catch (rollbackError) {
|
||||
this.logger.error('Failed to rollback transaction:', rollbackError);
|
||||
}
|
||||
|
||||
throw error;
|
||||
} finally {
|
||||
client.release();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,206 +1,218 @@
|
|||
import type { Pool, PoolClient, QueryResult as PgQueryResult, QueryResultRow } from 'pg';
|
||||
|
||||
/**
|
||||
* PostgreSQL Client Configuration
|
||||
*/
|
||||
export interface PostgreSQLClientConfig {
|
||||
host: string;
|
||||
port: number;
|
||||
database: string;
|
||||
username: string;
|
||||
password: string;
|
||||
poolSettings?: {
|
||||
min: number;
|
||||
max: number;
|
||||
idleTimeoutMillis: number;
|
||||
};
|
||||
ssl?: {
|
||||
enabled: boolean;
|
||||
rejectUnauthorized: boolean;
|
||||
};
|
||||
timeouts?: {
|
||||
query: number;
|
||||
connection: number;
|
||||
statement: number;
|
||||
lock: number;
|
||||
idleInTransaction: number;
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* PostgreSQL Connection Options
|
||||
*/
|
||||
export interface PostgreSQLConnectionOptions {
|
||||
retryAttempts?: number;
|
||||
retryDelay?: number;
|
||||
healthCheckInterval?: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Health Status Types
|
||||
*/
|
||||
export type PostgreSQLHealthStatus = 'healthy' | 'degraded' | 'unhealthy';
|
||||
|
||||
export interface PostgreSQLHealthCheck {
|
||||
status: PostgreSQLHealthStatus;
|
||||
timestamp: Date;
|
||||
latency: number;
|
||||
connections: {
|
||||
active: number;
|
||||
idle: number;
|
||||
total: number;
|
||||
};
|
||||
errors?: string[];
|
||||
}
|
||||
|
||||
export interface PostgreSQLMetrics {
|
||||
queriesPerSecond: number;
|
||||
averageQueryTime: number;
|
||||
errorRate: number;
|
||||
connectionPoolUtilization: number;
|
||||
slowQueries: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Query Result Types
|
||||
*/
|
||||
export interface QueryResult<T extends QueryResultRow = any> extends PgQueryResult<T> {
|
||||
executionTime?: number;
|
||||
}
|
||||
|
||||
export type TransactionCallback<T> = (client: PoolClient) => Promise<T>;
|
||||
|
||||
/**
|
||||
* Schema and Table Names
|
||||
*/
|
||||
export type SchemaNames = 'trading' | 'strategy' | 'risk' | 'audit';
|
||||
|
||||
export type TableNames =
|
||||
| 'trades'
|
||||
| 'orders'
|
||||
| 'positions'
|
||||
| 'portfolios'
|
||||
| 'strategies'
|
||||
| 'risk_limits'
|
||||
| 'audit_logs'
|
||||
| 'users'
|
||||
| 'accounts'
|
||||
| 'symbols'
|
||||
| 'exchanges';
|
||||
|
||||
/**
|
||||
* Trading Domain Types
|
||||
*/
|
||||
export interface Trade {
|
||||
id: string;
|
||||
order_id: string;
|
||||
symbol: string;
|
||||
side: 'buy' | 'sell';
|
||||
quantity: number;
|
||||
price: number;
|
||||
executed_at: Date;
|
||||
commission: number;
|
||||
fees: number;
|
||||
portfolio_id: string;
|
||||
strategy_id?: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface Order {
|
||||
id: string;
|
||||
symbol: string;
|
||||
side: 'buy' | 'sell';
|
||||
type: 'market' | 'limit' | 'stop' | 'stop_limit';
|
||||
quantity: number;
|
||||
price?: number;
|
||||
stop_price?: number;
|
||||
status: 'pending' | 'filled' | 'cancelled' | 'rejected';
|
||||
portfolio_id: string;
|
||||
strategy_id?: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
expires_at?: Date;
|
||||
}
|
||||
|
||||
export interface Position {
|
||||
id: string;
|
||||
symbol: string;
|
||||
quantity: number;
|
||||
average_cost: number;
|
||||
market_value: number;
|
||||
unrealized_pnl: number;
|
||||
realized_pnl: number;
|
||||
portfolio_id: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface Portfolio {
|
||||
id: string;
|
||||
name: string;
|
||||
cash_balance: number;
|
||||
total_value: number;
|
||||
unrealized_pnl: number;
|
||||
realized_pnl: number;
|
||||
user_id: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface Strategy {
|
||||
id: string;
|
||||
name: string;
|
||||
description: string;
|
||||
parameters: Record<string, any>;
|
||||
status: 'active' | 'inactive' | 'paused';
|
||||
performance_metrics: Record<string, number>;
|
||||
portfolio_id: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface RiskLimit {
|
||||
id: string;
|
||||
type: 'position_size' | 'daily_loss' | 'max_drawdown' | 'concentration';
|
||||
value: number;
|
||||
threshold: number;
|
||||
status: 'active' | 'breached' | 'disabled';
|
||||
portfolio_id?: string;
|
||||
strategy_id?: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface AuditLog {
|
||||
id: string;
|
||||
action: string;
|
||||
entity_type: string;
|
||||
entity_id: string;
|
||||
old_values?: Record<string, any>;
|
||||
new_values?: Record<string, any>;
|
||||
user_id?: string;
|
||||
ip_address?: string;
|
||||
user_agent?: string;
|
||||
timestamp: Date;
|
||||
}
|
||||
|
||||
/**
|
||||
* Query Builder Types
|
||||
*/
|
||||
export interface WhereCondition {
|
||||
column: string;
|
||||
operator: '=' | '!=' | '>' | '<' | '>=' | '<=' | 'IN' | 'NOT IN' | 'LIKE' | 'ILIKE' | 'IS NULL' | 'IS NOT NULL';
|
||||
value?: any;
|
||||
}
|
||||
|
||||
export interface JoinCondition {
|
||||
type: 'INNER' | 'LEFT' | 'RIGHT' | 'FULL';
|
||||
table: string;
|
||||
on: string;
|
||||
}
|
||||
|
||||
export interface OrderByCondition {
|
||||
column: string;
|
||||
direction: 'ASC' | 'DESC';
|
||||
}
|
||||
import type { QueryResult as PgQueryResult, Pool, PoolClient, QueryResultRow } from 'pg';
|
||||
|
||||
/**
|
||||
* PostgreSQL Client Configuration
|
||||
*/
|
||||
export interface PostgreSQLClientConfig {
|
||||
host: string;
|
||||
port: number;
|
||||
database: string;
|
||||
username: string;
|
||||
password: string;
|
||||
poolSettings?: {
|
||||
min: number;
|
||||
max: number;
|
||||
idleTimeoutMillis: number;
|
||||
};
|
||||
ssl?: {
|
||||
enabled: boolean;
|
||||
rejectUnauthorized: boolean;
|
||||
};
|
||||
timeouts?: {
|
||||
query: number;
|
||||
connection: number;
|
||||
statement: number;
|
||||
lock: number;
|
||||
idleInTransaction: number;
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* PostgreSQL Connection Options
|
||||
*/
|
||||
export interface PostgreSQLConnectionOptions {
|
||||
retryAttempts?: number;
|
||||
retryDelay?: number;
|
||||
healthCheckInterval?: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Health Status Types
|
||||
*/
|
||||
export type PostgreSQLHealthStatus = 'healthy' | 'degraded' | 'unhealthy';
|
||||
|
||||
export interface PostgreSQLHealthCheck {
|
||||
status: PostgreSQLHealthStatus;
|
||||
timestamp: Date;
|
||||
latency: number;
|
||||
connections: {
|
||||
active: number;
|
||||
idle: number;
|
||||
total: number;
|
||||
};
|
||||
errors?: string[];
|
||||
}
|
||||
|
||||
export interface PostgreSQLMetrics {
|
||||
queriesPerSecond: number;
|
||||
averageQueryTime: number;
|
||||
errorRate: number;
|
||||
connectionPoolUtilization: number;
|
||||
slowQueries: number;
|
||||
}
|
||||
|
||||
/**
|
||||
* Query Result Types
|
||||
*/
|
||||
export interface QueryResult<T extends QueryResultRow = any> extends PgQueryResult<T> {
|
||||
executionTime?: number;
|
||||
}
|
||||
|
||||
export type TransactionCallback<T> = (client: PoolClient) => Promise<T>;
|
||||
|
||||
/**
|
||||
* Schema and Table Names
|
||||
*/
|
||||
export type SchemaNames = 'trading' | 'strategy' | 'risk' | 'audit';
|
||||
|
||||
export type TableNames =
|
||||
| 'trades'
|
||||
| 'orders'
|
||||
| 'positions'
|
||||
| 'portfolios'
|
||||
| 'strategies'
|
||||
| 'risk_limits'
|
||||
| 'audit_logs'
|
||||
| 'users'
|
||||
| 'accounts'
|
||||
| 'symbols'
|
||||
| 'exchanges';
|
||||
|
||||
/**
|
||||
* Trading Domain Types
|
||||
*/
|
||||
export interface Trade {
|
||||
id: string;
|
||||
order_id: string;
|
||||
symbol: string;
|
||||
side: 'buy' | 'sell';
|
||||
quantity: number;
|
||||
price: number;
|
||||
executed_at: Date;
|
||||
commission: number;
|
||||
fees: number;
|
||||
portfolio_id: string;
|
||||
strategy_id?: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface Order {
|
||||
id: string;
|
||||
symbol: string;
|
||||
side: 'buy' | 'sell';
|
||||
type: 'market' | 'limit' | 'stop' | 'stop_limit';
|
||||
quantity: number;
|
||||
price?: number;
|
||||
stop_price?: number;
|
||||
status: 'pending' | 'filled' | 'cancelled' | 'rejected';
|
||||
portfolio_id: string;
|
||||
strategy_id?: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
expires_at?: Date;
|
||||
}
|
||||
|
||||
export interface Position {
|
||||
id: string;
|
||||
symbol: string;
|
||||
quantity: number;
|
||||
average_cost: number;
|
||||
market_value: number;
|
||||
unrealized_pnl: number;
|
||||
realized_pnl: number;
|
||||
portfolio_id: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface Portfolio {
|
||||
id: string;
|
||||
name: string;
|
||||
cash_balance: number;
|
||||
total_value: number;
|
||||
unrealized_pnl: number;
|
||||
realized_pnl: number;
|
||||
user_id: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface Strategy {
|
||||
id: string;
|
||||
name: string;
|
||||
description: string;
|
||||
parameters: Record<string, any>;
|
||||
status: 'active' | 'inactive' | 'paused';
|
||||
performance_metrics: Record<string, number>;
|
||||
portfolio_id: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface RiskLimit {
|
||||
id: string;
|
||||
type: 'position_size' | 'daily_loss' | 'max_drawdown' | 'concentration';
|
||||
value: number;
|
||||
threshold: number;
|
||||
status: 'active' | 'breached' | 'disabled';
|
||||
portfolio_id?: string;
|
||||
strategy_id?: string;
|
||||
created_at: Date;
|
||||
updated_at: Date;
|
||||
}
|
||||
|
||||
export interface AuditLog {
|
||||
id: string;
|
||||
action: string;
|
||||
entity_type: string;
|
||||
entity_id: string;
|
||||
old_values?: Record<string, any>;
|
||||
new_values?: Record<string, any>;
|
||||
user_id?: string;
|
||||
ip_address?: string;
|
||||
user_agent?: string;
|
||||
timestamp: Date;
|
||||
}
|
||||
|
||||
/**
|
||||
* Query Builder Types
|
||||
*/
|
||||
export interface WhereCondition {
|
||||
column: string;
|
||||
operator:
|
||||
| '='
|
||||
| '!='
|
||||
| '>'
|
||||
| '<'
|
||||
| '>='
|
||||
| '<='
|
||||
| 'IN'
|
||||
| 'NOT IN'
|
||||
| 'LIKE'
|
||||
| 'ILIKE'
|
||||
| 'IS NULL'
|
||||
| 'IS NOT NULL';
|
||||
value?: any;
|
||||
}
|
||||
|
||||
export interface JoinCondition {
|
||||
type: 'INNER' | 'LEFT' | 'RIGHT' | 'FULL';
|
||||
table: string;
|
||||
on: string;
|
||||
}
|
||||
|
||||
export interface OrderByCondition {
|
||||
column: string;
|
||||
direction: 'ASC' | 'DESC';
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue