dto对entity的赋值(后端, node, ts, nestjs, typeorm)

问题来源

    我在写后端创建或更新数据的时候,通常是将dto的属性值赋给entity类,然后保存entity,原本的写法,是挨个属性进行赋值,就是下面的做法;
dto:

export class CatDto {

    @ApiProperty({example:"小白"})
    @IsString()
    name: string;
    
    @ApiProperty({example:2})
    @IsInt()
    age: number;

    @ApiProperty({example:"我家的" ,required:false})
    @IsString()
    description: string;
}

entity:

@Entity()
export class Cat {
    @PrimaryGeneratedColumn()
    id: number;

    @Column({ type: 'text' })
    name: string;

    @Column({ type: 'int' })
    age: number; 

    @Column({ type: 'text',default:''})
    description: string;
}

创建数据时,赋值如下:

let cat = new Cat()
cat.name = catDto.name
cat.age = catDto.age
cat.description = catDto.description
// 然后保存cat

    问题是没啥问题,但是呢,如果dto的属性非常多,那么这个赋值就会又臭又长,不优雅了,那怎么办呢?

解决过程

    我遍历下entity的属性,dto里面有一致的属性就赋值不就行了,于是有了下面的类对类的赋值方法,动态遍历dto与entity一致的属性进行赋值。
赋值方法:

export const dto2entity = (dto: any, entity:any) => {
    Object.keys(entity).forEach(key => {
        if(dto[key]){
            entity[key] = dto[key];
        }
    })
}

    上面的赋值操作就可以写成这个样子:

let cat = new Cat()
dto2entity(catDto, cat)
// 保存cat

    不管dto和entity有多少个属性,只要属性一致,我一行代码就可以赋值了。
    自行满满,用一下试试。cat竟然全是空的!什么鬼!我好像忘了什么重要的事情….. ts看起来像是java或c#那样的静态语言,但它不是,它本质上是JavaScript!new Cat() 创建的对象,是空的,毛线属性都没有!
    怎么让 new Cat() 创建的对象带属性呢?于是我改造下entity类:
entity:

@Entity()
export class Cat {
    constructor() {
        this.id = undefined;
        this.name = undefined;
        this.age = undefined;
        this.description = undefined;
    }
    @PrimaryGeneratedColumn()
    id: number;

    @Column({ type: 'text' })
    name: string;

    @Column({ type: 'int' })
    age: number; 

    @Column({ type: 'text',default:''})
    description: string;
}

    好么,成功了, new Cat() 的对象都有属性了….但是,貌似这样写也挺麻烦的,具体要不要这么搞,看个人选择了。对业务逻辑层(service层)代码优雅性有点执念的我,还是决定就这么搞了。
    然后,在测试的时候,我又碰到了另外一个问题,测试的dto数据如下:

{
    "name": "Tom",
    "age": 0,
    "description": "这个猫老倒霉了"
}

    数据库保存cat的时候,报出异常,告诉我age不可以为空,打出cat来是下面的样子,没有age的值,搞毛线,我的age给我丢哪去了!!!!

{
    "id": undefined,
    "name": "Tom",
    "age": undefined,
    "description": "这个猫老倒霉了"
}

    然后发现,问题出自赋值函数 dto2entity 中,”if(dto[key])” 这个判断,将0认为是false,然后就把age属性赋值给丢掉了。改造一下:

export const dto2entity = (dto: any, entity:any) => {
    Object.keys(entity).forEach(key => {
        if(dto[key] || dto[key] === 0){
            entity[key] = dto[key];
        }
    })
}

    没问题了,但不太好看,想了下还是直接判断有没有属性得了,再改造下:

export const dto2entity = (dto: any, entity:any) => {
    for (const key in entity) {
        if (dto.hasOwnProperty(key)) {
            entity[key]= dto[key];
        }
    }
}

    打完收工,完美~。

点击数:121

发表评论

您的电子邮箱地址不会被公开。

*

code