Persist offer assistant jobs and recover results across closed views
This commit is contained in:
@@ -0,0 +1,71 @@
|
||||
import {DatabaseSync} from 'node:sqlite';
|
||||
import {mkdirSync,chmodSync} from 'node:fs';
|
||||
import {dirname,join} from 'node:path';
|
||||
import {fileURLToPath} from 'node:url';
|
||||
import {randomUUID,createHash} from 'node:crypto';
|
||||
const root=process.env.IHALEIST_USER_DATA_DIR || join(dirname(fileURLToPath(import.meta.url)),'data');
|
||||
const operations=new Set(['belge-sor','teklif-denetim','teklif-taslak']);
|
||||
function dbWork(fn) {
|
||||
mkdirSync(root,{recursive:true,mode:0o700});const db=new DatabaseSync(join(root,'ai-jobs.sqlite'));chmodSync(join(root,'ai-jobs.sqlite'),0o600);
|
||||
try {
|
||||
db.exec(`PRAGMA busy_timeout=5000;
|
||||
CREATE TABLE IF NOT EXISTS ai_jobs(id TEXT PRIMARY KEY,email TEXT NOT NULL,request_id TEXT NOT NULL,operation TEXT NOT NULL,ikn TEXT,
|
||||
input TEXT NOT NULL,input_hash TEXT NOT NULL,state TEXT NOT NULL,created_at TEXT NOT NULL,updated_at TEXT NOT NULL,
|
||||
result TEXT,error TEXT,run_id TEXT,heartbeat INTEGER,UNIQUE(email,request_id));
|
||||
CREATE INDEX IF NOT EXISTS job_owner ON ai_jobs(email,created_at);
|
||||
CREATE INDEX IF NOT EXISTS job_queue ON ai_jobs(state,created_at);`);
|
||||
return fn(db);
|
||||
}finally {db.close();}
|
||||
}
|
||||
function publicJob(row) {
|
||||
if(!row)return null;
|
||||
return {id:row.id,operation:row.operation,ikn:row.ikn,state:row.state,createdAt:row.created_at,updatedAt:row.updated_at,
|
||||
result:row.result?JSON.parse(row.result):null,error:row.error,inputHash:row.input_hash};
|
||||
}
|
||||
export function createAiJob(email,requestId,operation,input) {
|
||||
if(typeof requestId!=='string'||requestId.length<8||requestId.length>100||!operations.has(operation)||!input||typeof input!=='object'||Array.isArray(input))throw new Error('Geçersiz AI işi.');
|
||||
const serialized=JSON.stringify(input),hash=createHash('sha256').update(operation+'\n'+serialized).digest('hex');
|
||||
if(Buffer.byteLength(serialized)>2*1024*1024)throw new Error('AI kaynak verisi çok büyük.');
|
||||
return dbWork(db=>{
|
||||
db.exec('BEGIN IMMEDIATE');
|
||||
try {
|
||||
const previous=db.prepare('SELECT * FROM ai_jobs WHERE email=? AND request_id=?').get(email,requestId);
|
||||
if(previous) {if(previous.input_hash!==hash)throw new Error('İstek kimliği farklı kaynak için kullanılmış.');db.exec('COMMIT');return publicJob(previous);}
|
||||
if(db.prepare("SELECT COUNT(*) n FROM ai_jobs WHERE email=? AND state IN ('queued','running')").get(email).n>=5)throw new Error('En fazla beş etkin AI işi olabilir. Mevcut işleri tamamlayın veya iptal edin.');
|
||||
const id=randomUUID(),now=new Date().toISOString();
|
||||
db.prepare("INSERT INTO ai_jobs(id,email,request_id,operation,ikn,input,input_hash,state,created_at,updated_at) VALUES (?,?,?,?,?,?,?,'queued',?,?)")
|
||||
.run(id,email,requestId,operation,input.ikn||null,serialized,hash,now,now);
|
||||
const result=publicJob(db.prepare('SELECT * FROM ai_jobs WHERE id=?').get(id));db.exec('COMMIT');return result;
|
||||
} catch(error){db.exec('ROLLBACK');throw error;}
|
||||
});
|
||||
}
|
||||
export const getAiJob=(email,id)=>dbWork(db=>{const row=db.prepare('SELECT * FROM ai_jobs WHERE email=? AND id=?').get(email,id);return row?{...publicJob(row),input:JSON.parse(row.input)}:null;});
|
||||
export const listAiJobs=(email,ikn)=>dbWork(db=>db.prepare('SELECT * FROM ai_jobs WHERE email=? AND (? IS NULL OR ikn=?) ORDER BY created_at DESC LIMIT 50').all(email,ikn||null,ikn||null).map(publicJob));
|
||||
export function claimAiJob(clock=Date.now()) {
|
||||
return dbWork(db=>{
|
||||
db.exec('BEGIN IMMEDIATE');
|
||||
try {
|
||||
// Do not replay an uncertain provider call automatically after a lost worker.
|
||||
db.prepare("UPDATE ai_jobs SET state='interrupted',error=?,updated_at=? WHERE state='running' AND heartbeat<?")
|
||||
.run('İşlem kesildi. Kaynak korunuyor; yeniden başlatabilirsiniz.',new Date(clock).toISOString(),clock-60000);
|
||||
const row=db.prepare("SELECT * FROM ai_jobs WHERE state='queued' ORDER BY created_at LIMIT 1").get();
|
||||
if(!row){db.exec('COMMIT');return null;}
|
||||
const runId=randomUUID();
|
||||
db.prepare("UPDATE ai_jobs SET state='running',run_id=?,heartbeat=?,updated_at=? WHERE id=?").run(runId,clock,new Date(clock).toISOString(),row.id);
|
||||
db.exec('COMMIT');return {id:row.id,email:row.email,operation:row.operation,input:JSON.parse(row.input),runId};
|
||||
}catch(error){db.exec('ROLLBACK');throw error;}
|
||||
});
|
||||
}
|
||||
export const heartbeatAiJob=(id,runId)=>dbWork(db=>db.prepare("UPDATE ai_jobs SET heartbeat=? WHERE id=? AND run_id=? AND state='running'").run(Date.now(),id,runId).changes>0);
|
||||
export const finishAiJob=(id,runId,result,error=null)=>dbWork(db=>db.prepare("UPDATE ai_jobs SET state=?,result=?,error=?,updated_at=? WHERE id=? AND run_id=? AND state='running'")
|
||||
.run(error?'failed':'completed',error?null:JSON.stringify(result),error,new Date().toISOString(),id,runId).changes>0);
|
||||
export const cancelAiJob=(email,id)=>dbWork(db=>db.prepare("UPDATE ai_jobs SET state='cancelled',error=NULL,updated_at=? WHERE id=? AND email=? AND state IN ('queued','running','interrupted','failed')").run(new Date().toISOString(),id,email).changes>0);
|
||||
export const retryAiJob=(email,id)=>dbWork(db=>{
|
||||
db.exec('BEGIN IMMEDIATE');
|
||||
try {
|
||||
const active=db.prepare("SELECT COUNT(*) n FROM ai_jobs WHERE email=? AND state IN ('queued','running')").get(email).n;
|
||||
if(active>=5)throw new Error('En fazla beş etkin AI işi olabilir.');
|
||||
const changed=db.prepare("UPDATE ai_jobs SET state='queued',error=NULL,result=NULL,run_id=NULL,updated_at=? WHERE id=? AND email=? AND state IN ('interrupted','failed')").run(new Date().toISOString(),id,email).changes>0;
|
||||
db.exec('COMMIT');return changed;
|
||||
}catch(error){db.exec('ROLLBACK');throw error;}
|
||||
});
|
||||
Reference in New Issue
Block a user