holmes-actions/src/main/java/org/onap/holmes/common/dmaap/Publisher.java

   1 /*\r
   2  * Copyright 2017 ZTE Corporation.\r
   3  *\r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  * http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 package org.onap.holmes.common.dmaap;\r
  17 \r
  18 import java.io.IOException;\r
  19 import lombok.extern.slf4j.Slf4j;\r
  20 import org.apache.http.client.methods.HttpEntityEnclosingRequestBase;\r
  21 import org.apache.http.client.methods.HttpPost;\r
  22 import org.apache.http.impl.client.CloseableHttpClient;\r
  23 import org.onap.holmes.common.dmaap.entity.PolicyMsg;\r
  24 import org.onap.holmes.common.exception.CorrelationException;\r
  25 import com.alibaba.fastjson.JSON;\r
  26 import com.google.gson.Gson;\r
  27 import java.util.HashMap;\r
  28 import javax.ws.rs.core.MediaType;\r
  29 import lombok.Getter;\r
  30 import lombok.Setter;\r
  31 import org.apache.http.HttpResponse;\r
  32 import org.apache.http.HttpStatus;\r
  33 import org.apache.http.entity.StringEntity;\r
  34 import org.jvnet.hk2.annotations.Service;\r
  35 import org.onap.holmes.common.utils.HttpsUtils;\r
  36 \r
  37 @Getter\r
  38 @Setter\r
  39 @Service\r
  40 @Slf4j\r
  41 public class Publisher {\r
  42 \r
  43     private String topic;\r
  44     private String url;\r
  45     private String authInfo;\r
  46     private String authExpDate;\r
  47 \r
  48     public boolean publish(PolicyMsg msg) throws CorrelationException {\r
  49         String content;\r
  50         try {\r
  51             //content = JSON.toJSONString(msg);\r
  52                 content = new Gson().toJson(msg);\r
  53         } catch (Exception e) {\r
  54             throw new CorrelationException("Failed to convert the message object to a json string.",\r
  55                     e);\r
  56         }\r
  57         HttpResponse httpResponse;\r
  58         HashMap<String, String> headers = new HashMap<>();\r
  59         headers.put("Accept", MediaType.APPLICATION_JSON);\r
  60         headers.put("Content-Type", MediaType.APPLICATION_JSON);\r
  61         CloseableHttpClient httpClient = null;\r
  62         HttpPost httpPost = new HttpPost(url);\r
  63         try {\r
  64             httpClient = HttpsUtils.getHttpClient(HttpsUtils.DEFUALT_TIMEOUT);\r
  65             httpResponse = HttpsUtils.post(httpPost, headers, new HashMap<>(), new StringEntity(content, "utf-8"), httpClient);\r
  66         } catch (Exception e) {\r
  67             throw new CorrelationException("Failed to connect to DCAE.", e);\r
  68         } finally {\r
  69             httpPost.releaseConnection();\r
  70             if (httpClient != null) {\r
  71                 try {\r
  72                     httpClient.close();\r
  73                 } catch (IOException e) {\r
  74                     log.warn("Failed to close http client!");\r
  75                 }\r
  76             }\r
  77         }\r
  78         return checkStatus(httpResponse);\r
  79     }\r
  80 \r
  81     private boolean checkStatus(HttpResponse httpResponse) {\r
  82         return (httpResponse.getStatusLine().getStatusCode() == HttpStatus.SC_OK) ? true : false;\r
  83     }\r
  84 }\r