问题来源
我在写后端创建或更新数据的时候,通常是将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];
}
}
}
打完收工,完美~。