"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : new P(function (resolve) { resolve(result.value); }).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const moment = require("moment");
const task_1 = require("./mongoose/model/task");
const factory = require("../factory");
/**
* タスク実行時のソート条件
*/
const sortOrder4executionOfTasks = {
numberOfTried: 1,
runsAt: 1 // 実行予定日時の早さ優先
};
/**
* タスクリポジトリ
*/
class MongoRepository {
constructor(connection) {
this.taskModel = connection.model(task_1.default.modelName);
}
static CREATE_MONGO_CONDITIONS(params) {
const andConditions = [{
name: { $exists: true }
}];
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.project !== undefined) {
if (Array.isArray(params.project.ids)) {
andConditions.push({
'project.id': {
$exists: true,
$in: params.project.ids
}
});
}
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.name !== undefined) {
andConditions.push({
name: params.name
});
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (Array.isArray(params.statuses)) {
andConditions.push({
status: { $in: params.statuses }
});
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.runsFrom !== undefined) {
andConditions.push({
runsAt: { $gte: params.runsFrom }
});
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.runsThrough !== undefined) {
andConditions.push({
runsAt: { $lte: params.runsThrough }
});
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.lastTriedFrom !== undefined) {
andConditions.push({
lastTriedAt: {
$type: 'date',
$gte: params.lastTriedFrom
}
});
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.lastTriedThrough !== undefined) {
andConditions.push({
lastTriedAt: {
$type: 'date',
$lte: params.lastTriedThrough
}
});
}
return andConditions;
}
save(taskAttributes) {
return __awaiter(this, void 0, void 0, function* () {
return this.taskModel.create(taskAttributes)
.then((doc) => doc.toObject());
});
}
executeOneByName(taskName) {
return __awaiter(this, void 0, void 0, function* () {
const doc = yield this.taskModel.findOneAndUpdate({
status: factory.taskStatus.Ready,
runsAt: { $lt: new Date() },
name: taskName
}, {
status: factory.taskStatus.Running,
lastTriedAt: new Date(),
$inc: {
remainingNumberOfTries: -1,
numberOfTried: 1 // トライ回数増やす
}
}, { new: true })
.sort(sortOrder4executionOfTasks)
.exec();
if (doc === null) {
// tslint:disable-next-line:no-null-keyword
return null;
}
return doc.toObject();
});
}
retry(intervalInMinutes) {
return __awaiter(this, void 0, void 0, function* () {
const lastTriedAtShoudBeLessThan = moment()
.add(-intervalInMinutes, 'minutes')
.toDate();
yield this.taskModel.update({
status: factory.taskStatus.Running,
lastTriedAt: {
$type: 'date',
$lt: lastTriedAtShoudBeLessThan
},
remainingNumberOfTries: { $gt: 0 }
}, {
status: factory.taskStatus.Ready // 実行前に変更
}, { multi: true })
.exec();
});
}
abortOne(intervalInMinutes) {
return __awaiter(this, void 0, void 0, function* () {
const lastTriedAtShoudBeLessThan = moment()
.add(-intervalInMinutes, 'minutes')
.toDate();
const doc = yield this.taskModel.findOneAndUpdate({
status: factory.taskStatus.Running,
lastTriedAt: {
$type: 'date',
$lt: lastTriedAtShoudBeLessThan
},
remainingNumberOfTries: 0
}, {
status: factory.taskStatus.Aborted
}, { new: true })
.exec();
if (doc === null) {
// tslint:disable-next-line:no-null-keyword
return null;
}
return doc.toObject();
});
}
pushExecutionResultById(id, status, executionResult) {
return __awaiter(this, void 0, void 0, function* () {
yield this.taskModel.findByIdAndUpdate(id, {
status: status,
$push: { executionResults: executionResult }
})
.exec();
});
}
/**
* 特定タスク検索
*/
findById(params) {
return __awaiter(this, void 0, void 0, function* () {
const doc = yield this.taskModel.findOne({
name: params.name,
_id: params.id
}, {
__v: 0,
createdAt: 0,
updatedAt: 0
})
.exec();
if (doc === null) {
throw new factory.errors.NotFound('Task');
}
return doc.toObject();
});
}
count(params) {
return __awaiter(this, void 0, void 0, function* () {
const conditions = MongoRepository.CREATE_MONGO_CONDITIONS(params);
return this.taskModel.countDocuments((conditions.length > 0) ? { $and: conditions } : {})
.setOptions({ maxTimeMS: 10000 })
.exec();
});
}
/**
* 検索する
*/
search(params) {
return __awaiter(this, void 0, void 0, function* () {
const conditions = MongoRepository.CREATE_MONGO_CONDITIONS(params);
const query = this.taskModel.find((conditions.length > 0) ? { $and: conditions } : {}, {
__v: 0,
createdAt: 0,
updatedAt: 0
});
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.limit !== undefined && params.page !== undefined) {
query.limit(params.limit)
.skip(params.limit * (params.page - 1));
}
// tslint:disable-next-line:no-single-line-block-comment
/* istanbul ignore else */
if (params.sort !== undefined) {
query.sort(params.sort);
}
return query.setOptions({ maxTimeMS: 10000 })
.exec()
.then((docs) => docs.map((doc) => doc.toObject()));
});
}
}
exports.MongoRepository = MongoRepository;