n8n
Version:
n8n Workflow Automation Tool
58 lines • 2.88 kB
JavaScript
;
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.AgentCheckpointRepository = void 0;
const di_1 = require("@n8n/di");
const typeorm_1 = require("@n8n/typeorm");
const agent_checkpoint_entity_1 = require("../entities/agent-checkpoint.entity");
let AgentCheckpointRepository = class AgentCheckpointRepository extends typeorm_1.Repository {
constructor(dataSource) {
super(agent_checkpoint_entity_1.AgentCheckpoint, dataSource.manager);
}
async findByRunId(runId) {
return await this.findOneBy({ runId });
}
async findByRunIdAndAgentId(runId, agentId) {
return await this.findOneBy({ runId, agentId });
}
async findActiveForAgent(agentId) {
return await this.find({
where: { agentId, expired: false },
order: { updatedAt: 'DESC' },
});
}
async claimForResume(runId, agentId, suspendedState, runningState) {
const result = await this.update({ runId, agentId, expired: false, state: suspendedState }, { state: runningState });
return (result.affected ?? 0) > 0;
}
async cancelSuspended(runId, agentId, suspendedState) {
const result = await this.update({ runId, agentId, expired: false, state: suspendedState }, { expired: true });
return (result.affected ?? 0) > 0;
}
async expireByRunIdAndAgentId(runId, agentId) {
await this.update({ runId, agentId }, { expired: true, state: null });
}
async markExpired(olderThan) {
const result = await this.createQueryBuilder()
.update()
.set({ expired: true, state: null })
.where('updatedAt < :olderThan', { olderThan })
.andWhere('state IS NOT NULL')
.execute();
return result.affected ?? 0;
}
};
exports.AgentCheckpointRepository = AgentCheckpointRepository;
exports.AgentCheckpointRepository = AgentCheckpointRepository = __decorate([
(0, di_1.Service)(),
__metadata("design:paramtypes", [typeorm_1.DataSource])
], AgentCheckpointRepository);
//# sourceMappingURL=agent-checkpoint.repository.js.map