Added Redis subscription, event, and graceful shutdown
parent
7ef10dbb0f
commit
b7a01becb7
67
src/index.ts
67
src/index.ts
|
@ -5,8 +5,9 @@ import WorkflowOrchestrator from './workflows/WorkflowOrchestrator';
|
||||||
import rDramaSession from './rdrama/session/SessionManager';
|
import rDramaSession from './rdrama/session/SessionManager';
|
||||||
import redditSession from './reddit/session/SessionManager';
|
import redditSession from './reddit/session/SessionManager';
|
||||||
import { DatabaseInitializer } from './db/initializeDatabase';
|
import { DatabaseInitializer } from './db/initializeDatabase';
|
||||||
import { DatabaseService } from './db/services/Database';
|
import RedisSessionManager from './redis/session/SessionManager';
|
||||||
import { DatabaseMaintenanceService } from './db/services/DatabaseMaintenance';
|
import { Comment } from './rdrama/models/Comment';
|
||||||
|
|
||||||
// Import other necessary services or configurations
|
// Import other necessary services or configurations
|
||||||
|
|
||||||
async function startApplication() {
|
async function startApplication() {
|
||||||
|
@ -16,12 +17,6 @@ async function startApplication() {
|
||||||
if (!db) {
|
if (!db) {
|
||||||
throw new Error('Failed to initialize the database.');
|
throw new Error('Failed to initialize the database.');
|
||||||
}
|
}
|
||||||
const canSend = await DatabaseService.canSendNotification();
|
|
||||||
const coolDownHours = process.env.NOTIFICATION_COOLDOWN_HOURS
|
|
||||||
if (!canSend) {
|
|
||||||
console.log(`Last Message Sent less than ${coolDownHours ? coolDownHours : 4} hours ago. Set NOTIFICATION_COOLDOWN_HOURS to change this`)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
console.log('RDrama Session Start')
|
console.log('RDrama Session Start')
|
||||||
// Initialize SessionManager or other global configurations
|
// Initialize SessionManager or other global configurations
|
||||||
|
@ -31,18 +26,60 @@ async function startApplication() {
|
||||||
}
|
}
|
||||||
rDramaSessionManager.setAuthorizationToken(process.env.RDRAMA_API_KEY);
|
rDramaSessionManager.setAuthorizationToken(process.env.RDRAMA_API_KEY);
|
||||||
|
|
||||||
console.log('Database Maintenance Start')
|
console.log('Redis Start');
|
||||||
const databaseMaintenance = new DatabaseMaintenanceService()
|
const redisManager = RedisSessionManager.getInstance();
|
||||||
await databaseMaintenance.runMaintenanceTasks()
|
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')
|
console.log('Reddit Session Start')
|
||||||
await redditSession.getInstance()
|
const redditSessionManager = await redditSession.getInstance()
|
||||||
|
|
||||||
// Initialize and start your workflow
|
|
||||||
const workflowOrchestrator = new WorkflowOrchestrator();
|
const workflowOrchestrator = new WorkflowOrchestrator();
|
||||||
await workflowOrchestrator.executeWorkflow();
|
|
||||||
|
|
||||||
await rDramaSessionManager.shutdown()
|
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()
|
startApplication()
|
||||||
|
|
Loading…
Reference in New Issue