From d7910b2f4789fe6c76cb104e36edb8ccbb67d25e Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期五, 23 四月 2021 11:08:39 +0800
Subject: [PATCH] 优化地爱吗

---
 java110-generator/src/main/java/com/java110/code/back/GeneratorDtoBean.java |   75 +++++++++++++++++++++++++++++++++++++
 1 files changed, 75 insertions(+), 0 deletions(-)

diff --git a/java110-generator/src/main/java/com/java110/code/back/GeneratorDtoBean.java b/java110-generator/src/main/java/com/java110/code/back/GeneratorDtoBean.java
new file mode 100755
index 0000000..59d72f1
--- /dev/null
+++ b/java110-generator/src/main/java/com/java110/code/back/GeneratorDtoBean.java
@@ -0,0 +1,75 @@
+package com.java110.code.back;
+
+import com.java110.code.util.FileUtilBase;
+
+import java.util.Map;
+
+/**
+ *
+ */
+public class GeneratorDtoBean extends BaseGenerator {
+
+
+    /**
+     * 鎷艰 鏌ヨ鏁伴噺
+     *
+     * @param data        鏁版嵁
+     * @param fileContext 鏂囦欢鍐呭
+     * @return filContext 鏁版嵁
+     */
+    private String dealVariableAndGetSet(Data data, String fileContext) {
+
+        Map<String, String> params = data.getParams();
+
+        String variable = "";
+        String variableGetSet = "";
+
+        for (String key : params.keySet()) {
+            if ("operate".equals(key) || "bId".equals(key) || "statusCd".equals(key)|| "createTime".equals(key)) {
+                continue;
+            }
+            variable += "private String " + key + ";\n";
+
+            variableGetSet += "public String get" + toUpperCaseFirstOne(key) + "() {\n"
+                    + "        return " + key + ";\n"
+                    + "    }\n";
+            variableGetSet += "public void set" + toUpperCaseFirstOne(key) + "(String " + key + ") {\n"
+                    + "        this." + key + " = " + key + ";\n"
+                    + "    }\n";
+
+
+        }
+
+
+        fileContext = fileContext.replace("$beanVariable$", variable);
+        fileContext = fileContext.replace("$beanVariableGetSet$", variableGetSet);
+
+        return fileContext;
+    }
+
+
+    /**
+     * 鐢熸垚浠g爜
+     *
+     * @param data 鏁版嵁
+     */
+    public void generator(Data data) throws Exception {
+        StringBuffer sb = readFile(this.getClass().getResource("/template/dto.txt").getFile());
+        String fileContext = sb.toString();
+        fileContext = fileContext.replace("store", toLowerCaseFirstOne(data.getName()))
+                .replace("@@templateCode@@",data.getName())
+                .replace("Store", toUpperCaseFirstOne(data.getName()))
+                .replace("鍟嗘埛", data.getDesc());
+
+        fileContext = dealVariableAndGetSet(data, fileContext);
+        String writePath = this.getClass().getResource("/").getPath()
+                + "out/back/dto/" + data.getName() + "/" + toUpperCaseFirstOne(data.getName()) + "Dto.java";
+
+        writeFile(writePath,
+                fileContext);
+        //澶嶅埗鐢熸垚鐨勬枃浠跺埌瀵瑰簲鍒嗗尯鐩綍涓�
+        if (data.isAutoMove()) {
+            FileUtilBase.copyfile(writePath, "java110-bean\\src\\main\\java\\com\\java110\\dto\\" + data.getName() + "/" + toUpperCaseFirstOne(data.getName()) + "Dto.java");
+        }
+    }
+}

--
Gitblit v1.8.0