From ccc8ba1167f4b447ee4c19bee42b813c32f61027 Mon Sep 17 00:00:00 2001
From: dailidong <dailidong66@gmail.com>
Date: Thu, 22 Oct 2020 13:35:57 +0800
Subject: [PATCH] solve too many files, close logClientService (#3971)

---
 .../service/process/ProcessService.java       | 46 +++++++++++--------
 1 file changed, 26 insertions(+), 20 deletions(-)

diff --git a/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java b/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java
index 903f5cd2a1..9bc3fa73ed 100644
--- a/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java
+++ b/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java
@@ -307,31 +307,37 @@ public class ProcessService {
      */
     public void removeTaskLogFile(Integer processInstanceId){
 
-        LogClientService logClient = new LogClientService();
+        LogClientService logClient = null;
 
-        List<TaskInstance> taskInstanceList = findValidTaskListByProcessId(processInstanceId);
+        try {
+            logClient = new LogClientService();
+            List<TaskInstance> taskInstanceList = findValidTaskListByProcessId(processInstanceId);
 
-        if (CollectionUtils.isEmpty(taskInstanceList)){
-            return;
-        }
-
-        for (TaskInstance taskInstance : taskInstanceList){
-            String taskLogPath = taskInstance.getLogPath();
-            if (StringUtils.isEmpty(taskInstance.getHost())){
-                continue;
-            }
-            int port = Constants.RPC_PORT;
-            String ip = "";
-            try {
-                ip = Host.of(taskInstance.getHost()).getIp();
-            }catch (Exception e){
-                // compatible old version
-                ip = taskInstance.getHost();
+            if (CollectionUtils.isEmpty(taskInstanceList)) {
+                return;
             }
 
+            for (TaskInstance taskInstance : taskInstanceList) {
+                String taskLogPath = taskInstance.getLogPath();
+                if (StringUtils.isEmpty(taskInstance.getHost())) {
+                    continue;
+                }
+                int port = Constants.RPC_PORT;
+                String ip = "";
+                try {
+                    ip = Host.of(taskInstance.getHost()).getIp();
+                } catch (Exception e) {
+                    // compatible old version
+                    ip = taskInstance.getHost();
+                }
 
-            // remove task log from loggerserver
-            logClient.removeTaskLog(ip,port,taskLogPath);
+                // remove task log from loggerserver
+                logClient.removeTaskLog(ip, port, taskLogPath);
+            }
+        }finally {
+            if (logClient != null) {
+                logClient.close();
+            }
         }
     }