From 4daec329f36cc9d86911c08b8ec8e00270792189 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期五, 19 十一月 2021 15:08:51 +0800
Subject: [PATCH] 自动化代码中的service优化

---
 src/main/java/com/ks/codegenerator/utils/ServiceBuilder.java |   46 ++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/ks/codegenerator/utils/ServiceBuilder.java b/src/main/java/com/ks/codegenerator/utils/ServiceBuilder.java
index 9381e47..3ae6de7 100644
--- a/src/main/java/com/ks/codegenerator/utils/ServiceBuilder.java
+++ b/src/main/java/com/ks/codegenerator/utils/ServiceBuilder.java
@@ -1,12 +1,19 @@
 package com.ks.codegenerator.utils;
 
+import org.w3c.dom.Document;
+import org.w3c.dom.Node;
+import org.w3c.dom.NodeList;
 import org.yeshi.utils.FileUtil;
 import org.yeshi.utils.StringUtil;
 
+import javax.xml.parsers.DocumentBuilder;
+import javax.xml.parsers.DocumentBuilderFactory;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.stream.StreamResult;
 import java.io.File;
-import java.io.IOException;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
 
 /**
@@ -66,7 +73,7 @@
         //淇敼鏂囦欢鍚嶇О涓庢枃浠跺す鍚嶇О
         renameDir(destPath);
         //閲嶆柊鍛藉悕鍖呭悕
-        renamePackage(destPath + "/app/src/main/java/com/ks/app");
+        renamePackage(destPath + "/app/src/main/java/" + subpks.replace(".", "/"));
         replaceSettings(destPath);
         //鍘嬬缉鏂囦欢澶�
         File zip = new File(new File(destPath).getParent(), new File(destPath).getName() + ".zip");
@@ -81,8 +88,35 @@
         FileUtils.replaceFileContent(path + "/pom.xml", "<groupId>com.ks</groupId>", String.format("<groupId>%s</groupId>", getGroupId(pks)));
         FileUtils.replaceFileContent(path + "/pom.xml", "<artifactId>demo</artifactId>", String.format("<artifactId>%s</artifactId>", getArtifactId(pks)));
 
-        FileUtils.replaceFileContent(path + "/app/pom.xml", "<groupId>com.ks</groupId>", String.format("<groupId>%s</groupId>", getGroupId(subpks)));
-        FileUtils.replaceFileContent(path + "/app/pom.xml", "<artifactId>demo</artifactId>", String.format("<artifactId>%s</artifactId>", getArtifactId(subpks)));
+
+        //鏇存敼parent浠ュ強artifactId
+        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
+        DocumentBuilder db = dbf.newDocumentBuilder();
+        Document doc = db.parse(path + "/app/pom.xml");
+        NodeList list = doc.getElementsByTagName("project").item(0).getChildNodes();
+        for (int i = 0; i < list.getLength(); i++) {
+            if (list.item(i).getNodeName().equalsIgnoreCase("parent")) {
+                NodeList parent = list.item(i).getChildNodes();
+                for (int j = 0; j < parent.getLength(); j++) {
+                    Node node = parent.item(j);
+                    switch (node.getNodeName()) {
+                        case "artifactId":
+                            node.setTextContent(getArtifactId(pks));
+                            break;
+                        case "groupId":
+                            node.setTextContent(getGroupId(pks));
+                            break;
+                    }
+                }
+            } else if (list.item(i).getNodeName().equalsIgnoreCase("artifactId")) {
+                list.item(i).setTextContent(getArtifactId(subpks));
+            }
+        }
+
+        //淇濆瓨
+        Transformer ts = TransformerFactory.newInstance().newTransformer();
+        ts.transform(new DOMSource(doc), new StreamResult(path + "/app/pom.xml"));
+
 
         FileUtils.replaceFileContent(path + "/app/src/main/resources/application-dev.yml", "com.ks.app", subpks);
         FileUtils.replaceFileContent(path + "/app/src/main/resources/application-pro.yml", "com.ks.app", subpks);
@@ -93,7 +127,7 @@
         FileUtils.renameFile(path + "/service.iml", name + ".iml");
         //鏇存敼java鏂囦欢鐨勮矾寰�
         FileUtils.copyFileDir(path + "/app/src/main/java/com/ks/app", path + "/app/src/main/java/" + subpks.replace(".", "/"));
-        new File(path + "/app/src/main/java/com/ks/app").delete();
+        FileUtil.deleteFileDir(new File(path + "/app/src/main/java/com/ks/app"));
     }
 
 

--
Gitblit v1.8.0