Typeorm 自定义装饰器

时间:2021-03-09 22:48:55

标签: javascript node.js typescript typeorm

我希望能够向我的实体模型添加自定义装饰器,以标记特定字段以进行用户元数据迁移。

例如

import {Entity, PrimaryGeneratedColumn, Column} from "typeorm";
import { CreatedBy } from "../subscribers/test";

@Entity()
export class User {

    @PrimaryGeneratedColumn()
    id!: number;

    @Column()
    name!: string;

    @Migrate()
    userValue!: string;
 
    @Migrate()
    specficValue!: string;

}

其中 migrate 可能类似于 Column 装饰器,但我可以采用 User 模型,并找到具有 migrate 装饰器的所有属性,以便我知道要推进哪些属性。

有没有办法在 TypeORM 中实现这一点?

感谢您的任何帮助。

1 个答案:

答案 0 :(得分:2)

检查以下 github Tapeorm 问题: https://github.com/typeorm/typeorm/issues/2711

具体来说,我会检查“ColumnCommonOptions”装饰器的工作原理: https://github.com/typeorm/typeorm/blob/master/src/decorator/options/ColumnCommonOptions.ts

这似乎很简单:

// MigrateDecorator.ts

import { getMetadataArgsStorage } from "typeorm";
//Optional
import { MigrateOptionsInterface } from './MigrateOptionsInterface'

 export function MigrateDecorator(options?: MigrateOptionsInterface): Function { 
     return function (object: Object, propertyName: string) { 
         getMetadataArgsStorage().columns.push({
           propertyName,
           migrate: options.migrate || false // just an example
         }); 
     }; 
 }