67 lines
2.3 KiB
Java
67 lines
2.3 KiB
Java
package cn.bunny.controller;
|
|
|
|
import cn.bunny.domain.dto.VmsArgumentDto;
|
|
import cn.bunny.domain.result.Result;
|
|
import cn.bunny.domain.vo.GeneratorVo;
|
|
import cn.bunny.service.GeneratorService;
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
import jakarta.validation.Valid;
|
|
import lombok.RequiredArgsConstructor;
|
|
import org.apache.logging.log4j.util.Strings;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* 代码生成器控制器
|
|
* 提供代码生成和打包下载功能
|
|
*/
|
|
@Tag(name = "生成模板", description = "生成模板接口")
|
|
@RestController
|
|
@RequestMapping("/api/generator")
|
|
@RequiredArgsConstructor
|
|
public class GeneratorController {
|
|
|
|
private final GeneratorService generatorService;
|
|
|
|
/**
|
|
* 生成代码
|
|
*
|
|
* @param dto 生成参数DTO
|
|
* @return 生成的代码结果，按表名分组
|
|
*/
|
|
@Operation(summary = "生成代码", description = "根据SQL或数据库表生成代码")
|
|
@PostMapping
|
|
public Result<Map<String, List<GeneratorVo>>> generator(@Valid @RequestBody VmsArgumentDto dto) {
|
|
// 判断当前是使用 SQL语句 生成还是 数据库生成
|
|
String sql = dto.getSql();
|
|
|
|
Map<String, List<GeneratorVo>> result = Strings.isEmpty(sql)
|
|
? generatorService.generateCodeByDatabase(dto)
|
|
: generatorService.generateCodeBySql(dto);
|
|
return Result.success(result);
|
|
}
|
|
|
|
/**
|
|
* 打包代码为ZIP下载
|
|
*
|
|
* @param dto 生成参数DTO
|
|
* @return ZIP文件响应实体
|
|
*/
|
|
@Operation(summary = "打包下载", description = "将生成的代码打包为ZIP文件下载")
|
|
@PostMapping("downloadByZip")
|
|
public ResponseEntity<byte[]> downloadByZip(@Valid @RequestBody VmsArgumentDto dto) {
|
|
// 判断当前是使用 SQL语句 生成还是 数据库生成
|
|
String sql = dto.getSql();
|
|
|
|
return Strings.isEmpty(sql)
|
|
? generatorService.downloadByZipByDatabase(dto)
|
|
: generatorService.downloadByZipBySqL(dto);
|
|
}
|
|
} |