package com.monkeylessey.gen.handler;
|
|
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
|
import com.monkeylessey.gen.FormTemplate;
|
import com.monkeylessey.gen.domain.GenerateData;
|
import com.monkeylessey.sys.domain.vo.TableColumnVO;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author xp
|
* @version 1.0
|
* @date 2022/5/18
|
*/
|
|
public class FormHandler {
|
|
public Map handler(Map objectMap, GenerateData data) {
|
// 获取table
|
TableInfo tableInfo = (TableInfo) objectMap.get("table");
|
// 获取包信息
|
Map<String, String> packageMap = (Map<String, String>) objectMap.get("package");
|
// 实体类名称
|
String entityName = (String) objectMap.get("entity");
|
// 设置Form类相关
|
FormTemplate formTemplate = new FormTemplate();
|
// 设置Form名称
|
formTemplate.setFormName(entityName + "Form");
|
// 设置Form路径
|
formTemplate.setFormPath(packageMap.get("Parent") + packageMap.get("ModuleName") + ".sys.domain.form");
|
// 设置form的字段，排除entity的一些字段即可
|
|
List<TableColumnVO> formColumns = data.getColumns().stream()
|
.filter(item -> !FormTemplate.removeFields.contains(item.getPropertyName()))
|
.collect(Collectors.toList());
|
formTemplate.setFields(formColumns);
|
// 设置form的父类名
|
formTemplate.setBaseFormName("AbsForm");
|
formTemplate.setBaseFormPath("com.monkeylessey.sys.domain.base");
|
objectMap.put("formInfo", formTemplate);
|
return new VoHandler().handler(objectMap, data);
|
}
|
}
|
