rulemgt/src/main/java/org/onap/holmes/rulemgt/bolt/enginebolt/EngineWrapper.java

   1 /**\r
   2  * Copyright 2017-2020 ZTE Corporation.\r
   3  * <p>\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  * <p>\r
   8  * http://www.apache.org/licenses/LICENSE-2.0\r
   9  * <p>\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.rulemgt.bolt.enginebolt;\r
  17 \r
  18 import com.google.gson.JsonObject;\r
  19 import com.google.gson.JsonParser;\r
  20 import lombok.extern.slf4j.Slf4j;\r
  21 import org.apache.http.HttpResponse;\r
  22 import org.jvnet.hk2.annotations.Service;\r
  23 import org.onap.holmes.common.exception.CorrelationException;\r
  24 import org.onap.holmes.common.utils.HttpsUtils;\r
  25 import org.onap.holmes.rulemgt.bean.request.CorrelationCheckRule4Engine;\r
  26 import org.onap.holmes.rulemgt.bean.request.CorrelationDeployRule4Engine;\r
  27 import org.onap.holmes.rulemgt.constant.RuleMgtConstant;\r
  28 \r
  29 import javax.inject.Inject;\r
  30 \r
  31 @Service\r
  32 @Slf4j\r
  33 public class EngineWrapper {\r
  34 \r
  35     @Inject\r
  36     private EngineService engineService;\r
  37 \r
  38     public String deployEngine(CorrelationDeployRule4Engine correlationRule, String ip) throws CorrelationException {\r
  39         HttpResponse response;\r
  40         try {\r
  41             response = engineService.deploy(correlationRule, ip);\r
  42         } catch (Exception e) {\r
  43             throw new CorrelationException("Failed to call the rule deployment RESTful API.", e);\r
  44         }\r
  45         if (response.getStatusLine().getStatusCode() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
  46             log.info("Succeeded in calling the rule deployment RESTful API from the engine management service.");\r
  47             try {\r
  48                 JsonObject json = JsonParser.parseString(HttpsUtils.extractResponseEntity(response)).getAsJsonObject();\r
  49                 return json.get(RuleMgtConstant.PACKAGE).getAsString();\r
  50             } catch (Exception e) {\r
  51                 throw new CorrelationException("Failed to parse the value returned by the engine management service.", e);\r
  52             }\r
  53         } else {\r
  54             throw new CorrelationException("Failed to deploy the rule!");\r
  55         }\r
  56     }\r
  57 \r
  58     public boolean deleteRuleFromEngine(String packageName, String ip) throws CorrelationException {\r
  59         HttpResponse response;\r
  60         try {\r
  61             response = engineService.delete(packageName, ip);\r
  62         } catch (Exception e) {\r
  63             throw new CorrelationException("Failed to call the rule deleting RESTful API.", e);\r
  64         }\r
  65         if (response.getStatusLine().getStatusCode() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
  66             log.info("Succeeded in calling the rule deleting RESTful API from the engine management service.");\r
  67             return true;\r
  68         } else {\r
  69             throw new CorrelationException("Failed to delete the rule!");\r
  70         }\r
  71     }\r
  72 \r
  73     public boolean checkRuleFromEngine(CorrelationCheckRule4Engine correlationCheckRule4Engine, String ip)\r
  74             throws CorrelationException {\r
  75         log.info("Rule Contents: " + correlationCheckRule4Engine.getContent());\r
  76         HttpResponse response;\r
  77         try {\r
  78             response = engineService.check(correlationCheckRule4Engine, ip);\r
  79         } catch (Exception e) {\r
  80             throw new CorrelationException("Failed to call the rule verification RESTful API.", e);\r
  81         }\r
  82         if (response.getStatusLine().getStatusCode() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
  83             log.info("Succeeded in calling the rule verification RESTful API from the engine management service.");\r
  84             return true;\r
  85         } else {\r
  86             log.info(response.getStatusLine().getStatusCode() + " " + response.getEntity());\r
  87             throw new CorrelationException("Failed to verify the rule. The contents of the rule are invalid.");\r
  88         }\r
  89     }\r
  90 }\r