当前位置：首页 > news >正文

使用过程宏实现自动化新增功能

news 2025/9/30 9:11:13

使用过程宏实现自动化新增功能

简化后端开发：使用Rust过程宏自动化CRUD新增操作

在现代Web开发中，重复的CRUD（增删改查）操作占据了大量开发时间。今天我们将探讨如何使用Rust过程宏来自动化新增功能的实现。

问题背景

在传统的Rust Web开发中，每个实体都需要手动编写新增操作的代码：

pub async fn create_category(db: &DatabaseConnection,data: web::Json<CreateCategoryRequest>,
) -> Result<HttpResponse, AppError> {let active_model = category::ActiveModel {name: Set(data.name.clone()),slug: Set(data.slug.clone()),description: Set(data.description.clone()),..Default::default()};let category = active_model.insert(db).await?;Ok(HttpResponse::Created().json(category))
}

这种重复性工作不仅耗时，而且容易出错。

解决方案：过程宏

我们使用Rust的过程宏来自动生成这些重复代码：

crud_entity!({entity: categories,route_prefix: "/api/categories",permission_prefix: "categories",id_type: "integer",operations: ["create"],create_request_type: CreateCategoryRequest
});

核心实现

1. 请求类型处理

fn generate_create_code(entity: &Ident,route_prefix: &LitStr,permission_prefix: &LitStr,create_request_type: &Option<proc_macro2::Ident>,
) -> proc_macro2::TokenStream {let create_fn = format_ident!("create_{}", entity.to_string().to_lowercase());let create_handler = format_ident!("create_{}_handler", entity.to_string().to_lowercase());let create_request_type = match create_request_type {Some(ident) => ident,None => {return syn::Error::new_spanned(entity,"create_request_type is required for Create operation",).to_compile_error().into();}};// 生成代码...
}

2. 业务逻辑生成

quote! {pub async fn #create_fn(db: &DatabaseConnection,data: #create_request_type,) -> Result<#entity::Model, AppError> {let active_model = #entity::ActiveModel::from(data);let model = active_model.insert(db).await.map_err(|e| {println!("添加分类失败: {}", e);AppError::DatabaseConnectionError(db_err_map(e).to_owned())})?;Ok(model)}
}

使用示例

1. 定义请求DTO

#[derive(Debug, Deserialize)]
pub struct CreateCategoryRequest {pub name: String,pub slug: String,pub description: Option<String>,
}

2. 实现类型转换

impl From<CreateCategoryRequest> for category::ActiveModel {fn from(request: CreateCategoryRequest) -> Self {category::ActiveModel {name: Set(request.name),slug: Set(request.slug),description: Set(request.description),..Default::default()}}
}

3. 使用宏生成

crud_entity!({entity: categories,route_prefix: "/api/categories",permission_prefix: "categories",operations: ["create"],create_request_type: CreateCategoryRequest
});