mirrors/nocobase
1
0
Fork 0
mirror of https://github.com/nocobase/nocobase synced 2024-11-14 22:15:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(plugin-workflow): fix column name caused error when underscored (#5364)

Browse Source
This commit is contained in:
Junyi 2024-10-10 14:13:41 +08:00 committed by GitHub
parent 0183ef0561
commit 75efed6a19
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
packages/plugins/@nocobase/plugin-workflow/src/server/triggers/ScheduleTrigger/DateFieldScheduleTrigger.ts
View File

@ -12,7 +12,7 @@ import parser from 'cron-parser';
import type Plugin from '../../Plugin';
import type { WorkflowModel } from '../../types';
import { parseDateWithoutMs, SCHEDULE_MODE } from './utils';
import { parseCollectionName } from '@nocobase/data-source-manager';
import { parseCollectionName, SequelizeCollectionManager } from '@nocobase/data-source-manager';
export type ScheduleOnField = {
field: string;
@ -68,10 +68,10 @@ function getDataOptionTime(record, on, dir = 1) {
const DialectTimestampFnMap: { [key: string]: (col: string) => string } = {
postgres(col) {
return `CAST(FLOOR(extract(epoch from "${col}")) AS INTEGER)`;
return `CAST(FLOOR(extract(epoch from ${col})) AS INTEGER)`;
},
mysql(col) {
return `CAST(FLOOR(UNIX_TIMESTAMP(\`${col}\`)) AS SIGNED INTEGER)`;
return `CAST(FLOOR(UNIX_TIMESTAMP(${col})) AS SIGNED INTEGER)`;
},
sqlite(col) {
return `CAST(FLOOR(unixepoch(${col})) AS INTEGER)`;
@ -160,7 +160,7 @@ export default class ScheduleTrigger {
{ config: { collection, limit, startsOn, repeat, endsOn }, allExecuted }: WorkflowModel,
currentDate: Date,
) {
const { db } = this.workflow.app;
const { dataSourceManager } = this.workflow.app;
if (limit && allExecuted >= limit) {
return [];
}
@ -174,6 +174,14 @@ export default class ScheduleTrigger {
return [];
}
const [dataSourceName, collectionName] = parseCollectionName(collection);
const { collectionManager } = dataSourceManager.get(dataSourceName);
if (!(collectionManager instanceof SequelizeCollectionManager)) {
return [];
}
const { db } = collectionManager;
const { model } = collectionManager.getCollection(collectionName);
const range = this.cacheCycle * 2;
const conditions: any[] = [
@ -191,9 +199,12 @@ export default class ScheduleTrigger {
if (typeof repeat === 'number') {
const tsFn = DialectTimestampFnMap[db.options.dialect];
if (repeat > range && tsFn) {
const { field } = model.getAttributes()[startsOn.field];
const modExp = fn(
'MOD',
literal(`${Math.round(timestamp / 1000)} - ${tsFn(startsOn.field)}`),
literal(
`${Math.round(timestamp / 1000)} - ${db.sequelize.getQueryInterface().quoteIdentifiers(tsFn(field))}`,
),
Math.round(repeat / 1000),
);
conditions.push(where(modExp, { [Op.lt]: Math.round(range / 1000) }));
@ -230,7 +241,6 @@ export default class ScheduleTrigger {
});
}
const { model } = db.getCollection(collection);
return model.findAll({
where: {
[Op.and]: conditions,