88 lines
3.6 KiB
TypeScript
88 lines
3.6 KiB
TypeScript
import dotenv from 'dotenv';
|
|
dotenv.config();
|
|
|
|
import WorkflowOrchestrator from './workflows/WorkflowOrchestrator';
|
|
import rDramaSession from './rdrama/session/SessionManager';
|
|
import redditSession from './reddit/session/SessionManager';
|
|
import { DatabaseInitializer } from './db/initializeDatabase';
|
|
import RedisSessionManager from './redis/session/SessionManager';
|
|
import { Comment } from './rdrama/models/Comment';
|
|
|
|
// Import other necessary services or configurations
|
|
|
|
async function startApplication() {
|
|
console.log('Database Start')
|
|
const databaseInitializer = DatabaseInitializer.getInstance();
|
|
const db = await databaseInitializer.getDbInstance()
|
|
if (!db) {
|
|
throw new Error('Failed to initialize the database.');
|
|
}
|
|
|
|
console.log('RDrama Session Start')
|
|
// Initialize SessionManager or other global configurations
|
|
const rDramaSessionManager = rDramaSession.getInstance();
|
|
if (!process.env.RDRAMA_API_KEY) {
|
|
throw new Error('RDRAMA_API_KEY is undefined. Please set this environment variable.');
|
|
}
|
|
rDramaSessionManager.setAuthorizationToken(process.env.RDRAMA_API_KEY);
|
|
|
|
console.log('Redis Start');
|
|
const redisManager = RedisSessionManager.getInstance();
|
|
redisManager.client.subscribe('newCommentsChannel', (error, count) => {
|
|
if (error) throw new Error('Failed to subscribe to Redis channel.');
|
|
console.log(`Subscribed to ${'newCommentsChannel'}. Listening for new comments.`);
|
|
});
|
|
|
|
console.log('Reddit Session Start')
|
|
const redditSessionManager = await redditSession.getInstance()
|
|
|
|
const workflowOrchestrator = new WorkflowOrchestrator();
|
|
|
|
redisManager.client.on('message', async (channel, message) => {
|
|
if (channel === 'newCommentsChannel') {
|
|
const comment: Comment = JSON.parse(message)
|
|
console.log(`New comment ${comment.id} received.`);
|
|
await workflowOrchestrator.executeWorkflow(comment);
|
|
}
|
|
});
|
|
|
|
setupProcessListeners(redisManager, redditSessionManager, rDramaSessionManager);
|
|
}
|
|
|
|
|
|
function setupProcessListeners(redisManager: RedisSessionManager, redditSessionManager: redditSession, rDramaSessionManager: rDramaSession) {
|
|
process.on('SIGINT', async () => {
|
|
console.log('SIGINT received. Shutting down gracefully...');
|
|
await shutdownServices(redisManager, redditSessionManager, rDramaSessionManager);
|
|
});
|
|
|
|
process.on('SIGTERM', async () => {
|
|
console.log('SIGTERM received. Shutting down gracefully...');
|
|
await shutdownServices(redisManager, redditSessionManager, rDramaSessionManager);
|
|
});
|
|
|
|
process.on('uncaughtException', async (error) => {
|
|
console.error('Uncaught Exception:', error);
|
|
await shutdownServices(redisManager, redditSessionManager, rDramaSessionManager);
|
|
process.exit(1); // Exit with error
|
|
});
|
|
|
|
process.on('unhandledRejection', async (reason, promise) => {
|
|
console.error('Unhandled Rejection at:', promise, 'reason:', reason);
|
|
await shutdownServices(redisManager, redditSessionManager, rDramaSessionManager);
|
|
process.exit(1); // Exit with error
|
|
});
|
|
}
|
|
|
|
async function shutdownServices(redisManager: RedisSessionManager, redditSessionManager: redditSession, rDramaSessionManager: rDramaSession) {
|
|
await redisManager.shutdown();
|
|
await redditSessionManager.shutdown();
|
|
await rDramaSessionManager.shutdown();
|
|
console.log('Services shut down successfully.');
|
|
process.exit(0); // Exit cleanly
|
|
}
|
|
|
|
startApplication()
|
|
.then(() => console.log('Application started successfully.'))
|
|
.catch((error) => console.error('Application failed to start:', error));
|