188 lines
7.5 KiB
JavaScript
188 lines
7.5 KiB
JavaScript
"use strict";
|
||
Object.defineProperty(exports, "__esModule", { value: true });
|
||
exports.AsyncContext = void 0;
|
||
var tslib_1 = require("tslib");
|
||
var assert_1 = tslib_1.__importDefault(require("assert"));
|
||
var AsyncContext = /** @class */ (function () {
|
||
function AsyncContext(store, headers) {
|
||
this.rowStore = store;
|
||
this.opRecords = [];
|
||
this.events = {
|
||
commit: [],
|
||
rollback: [],
|
||
};
|
||
if (headers) {
|
||
this.headers = headers;
|
||
}
|
||
}
|
||
AsyncContext.prototype.setHeaders = function (headers) {
|
||
this.headers = headers;
|
||
};
|
||
AsyncContext.prototype.getHeader = function (key) {
|
||
if (this.headers) {
|
||
return this.headers[key];
|
||
}
|
||
};
|
||
AsyncContext.prototype.getScene = function () {
|
||
return this.scene;
|
||
};
|
||
AsyncContext.prototype.setScene = function (scene) {
|
||
this.scene = scene;
|
||
};
|
||
AsyncContext.prototype.resetEvents = function () {
|
||
this.events = {
|
||
commit: [],
|
||
rollback: [],
|
||
};
|
||
};
|
||
AsyncContext.prototype.on = function (event, callback) {
|
||
this.uuid && this.events[event].push(callback);
|
||
};
|
||
/**
|
||
* 一个context中不应该有并发的事务，这里将事务串行化，使用的时候千万要注意不要自己等自己
|
||
* @param options
|
||
*/
|
||
AsyncContext.prototype.begin = function (options) {
|
||
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
||
var _a;
|
||
return tslib_1.__generator(this, function (_b) {
|
||
switch (_b.label) {
|
||
case 0:
|
||
if (!!this.uuid) return [3 /*break*/, 2];
|
||
_a = this;
|
||
return [4 /*yield*/, this.rowStore.begin(options)];
|
||
case 1:
|
||
_a.uuid = _b.sent();
|
||
return [3 /*break*/, 3];
|
||
case 2:
|
||
(0, assert_1.default)(false);
|
||
_b.label = 3;
|
||
case 3: return [2 /*return*/];
|
||
}
|
||
});
|
||
});
|
||
};
|
||
AsyncContext.prototype.commit = function () {
|
||
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
||
var commitEvents, commitEvents_1, commitEvents_1_1, e, e_1_1;
|
||
var e_1, _a;
|
||
return tslib_1.__generator(this, function (_b) {
|
||
switch (_b.label) {
|
||
case 0:
|
||
if (!this.uuid) return [3 /*break*/, 9];
|
||
return [4 /*yield*/, this.rowStore.commit(this.uuid)];
|
||
case 1:
|
||
_b.sent();
|
||
this.uuid = undefined;
|
||
commitEvents = this.events.commit;
|
||
this.resetEvents();
|
||
_b.label = 2;
|
||
case 2:
|
||
_b.trys.push([2, 7, 8, 9]);
|
||
commitEvents_1 = tslib_1.__values(commitEvents), commitEvents_1_1 = commitEvents_1.next();
|
||
_b.label = 3;
|
||
case 3:
|
||
if (!!commitEvents_1_1.done) return [3 /*break*/, 6];
|
||
e = commitEvents_1_1.value;
|
||
return [4 /*yield*/, e()];
|
||
case 4:
|
||
_b.sent();
|
||
_b.label = 5;
|
||
case 5:
|
||
commitEvents_1_1 = commitEvents_1.next();
|
||
return [3 /*break*/, 3];
|
||
case 6: return [3 /*break*/, 9];
|
||
case 7:
|
||
e_1_1 = _b.sent();
|
||
e_1 = { error: e_1_1 };
|
||
return [3 /*break*/, 9];
|
||
case 8:
|
||
try {
|
||
if (commitEvents_1_1 && !commitEvents_1_1.done && (_a = commitEvents_1.return)) _a.call(commitEvents_1);
|
||
}
|
||
finally { if (e_1) throw e_1.error; }
|
||
return [7 /*endfinally*/];
|
||
case 9: return [2 /*return*/];
|
||
}
|
||
});
|
||
});
|
||
};
|
||
AsyncContext.prototype.rollback = function () {
|
||
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
||
var rollbackEvents, rollbackEvents_1, rollbackEvents_1_1, e, e_2_1;
|
||
var e_2, _a;
|
||
return tslib_1.__generator(this, function (_b) {
|
||
switch (_b.label) {
|
||
case 0:
|
||
if (!this.uuid) return [3 /*break*/, 9];
|
||
return [4 /*yield*/, this.rowStore.rollback(this.uuid)];
|
||
case 1:
|
||
_b.sent();
|
||
// console.log('rollback', this.uuid);
|
||
this.uuid = undefined;
|
||
rollbackEvents = this.events.rollback;
|
||
this.resetEvents();
|
||
_b.label = 2;
|
||
case 2:
|
||
_b.trys.push([2, 7, 8, 9]);
|
||
rollbackEvents_1 = tslib_1.__values(rollbackEvents), rollbackEvents_1_1 = rollbackEvents_1.next();
|
||
_b.label = 3;
|
||
case 3:
|
||
if (!!rollbackEvents_1_1.done) return [3 /*break*/, 6];
|
||
e = rollbackEvents_1_1.value;
|
||
return [4 /*yield*/, e()];
|
||
case 4:
|
||
_b.sent();
|
||
_b.label = 5;
|
||
case 5:
|
||
rollbackEvents_1_1 = rollbackEvents_1.next();
|
||
return [3 /*break*/, 3];
|
||
case 6: return [3 /*break*/, 9];
|
||
case 7:
|
||
e_2_1 = _b.sent();
|
||
e_2 = { error: e_2_1 };
|
||
return [3 /*break*/, 9];
|
||
case 8:
|
||
try {
|
||
if (rollbackEvents_1_1 && !rollbackEvents_1_1.done && (_a = rollbackEvents_1.return)) _a.call(rollbackEvents_1);
|
||
}
|
||
finally { if (e_2) throw e_2.error; }
|
||
return [7 /*endfinally*/];
|
||
case 9: return [2 /*return*/];
|
||
}
|
||
});
|
||
});
|
||
};
|
||
AsyncContext.prototype.operate = function (entity, operation, option) {
|
||
return this.rowStore.operate(entity, operation, this, option);
|
||
};
|
||
AsyncContext.prototype.select = function (entity, selection, option) {
|
||
return this.rowStore.select(entity, selection, this, option);
|
||
};
|
||
AsyncContext.prototype.aggregate = function (entity, aggregation, option) {
|
||
return this.rowStore.aggregate(entity, aggregation, this, option);
|
||
};
|
||
AsyncContext.prototype.count = function (entity, selection, option) {
|
||
return this.rowStore.count(entity, selection, this, option);
|
||
};
|
||
AsyncContext.prototype.mergeMultipleResults = function (toBeMerged) {
|
||
return this.rowStore.mergeMultipleResults(toBeMerged);
|
||
};
|
||
AsyncContext.prototype.getCurrentTxnId = function () {
|
||
return this.uuid;
|
||
};
|
||
AsyncContext.prototype.getSchema = function () {
|
||
return this.rowStore.getSchema();
|
||
};
|
||
AsyncContext.prototype.setMessage = function (message) {
|
||
this.message = message;
|
||
};
|
||
AsyncContext.prototype.getMessage = function () {
|
||
return this.message;
|
||
};
|
||
return AsyncContext;
|
||
}());
|
||
exports.AsyncContext = AsyncContext;
|
||
;
|
||
;
|