* Use of this software is governed by the Commercial License Agreement * obtained after purchasing a license from BladeX. *
* 1. This software is for development use only under a valid license * from BladeX. *
* 2. Redistribution of this software's source code to any third party * without a commercial license is strictly prohibited. *
* 3. Licensees may copyright their own code but cannot use segments * from this software for such purposes. Copyright of this software * remains with BladeX. *
* Using this software signifies agreement to this License, and the software * must not be used for illegal purposes. *
* THIS SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY. The author is * not liable for any claims arising from secondary or illegal development. *
* Author: Chill Zhuang (bladejava@qq.com)
*/
package org.springblade.develop.service.impl;
import com.baomidou.mybatisplus.generator.config.po.TableField;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import lombok.RequiredArgsConstructor;
import org.springblade.core.tool.jackson.JsonUtil;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.develop.constant.DevelopConstant;
import org.springblade.develop.pojo.dto.GeneratorDTO;
import org.springblade.develop.pojo.entity.*;
import org.springblade.develop.service.*;
import org.springblade.develop.support.BladeCodeGenerator;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
/**
* 服务实现类
*
* @author Chill
*/
@Service
@RequiredArgsConstructor
public class GenerateServiceImpl implements IGenerateService {
private final ICodeService codeService;
private final ICodeSettingService codeSettingService;
private final IDatasourceService datasourceService;
private final IModelService modelService;
private final IModelPrototypeService modelPrototypeService;
@Override
@Transactional(rollbackFor = Exception.class)
public boolean code(List codes = codeService.listByIds(ids);
codes.forEach(code -> {
// 创建代码生成器
BladeCodeGenerator generator = new BladeCodeGenerator();
// 设置菜单数据
this.generateMenu(generator, code);
// 设置配置信息
this.generateTemplate(generator, code);
// 设置基础模型
Model model = modelService.getById(code.getModelId());
this.generateModel(generator, code, model);
// 设置数据源
this.generateDatasource(generator, model);
// 启动代码生成
generator.run();
});
return true;
}
@Override
public boolean codeFast(GeneratorDTO dto) {
// 创建代码生成器
BladeCodeGenerator generator = new BladeCodeGenerator();
Code code = Objects.requireNonNull(BeanUtil.copyProperties(dto, Code.class));
Model model = Objects.requireNonNull(BeanUtil.copyProperties(dto, Model.class));
String modelForm = dto.getModelForm();
// 设置菜单数据
this.generateMenu(generator, code);
// 设置配置信息
this.generateForm(generator, modelForm);
this.generateTemplate(generator, code);
this.generateModel(generator, code, model);
// 设置数据源
this.generateDatasource(generator, model);
// 启动代码生成
generator.run();
return true;
}
private void generateMenu(BladeCodeGenerator generator, Code code) {
// 设置上级菜单id
generator.setMenuId(String.valueOf(code.getMenuId()));
// 设置是否生成菜单sql
generator.setHasMenuSql(Boolean.TRUE);
}
private void generateForm(BladeCodeGenerator generator, String modelForm) {
if (StringUtil.isNotBlank(modelForm)) {
CodeSetting codeSetting = codeSettingService.getById(Func.toLong(modelForm));
if (codeSetting != null) {
generator.setModelFormOption(codeSetting.getSettings());
}
}
}
private void generateTemplate(BladeCodeGenerator generator, Code code) {// 设置基础配置
generator.setCodeStyle(code.getCodeStyle());
generator.setCodeName(code.getCodeName());
generator.setServiceName(code.getServiceName());
generator.setPackageName(code.getPackageName());
generator.setPackageDir(code.getApiPath());
generator.setPackageWebDir(code.getWebPath());
generator.setTablePrefix(Func.toStrArray(code.getTablePrefix()));
generator.setIncludeTables(Func.toStrArray(code.getTableName()));
// 设置模版信息
generator.setTemplateType(Func.toStr(code.getTemplateType(), DevelopConstant.TEMPLATE_CRUD));
generator.setAuthor(code.getAuthor());
generator.setSubModelId(code.getSubModelId());
generator.setSubFkId(code.getSubFkId());
generator.setTreeId(code.getTreeId());
generator.setTreePid(code.getTreePid());
generator.setTreeName(code.getTreeName());
// 设置是否继承基础业务字段
generator.setHasSuperEntity(code.getBaseMode() == 2);
// 设置是否开启包装器模式
generator.setHasWrapper(code.getWrapMode() == 2);
// 设置是否开启远程调用模式
generator.setHasFeign(code.getFeignMode() == 2);
// 设置控制器服务名前缀
generator.setHasServiceName(Boolean.FALSE);
}
private void generateModel(BladeCodeGenerator generator, Code code, Model model) {
generator.setModelCode(model.getModelCode());
generator.setModelClass(model.getModelClass());
generator.setModel(JsonUtil.readMap(JsonUtil.toJson(model)));
// 设置模型集合
if (Func.isNotEmpty(model.getId())) {
List