129 lines
4.6 KiB
JavaScript
129 lines
4.6 KiB
JavaScript
import { TreeStore } from 'oak-memory-tree-store';
|
||
import { TriggerExecutor } from 'oak-domain/lib/store/TriggerExecutor';
|
||
import { RelationAuth } from "oak-domain/lib/store/RelationAuth";
|
||
;
|
||
;
|
||
export class DebugStore extends TreeStore {
|
||
executor;
|
||
relationAuth;
|
||
dataLoaded;
|
||
dataLoadedLock;
|
||
dataLoadedLockUnlocker = () => undefined;
|
||
constructor(storageSchema, contextBuilder, authDeduceRelationMap, selectFreeEntities, updateFreeDict) {
|
||
super(storageSchema);
|
||
this.dataLoaded = false;
|
||
this.dataLoadedLock = new Promise((resolve) => {
|
||
this.dataLoadedLockUnlocker = () => resolve();
|
||
});
|
||
this.executor = new TriggerExecutor(() => contextBuilder(this));
|
||
this.relationAuth = new RelationAuth(storageSchema, authDeduceRelationMap, selectFreeEntities, updateFreeDict);
|
||
}
|
||
checkRelationAsync(entity, operation, context) {
|
||
return this.relationAuth.checkRelationAsync(entity, operation, context);
|
||
}
|
||
async exec(script, txnId) {
|
||
throw new Error('debugStore dont support exec script directly');
|
||
}
|
||
aggregate(entity, aggregation, context, option) {
|
||
return this.aggregateAsync(entity, aggregation, context, option);
|
||
}
|
||
begin(option) {
|
||
return super.beginAsync();
|
||
}
|
||
commit(txnId) {
|
||
return super.commitAsync(txnId);
|
||
}
|
||
rollback(txnId) {
|
||
return super.rollbackAsync(txnId);
|
||
}
|
||
async cascadeUpdateAsync(entity, operation, context, option) {
|
||
// 如果是在modi处理过程中，所有的trigger也可以延时到apply时再处理（这时候因为modi中的数据并不实际存在，处理会有问题）
|
||
if (!option.blockTrigger) {
|
||
await this.executor.preOperation(entity, operation, context, option);
|
||
}
|
||
const result = await super.cascadeUpdateAsync(entity, operation, context, option);
|
||
if (!option.blockTrigger) {
|
||
await this.executor.postOperation(entity, operation, context, option);
|
||
}
|
||
return result;
|
||
}
|
||
async operate(entity, operation, context, option) {
|
||
if (!this.dataLoaded) {
|
||
await this.dataLoadedLock;
|
||
}
|
||
const autoCommit = !context.getCurrentTxnId();
|
||
let result;
|
||
if (autoCommit) {
|
||
await context.begin();
|
||
}
|
||
/**
|
||
* 这里似乎还有点问题，如果在后续的checker里增加了cascadeUpdate，是无法在一开始检查权限的
|
||
* 后台的DbStore也一样 by Xc 20230801
|
||
*/
|
||
try {
|
||
await this.relationAuth.checkRelationAsync(entity, operation, context);
|
||
result = await super.operateAsync(entity, operation, context, option);
|
||
}
|
||
catch (err) {
|
||
await context.rollback();
|
||
throw err;
|
||
}
|
||
if (autoCommit) {
|
||
await context.commit();
|
||
}
|
||
return result;
|
||
}
|
||
async select(entity, selection, context, option) {
|
||
if (!this.dataLoaded) {
|
||
await this.dataLoadedLock;
|
||
}
|
||
const autoCommit = !context.getCurrentTxnId();
|
||
if (autoCommit) {
|
||
await context.begin();
|
||
}
|
||
Object.assign(selection, {
|
||
action: 'select',
|
||
});
|
||
// select的trigger应加在根结点的动作之前
|
||
try {
|
||
if (!option.blockTrigger) {
|
||
await this.executor.preOperation(entity, selection, context, option);
|
||
}
|
||
if (!option.dontCollect) {
|
||
await this.relationAuth.checkRelationAsync(entity, selection, context);
|
||
}
|
||
const result = await super.selectAsync(entity, selection, context, option);
|
||
if (!option.blockTrigger) {
|
||
await this.executor.postOperation(entity, selection, context, option, result);
|
||
}
|
||
if (autoCommit) {
|
||
await context.commit();
|
||
}
|
||
return result;
|
||
}
|
||
catch (err) {
|
||
if (autoCommit) {
|
||
await context.rollback();
|
||
}
|
||
throw err;
|
||
}
|
||
}
|
||
async count(entity, selection, context, option) {
|
||
if (!this.dataLoaded) {
|
||
await this.dataLoadedLock;
|
||
}
|
||
return super.countAsync(entity, selection, context, option);
|
||
}
|
||
registerTrigger(trigger) {
|
||
this.executor.registerTrigger(trigger);
|
||
}
|
||
registerChecker(checker) {
|
||
this.executor.registerChecker(checker, this.getSchema());
|
||
}
|
||
resetInitialData(initialData, stat) {
|
||
super.resetInitialData(initialData, stat);
|
||
this.dataLoaded = true;
|
||
this.dataLoadedLockUnlocker();
|
||
}
|
||
}
|