fix(reliability): дневной лимит imggen в БД + ретеншн error_log (Спринт3)
- imggen: дневной счётчик генераций перенесён из in-memory Map в таблицу imggen_usage (миграция 070) — переживает рестарт. Cooldown остаётся в памяти, но добавлена периодическая чистка Map + старых строк imggen_usage (>7 дн). - classroom-cleanup: ретеншн error_log (app_settings.error_log_retention_days, по умолч. 30; 0 = выкл) в том же суточном джобе. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
@@ -9,11 +9,26 @@ const db = require('../db/db');
|
||||
const { isFeatureEnabledForUser } = require('../middleware/features');
|
||||
|
||||
const GEN_DIR = path.join(__dirname, '../../uploads/generated');
|
||||
const _cooldown = new Map(); // userId → last ts (антиспам)
|
||||
const _daily = new Map(); // userId → { day, count }
|
||||
const _cooldown = new Map(); // userId → last ts (антиспам, короткоживущий, ок в памяти)
|
||||
const COOLDOWN_DEFAULT = 4000;
|
||||
const DAILY_DEFAULT = 40;
|
||||
|
||||
// Дневной счётчик — в БД (imggen_usage), переживает рестарт.
|
||||
function _dailyCount(uid, day) {
|
||||
try { const r = db.prepare('SELECT count FROM imggen_usage WHERE user_id=? AND day=?').get(uid, day); return r ? r.count : 0; }
|
||||
catch (e) { return 0; }
|
||||
}
|
||||
function _dailyBump(uid, day) {
|
||||
try { db.prepare('INSERT INTO imggen_usage (user_id, day, count) VALUES (?,?,1) ON CONFLICT(user_id, day) DO UPDATE SET count = count + 1').run(uid, day); }
|
||||
catch (e) {}
|
||||
}
|
||||
// Чистка in-memory cooldown (старше минуты) + старых строк imggen_usage — не растём бесконечно.
|
||||
setInterval(() => {
|
||||
const cut = Date.now() - 60_000;
|
||||
for (const [k, t] of _cooldown) if (t < cut) _cooldown.delete(k);
|
||||
try { db.prepare("DELETE FROM imggen_usage WHERE day < date('now','-7 days')").run(); } catch (e) {}
|
||||
}, 5 * 60_000).unref();
|
||||
|
||||
function _cfg() {
|
||||
try { const r = db.prepare("SELECT value FROM app_settings WHERE key='imggen_provider'").get(); return r ? JSON.parse(r.value) : null; } catch (e) { return null; }
|
||||
}
|
||||
@@ -132,13 +147,12 @@ async function generate(req, res) {
|
||||
const L = _limits();
|
||||
if (now - (_cooldown.get(uid) || 0) < L.cooldownMs) return res.status(429).json({ error: 'Чуть помедленнее — подожди пару секунд' });
|
||||
const today = new Date().toISOString().slice(0, 10);
|
||||
const d = _daily.get(uid);
|
||||
if (L.dailyCap > 0 && d && d.day === today && d.count >= L.dailyCap) return res.status(429).json({ error: 'Дневной лимит генераций исчерпан, попробуй завтра' });
|
||||
if (L.dailyCap > 0 && _dailyCount(uid, today) >= L.dailyCap) return res.status(429).json({ error: 'Дневной лимит генераций исчерпан, попробуй завтра' });
|
||||
_cooldown.set(uid, now);
|
||||
|
||||
try {
|
||||
const out = await generateImage(prompt, uid);
|
||||
_daily.set(uid, { day: today, count: (d && d.day === today ? d.count : 0) + 1 });
|
||||
_dailyBump(uid, today);
|
||||
res.json({ url: out.url });
|
||||
} catch (e) {
|
||||
res.status(502).json({ error: e.message || 'Не удалось сгенерировать', detail: e.detail });
|
||||
|
||||
Reference in New Issue
Block a user