From 12ddbc174a7b5dccd8189d2c4404104657ab4086 Mon Sep 17 00:00:00 2001
From: o2sword <171715986@qq.com>
Date: Mon, 31 Aug 2020 17:03:35 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BA=94=E7=94=A8=E5=B8=82=E5=9C=BA=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../center/jaxrs/market/ActionUninstall.java  | 48 ++++++++++++++++---
 1 file changed, 42 insertions(+), 6 deletions(-)

diff --git a/o2server/x_program_center/src/main/java/com/x/program/center/jaxrs/market/ActionUninstall.java b/o2server/x_program_center/src/main/java/com/x/program/center/jaxrs/market/ActionUninstall.java
index 05bbdf7642..563e25132a 100644
--- a/o2server/x_program_center/src/main/java/com/x/program/center/jaxrs/market/ActionUninstall.java
+++ b/o2server/x_program_center/src/main/java/com/x/program/center/jaxrs/market/ActionUninstall.java
@@ -5,14 +5,17 @@ import com.x.base.core.container.factory.EntityManagerContainerFactory;
 import com.x.base.core.entity.enums.CommonStatus;
 import com.x.base.core.project.*;
 import com.x.base.core.project.config.Config;
+import com.x.base.core.project.config.Nodes;
 import com.x.base.core.project.connection.CipherConnectionAction;
 import com.x.base.core.project.exception.ExceptionAccessDenied;
 import com.x.base.core.project.exception.ExceptionEntityNotExist;
+import com.x.base.core.project.gson.XGsonBuilder;
 import com.x.base.core.project.http.ActionResult;
 import com.x.base.core.project.http.EffectivePerson;
 import com.x.base.core.project.jaxrs.WrapBoolean;
 import com.x.base.core.project.logger.Logger;
 import com.x.base.core.project.logger.LoggerFactory;
+import com.x.base.core.project.tools.Crypto;
 import com.x.cms.core.entity.element.wrap.WrapCms;
 import com.x.portal.core.entity.wrap.WrapPortal;
 import com.x.processplatform.core.entity.element.wrap.WrapProcessPlatform;
@@ -24,8 +27,15 @@ import com.x.program.center.core.entity.wrap.WrapAgent;
 import com.x.program.center.core.entity.wrap.WrapInvoke;
 import com.x.program.center.core.entity.wrap.WrapServiceModule;
 import com.x.query.core.entity.wrap.WrapQuery;
+import org.apache.commons.lang3.StringUtils;
 
+import java.io.ByteArrayInputStream;
+import java.io.DataInputStream;
+import java.io.DataOutputStream;
+import java.net.Socket;
 import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
 
 class ActionUninstall extends BaseAction {
 
@@ -51,13 +61,15 @@ class ActionUninstall extends BaseAction {
 			WrapModule module = installData.getWrapModule();
 			if(module!=null) {
 				this.uninstall(module);
-
-				emc.beginTransaction(InstallLog.class);
-				installLog.setStatus(CommonStatus.INVALID.getValue());
-				installLog.setUnInstallPerson(effectivePerson.getDistinguishedName());
-				installLog.setUnInstallTime(new Date());
-				emc.commit();
 			}
+			if(StringUtils.isNotEmpty(installData.getCustomApp())){
+				this.uninstallCustomApp(installData.getCustomApp());
+			}
+			emc.beginTransaction(InstallLog.class);
+			installLog.setStatus(CommonStatus.INVALID.getValue());
+			installLog.setUnInstallPerson(effectivePerson.getDistinguishedName());
+			installLog.setUnInstallTime(new Date());
+			emc.commit();
 			wo.setValue(true);
 			result.setData(wo);
 			return result;
@@ -132,6 +144,30 @@ class ActionUninstall extends BaseAction {
 
 	}
 
+	private void uninstallCustomApp(String fileName) throws Exception{
+		Nodes nodes = Config.nodes();
+		for (String node : nodes.keySet()){
+			if(nodes.get(node).getApplication().getEnable()) {
+				logger.print("socket uninstall custom app{} to {}:{}",fileName, node, nodes.get(node).nodeAgentPort());
+				try (Socket socket = new Socket(node, nodes.get(node).nodeAgentPort())) {
+					socket.setKeepAlive(true);
+					socket.setSoTimeout(5000);
+					try (DataOutputStream dos = new DataOutputStream(socket.getOutputStream());
+						 DataInputStream dis = new DataInputStream(socket.getInputStream())) {
+						Map<String, Object> commandObject = new HashMap<>();
+						commandObject.put("command", "uninstall:customWar");
+						commandObject.put("credential", Crypto.rsaEncrypt("o2@", Config.publicKey()));
+
+						dos.writeUTF(XGsonBuilder.toJson(commandObject));
+						dos.flush();
+						dos.writeUTF(fileName);
+						dos.flush();
+					}
+				}
+			}
+		}
+	}
+
 	public static class Wo extends WrapBoolean {
 
 	}
-- 
GitLab