feat(database): magic attribute model

This commit is contained in:
chenos 2022-01-07 20:26:23 +08:00
parent 2bf09bf9bb
commit ef6e2dd378
2 changed files with 47 additions and 0 deletions

View File

@ -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';

View 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,
};
}
}