代码生成工具超多字段的时候使用

modules/core/src/test/java/com/jeesite/test/GenTableToolTest.java

@@ -0,0 +1,69 @@
+/**
+ * Copyright (c) 2013-Now http://jeesite.com All rights reserved.
+ */
+package com.jeesite.test;
+
+import org.junit.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.annotation.Rollback;
+import org.springframework.test.context.ActiveProfiles;
+
+import com.jeesite.common.config.Global;
+import com.jeesite.common.lang.StringUtils;
+import com.jeesite.common.tests.BaseSpringContextTests;
+import com.jeesite.modules.gen.entity.GenTable;
+import com.jeesite.modules.gen.entity.GenTableColumn;
+import com.jeesite.modules.gen.service.GenTableService;
+
+/**
+ * 代码生成工具（API）
+ * @author ThinkGem
+ * @version 2020-2-1
+ */
+@ActiveProfiles("test")
+@SpringBootTest(classes=ApplicationTest.class)
+@Rollback(false)
+public class GenTableToolTest extends BaseSpringContextTests {
+
+	@Autowired
+	private GenTableService genTableService;
+	
+	@Test
+	public void execute() throws Exception{
+		GenTable genTable = new GenTable();
+		genTable.setIsNewRecord(true);
+		genTable.setTableName("test_data_demo"); // 表名
+		genTable = genTableService.getFromDb(genTable);
+		genTable.setClassName(StringUtils.capCamelCase(genTable.getTableName())); // 实体类名
+		genTable.setFunctionAuthor("ThinkGem");	// 作者名称
+		genTable.setTplCategory("crud");		// 生成模板：crud、treeGrid、service、dao、query
+		genTable.setPackageName("com.jeesite.modules");// 生成包路径
+		genTable.setModuleName("test");			// 生成模块名
+		genTable.setSubModuleName("");			// 生成子模块名
+		genTable.setFunctionName(genTable.getComments()); 				// 生成功能名
+		genTable.setFunctionNameSimple(genTable.getComments()); 		// 功能名（简称）
+		genTable.getOptionMap().put("isHaveDisableEnable", Global.NO);	// 是否有停用启用
+		genTable.getOptionMap().put("isHaveDelete", Global.YES);		// 是否有删除功能
+		genTable.getOptionMap().put("isFileUpload", Global.NO);			// 是否有上传附件
+		genTable.getOptionMap().put("isImageUpload", Global.NO);		// 是否有上传图片
+		for(GenTableColumn column : genTable.getColumnList()){
+			// 字段配置（下拉框）
+			if ("trade_type".equals(column.getColumnName())){
+				column.setShowType("select");
+				column.getOptionMap().put("dictType", "sys_menu_type");
+			}
+		}
+		// 删除旧配置
+		genTableService.delete(genTable);
+		// 保存新配置
+		genTable.setIsNewRecord(true);
+		genTableService.save(genTable);
+		// 编译并生成代码
+		genTable.setGenFlag("1"); // 1：编译输出；2：生成文件
+		genTable.setReplaceFile(Global.YES); // 如果存在则替换
+		String result = genTableService.generateCode(genTable);
+		System.out.println(result);
+	}
+	
+}