From e52452de27673c6d794fc2d8ae96f33abc25dc0c Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期日, 29 三月 2020 12:11:23 +0800
Subject: [PATCH] 调整代码生成器

---
 java110-code-generator/src/main/java/com/java110/code/back/BackCoderGeneratorStart.java |   20 +++++++++++++++-----
 1 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/java110-code-generator/src/main/java/com/java110/code/back/BackCoderGeneratorStart.java b/java110-code-generator/src/main/java/com/java110/code/back/BackCoderGeneratorStart.java
index 771e129..0467fbc 100644
--- a/java110-code-generator/src/main/java/com/java110/code/back/BackCoderGeneratorStart.java
+++ b/java110-code-generator/src/main/java/com/java110/code/back/BackCoderGeneratorStart.java
@@ -1,6 +1,7 @@
 package com.java110.code.back;
 
 
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.java110.code.back.Data;
 import com.java110.code.back.GeneratorAbstractBussiness;
@@ -15,8 +16,7 @@
 import com.java110.code.back.GeneratorUpdateInfoListener;
 import com.java110.code.web.GeneratorStart;
 
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
 
 /**
  * Hello world!
@@ -30,14 +30,15 @@
 
     /**
      * 浠g爜鐢熸垚鍣� 鍏ュ彛鏂规硶
-     *  姝ゅ鐢熸垚鐨刴apper鏂囦欢鍖呭惈杩囩▼琛ㄥ拰瀹炰緥琛ㄧ殑sql,鎵�浠ヨ姹備袱寮犺〃鐨勭壒娈婂瓧娈典篃瑕佸啓涓�
-     *   BusinessTypeCd
+     * 姝ゅ鐢熸垚鐨刴apper鏂囦欢鍖呭惈杩囩▼琛ㄥ拰瀹炰緥琛ㄧ殑sql,鎵�浠ヨ姹備袱寮犺〃鐨勭壒娈婂瓧娈典篃瑕佸啓涓�
+     * BusinessTypeCd
+     *
      * @param args 鍙傛暟
      */
     public static void main(String[] args) {
 
         //鍔犺浇閰嶇疆
-        StringBuffer sb = readFile(GeneratorStart.class.getResource("/web/template_1.json").getFile());
+        StringBuffer sb = readFile(GeneratorStart.class.getResource("/back/template_1.json").getFile());
 
         JSONObject dataJson = JSONObject.parseObject(sb.toString());
 
@@ -58,6 +59,15 @@
         data.setTableName(dataJson.getString("tableName"));
         data.setParams(dataJson.getJSONObject("param"));
 
+        JSONArray required = dataJson.getJSONArray("required");
+
+        List<String> paramList = new ArrayList<String>();
+        for (int reqIndex = 0; reqIndex < required.size(); reqIndex++) {
+            JSONObject require = required.getJSONObject(reqIndex);
+            paramList.add(require.getString("code"));
+        }
+        data.setRequiredParam(paramList.toArray(new String[required.size()]));
+
         GeneratorSaveInfoListener generatorSaveInfoListener = new GeneratorSaveInfoListener();
         generatorSaveInfoListener.generator(data);
 

--
Gitblit v1.8.0