From 37b3bb9b015069ba5e9ad1845de30467decb61d4 Mon Sep 17 00:00:00 2001 From: Shiwei Tian Date: Mon, 9 Apr 2018 09:32:39 +0800 Subject: fix https bug Issue-ID: HOLMES-104 Change-Id: I6922584f94aa0ec79fda1e2b3dcc7da6c874c9da Signed-off-by: Shiwei Tian --- .../rulemgt/bolt/enginebolt/EngineService.java | 36 +++++++++++++++++++-- .../rulemgt/dcae/DcaeConfigurationPolling.java | 37 +++++++++++++++++++--- .../org/onap/holmes/rulemgt/msb/EngineIpList.java | 16 +++++++++- 3 files changed, 80 insertions(+), 9 deletions(-) (limited to 'rulemgt/src/main/java/org/onap') diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineService.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineService.java index 13507d6..aa0bf32 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineService.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineService.java @@ -15,11 +15,13 @@ */ package org.onap.holmes.rulemgt.bolt.enginebolt; +import java.io.IOException; import java.util.HashMap; import javax.ws.rs.core.MediaType; import lombok.extern.slf4j.Slf4j; import org.apache.http.HttpResponse; import org.apache.http.entity.StringEntity; +import org.apache.http.impl.client.CloseableHttpClient; import org.jvnet.hk2.annotations.Service; import org.onap.holmes.common.utils.GsonUtil; import org.onap.holmes.common.utils.HttpsUtils; @@ -38,7 +40,13 @@ public class EngineService { protected HttpResponse delete(String packageName, String ip) throws Exception { HashMap headers = createHeaders(); String url = PREFIX + ip + PORT + RuleMgtConstant.ENGINE_PATH + "/" + packageName; - return HttpsUtils.delete(url, headers); + CloseableHttpClient httpClient = null; + try { + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); + return HttpsUtils.delete(url, headers, httpClient); + } finally { + closeHttpClient(httpClient); + } } protected HttpResponse check(CorrelationCheckRule4Engine correlationCheckRule4Engine, String ip) @@ -46,14 +54,36 @@ public class EngineService { String content = GsonUtil.beanToJson(correlationCheckRule4Engine); HashMap headers = createHeaders(); String url = PREFIX + ip + PORT + RuleMgtConstant.ENGINE_PATH; - return HttpsUtils.post(url, headers, new HashMap<>(), new StringEntity(content)); + CloseableHttpClient httpClient = null; + try { + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); + return HttpsUtils.post(url, headers, new HashMap<>(), new StringEntity(content), httpClient); + } finally { + closeHttpClient(httpClient); + } } protected HttpResponse deploy(CorrelationDeployRule4Engine correlationDeployRule4Engine, String ip) throws Exception { String content = GsonUtil.beanToJson(correlationDeployRule4Engine); HashMap headers = createHeaders(); String url = PREFIX + ip + PORT + RuleMgtConstant.ENGINE_PATH; - return HttpsUtils.put(url, headers, new HashMap<>(), new StringEntity(content)); + CloseableHttpClient httpClient = null; + try { + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); + return HttpsUtils.put(url, headers, new HashMap<>(), new StringEntity(content),httpClient); + } finally { + closeHttpClient(httpClient); + } + } + + private void closeHttpClient(CloseableHttpClient httpClient) { + if (httpClient != null) { + try { + httpClient.close(); + } catch (IOException e) { + log.warn("Failed to close http client!"); + } + } } private HashMap createHeaders() { diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/dcae/DcaeConfigurationPolling.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/dcae/DcaeConfigurationPolling.java index e8fa8b0..d1b2aba 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/dcae/DcaeConfigurationPolling.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/dcae/DcaeConfigurationPolling.java @@ -30,6 +30,7 @@ import javax.ws.rs.core.MediaType; import lombok.extern.slf4j.Slf4j; import org.apache.http.HttpResponse; import org.apache.http.entity.StringEntity; +import org.apache.http.impl.client.CloseableHttpClient; import org.onap.holmes.common.dcae.DcaeConfigurationQuery; import org.onap.holmes.common.dcae.entity.DcaeConfigurations; import org.onap.holmes.common.dcae.entity.Rule; @@ -83,6 +84,8 @@ public class DcaeConfigurationPolling implements Runnable { log.error("Failed to get right response!" + e.getMessage(), e); } catch (IOException e) { log.error("Failed to extract response entity. " + e.getMessage(), e); + } catch (Exception e) { + log.error("Failed to build http client. " + e.getMessage(), e); } } if (ruleQueryListResponse != null) { @@ -100,9 +103,15 @@ public class DcaeConfigurationPolling implements Runnable { public RuleQueryListResponse getAllCorrelationRules() throws CorrelationException, IOException { HashMap headers = new HashMap<>(); headers.put("Content-Type", MediaType.APPLICATION_JSON); - HttpResponse httpResponse = HttpsUtils.get(url, headers); - String response = HttpsUtils.extractResponseEntity(httpResponse); - return JSON.parseObject(response,RuleQueryListResponse.class); + CloseableHttpClient httpClient = null; + try { + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); + HttpResponse httpResponse = HttpsUtils.get(url, headers, httpClient); + String response = HttpsUtils.extractResponseEntity(httpResponse); + return JSON.parseObject(response,RuleQueryListResponse.class); + } finally { + closeHttpClient(httpClient); + } } private boolean addAllCorrelationRules(DcaeConfigurations dcaeConfigurations) throws CorrelationException { @@ -119,13 +128,17 @@ public class DcaeConfigurationPolling implements Runnable { headers.put("Content-Type", MediaType.APPLICATION_JSON); headers.put("Accept", MediaType.APPLICATION_JSON); HttpResponse httpResponse; + CloseableHttpClient httpClient = null; try { + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); httpResponse = HttpsUtils - .put(url, headers, new HashMap<>(), new StringEntity(content)); + .put(url, headers, new HashMap<>(), new StringEntity(content), httpClient); } catch (UnsupportedEncodingException e) { throw new CorrelationException("Failed to create https entity.", e); } catch (Exception e) { throw new CorrelationException(e.getMessage()); + } finally { + closeHttpClient(httpClient); } if (httpResponse != null) { suc = httpResponse.getStatusLine().getStatusCode() == 200; @@ -141,11 +154,15 @@ public class DcaeConfigurationPolling implements Runnable { ruleResult4APIs.forEach(correlationRule ->{ HashMap headers = new HashMap<>(); headers.put("Content-Type", MediaType.APPLICATION_JSON); + CloseableHttpClient httpClient = null; try { - HttpsUtils.delete(url + "/" + correlationRule.getRuleId(), headers); + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); + HttpsUtils.delete(url + "/" + correlationRule.getRuleId(), headers, httpClient); } catch (Exception e) { log.warn("Failed to delete rule, the rule id is : " + correlationRule.getRuleId() + " exception messge is : " + e.getMessage(), e); + } finally { + closeHttpClient(httpClient); } }); } @@ -159,4 +176,14 @@ public class DcaeConfigurationPolling implements Runnable { ruleCreateRequest.setEnabled(1); return ruleCreateRequest; } + + private void closeHttpClient(CloseableHttpClient httpClient) { + if (httpClient != null) { + try { + httpClient.close(); + } catch (IOException e) { + log.warn("Failed to close http client!"); + } + } + } } diff --git a/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/EngineIpList.java b/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/EngineIpList.java index cfccd18..992785f 100644 --- a/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/EngineIpList.java +++ b/rulemgt/src/main/java/org/onap/holmes/rulemgt/msb/EngineIpList.java @@ -16,7 +16,10 @@ package org.onap.holmes.rulemgt.msb; +import java.io.IOException; +import lombok.extern.slf4j.Slf4j; import org.apache.http.HttpResponse; +import org.apache.http.impl.client.CloseableHttpClient; import org.jvnet.hk2.annotations.Service; import org.onap.holmes.common.api.entity.ServiceEntity; import org.onap.holmes.common.api.entity.ServiceNode4Query; @@ -30,6 +33,7 @@ import java.util.HashMap; import java.util.List; @Service +@Slf4j public class EngineIpList { private String[] msbAddrInfo; @@ -47,12 +51,22 @@ public class EngineIpList { public List getServiceCount()throws Exception{ String response; + CloseableHttpClient httpClient = null; try { + httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT); HttpResponse httpResponse = HttpsUtils - .get(url, new HashMap<>()); + .get(url, new HashMap<>(), httpClient); response = HttpsUtils.extractResponseEntity(httpResponse); } catch (Exception e) { throw e; + } finally { + if (httpClient != null) { + try { + httpClient.close(); + } catch (IOException e) { + log.warn("Failed to close http client!"); + } + } } ServiceEntity service = GsonUtil.jsonToBean(response, ServiceEntity.class); List nodesList = service.getNodes(); -- cgit 1.2.3-korg