mirror of
https://github.com/xCyanGrizzly/DragonsStash.git
synced 2026-05-11 06:11:15 +00:00
9adbdb2a77
- Add invokeWithTimeout wrapper for TDLib API calls (2min timeout per call) - Add stuck detection to getChannelMessages: break if from_message_id doesn't advance - Add stuck detection to getTopicMessages: same protection for topic scanning - Add stuck detection to getForumTopicList: break if pagination offsets don't advance - Add max page limit (5000) to all scanning loops to prevent infinite pagination - Add mutex wait timeout (30min) to prevent indefinite blocking when holder hangs - Add cycle timeout (4h default, configurable via WORKER_CYCLE_TIMEOUT_MINUTES) - Fix end-of-page detection to use actual limit value instead of hardcoded 100 Co-authored-by: xCyanGrizzly <53275238+xCyanGrizzly@users.noreply.github.com>
50 lines
1.8 KiB
JavaScript
50 lines
1.8 KiB
JavaScript
import { mkdir } from "fs/promises";
|
|
import { config } from "./util/config.js";
|
|
import { logger } from "./util/logger.js";
|
|
import { markStaleRunsAsFailed } from "./db/queries.js";
|
|
import { cleanupTempDir } from "./worker.js";
|
|
import { startScheduler, stopScheduler } from "./scheduler.js";
|
|
import { startFetchListener, stopFetchListener } from "./fetch-listener.js";
|
|
import { db, pool } from "./db/client.js";
|
|
const log = logger.child({ module: "main" });
|
|
async function main() {
|
|
log.info("DragonsStash Telegram Worker starting");
|
|
log.info({ config: { ...config, databaseUrl: "***" } }, "Configuration loaded");
|
|
if (!config.telegramApiId || !config.telegramApiHash) {
|
|
log.fatal("TELEGRAM_API_ID and TELEGRAM_API_HASH are both required");
|
|
process.exit(1);
|
|
}
|
|
// Ensure temp directory exists
|
|
await mkdir(config.tempDir, { recursive: true });
|
|
await mkdir(config.tdlibStateDir, { recursive: true });
|
|
// Clean up stale state
|
|
await cleanupTempDir();
|
|
await markStaleRunsAsFailed();
|
|
// Start the fetch listener (pg_notify for on-demand channel fetching)
|
|
await startFetchListener();
|
|
// Start the scheduler
|
|
await startScheduler();
|
|
}
|
|
// Graceful shutdown
|
|
function shutdown(signal) {
|
|
log.info({ signal }, "Shutdown signal received");
|
|
stopScheduler();
|
|
stopFetchListener();
|
|
// Close DB connections
|
|
Promise.all([db.$disconnect(), pool.end()])
|
|
.then(() => {
|
|
log.info("Shutdown complete");
|
|
process.exit(0);
|
|
})
|
|
.catch((err) => {
|
|
log.error({ err }, "Error during shutdown");
|
|
process.exit(1);
|
|
});
|
|
}
|
|
process.on("SIGTERM", () => shutdown("SIGTERM"));
|
|
process.on("SIGINT", () => shutdown("SIGINT"));
|
|
main().catch((err) => {
|
|
log.fatal({ err }, "Worker failed to start");
|
|
process.exit(1);
|
|
});
|
|
//# sourceMappingURL=index.js.map
|