mirror of
https://github.com/nocobase/nocobase
synced 2024-11-15 09:47:10 +00:00
feat(database): magic attribute model
This commit is contained in:
parent
2bf09bf9bb
commit
ef6e2dd378
@ -12,3 +12,4 @@ export * from './relation-repository/multiple-relation-repository';
|
||||
export { Model, ModelCtor } from 'sequelize';
|
||||
export * from './fields';
|
||||
export * from './update-associations';
|
||||
export * from './magic-attribute-model';
|
||||
|
46
packages/database/src/magic-attribute-model.ts
Normal file
46
packages/database/src/magic-attribute-model.ts
Normal file
@ -0,0 +1,46 @@
|
||||
import { Model } from 'sequelize';
|
||||
import { merge } from '@nocobase/utils';
|
||||
import _ from 'lodash';
|
||||
|
||||
export class MagicAttributeModel extends Model {
|
||||
set(key: any, value?: any, options?: any) {
|
||||
if (typeof key === 'string') {
|
||||
const [column] = key.split('.');
|
||||
if ((this.constructor as any).hasAlias(column)) {
|
||||
return super.set(key, value, options);
|
||||
}
|
||||
if ((this.constructor as any).rawAttributes[column]) {
|
||||
return super.set(key, value, options);
|
||||
}
|
||||
if (_.isPlainObject(value)) {
|
||||
const opts = super.get(`options`) || {};
|
||||
return super.set(`options.${key}`, merge(opts?.[key], value), options);
|
||||
}
|
||||
return super.set(`options.${key}`, value, options);
|
||||
} else {
|
||||
Object.keys(key).forEach((k) => {
|
||||
this.set(k, key[k], options);
|
||||
});
|
||||
}
|
||||
return super.set(key, value, options);
|
||||
}
|
||||
|
||||
get(key?: any, value?: any): any {
|
||||
if (typeof key === 'string') {
|
||||
const [column] = key.split('.');
|
||||
if ((this.constructor as any).hasAlias(column)) {
|
||||
return super.get(key, value);
|
||||
}
|
||||
if ((this.constructor as any).rawAttributes[column]) {
|
||||
return super.get(key, value);
|
||||
}
|
||||
const options = super.get(`options`);
|
||||
return _.get(options, key);
|
||||
}
|
||||
const data = super.get(key, value);
|
||||
return {
|
||||
..._.omit(data, 'options'),
|
||||
...data.options,
|
||||
};
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user