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

   1 /**\r
   2  * Copyright 2017-2021 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.jvnet.hk2.annotations.Service;\r
  22 import org.onap.holmes.common.exception.CorrelationException;\r
  23 import org.onap.holmes.rulemgt.bean.request.CorrelationCheckRule4Engine;\r
  24 import org.onap.holmes.rulemgt.bean.request.CorrelationDeployRule4Engine;\r
  25 import org.onap.holmes.rulemgt.constant.RuleMgtConstant;\r
  26 \r
  27 import javax.inject.Inject;\r
  28 \r
  29 @Service\r
  30 @Slf4j\r
  31 public class EngineWrapper {\r
  32 \r
  33     @Inject\r
  34     private EngineService engineService;\r
  35 \r
  36     public String deployEngine(CorrelationDeployRule4Engine correlationRule, String ip) throws CorrelationException {\r
  37         String response = engineService.deploy(correlationRule, ip);\r
  38         if (response != null) {\r
  39             try {\r
  40                 JsonObject json = JsonParser.parseString(response).getAsJsonObject();\r
  41                 return json.get(RuleMgtConstant.PACKAGE).getAsString();\r
  42             } catch (Exception e) {\r
  43                 throw new CorrelationException("Failed to parse the value returned by the engine management service.", e);\r
  44             }\r
  45         } else {\r
  46             throw new CorrelationException("Failed to deploy the rule!");\r
  47         }\r
  48     }\r
  49 \r
  50     public boolean deleteRuleFromEngine(String packageName, String ip) throws CorrelationException {\r
  51         if (engineService.delete(packageName, ip)) {\r
  52             return true;\r
  53         } else {\r
  54             throw new CorrelationException("Failed to delete the rule!");\r
  55         }\r
  56     }\r
  57 \r
  58     public boolean checkRuleFromEngine(CorrelationCheckRule4Engine correlationCheckRule4Engine, String ip)\r
  59             throws CorrelationException {\r
  60         log.info("Rule Contents: " + correlationCheckRule4Engine.getContent());\r
  61         if (!engineService.check(correlationCheckRule4Engine, ip)) {\r
  62             throw new CorrelationException("Failed to verify the rule. The contents of the rule are invalid.");\r
  63         }\r
  64         return true;\r
  65     }\r
  66 }\r