733cae0982e0a38bc2fb99157532c27f9074ae0f
[holmes/rule-management.git] / rulemgt / src / main / java / org / onap / holmes / rulemgt / bolt / enginebolt / EngineWrapper.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.rulemgt.bolt.enginebolt;\r
17 \r
18 import javax.inject.Inject;\r
19 import javax.ws.rs.core.Response;\r
20 import lombok.extern.slf4j.Slf4j;\r
21 import net.sf.json.JSONObject;\r
22 import org.jvnet.hk2.annotations.Service;\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 import org.onap.holmes.common.exception.CorrelationException;\r
27 \r
28 @Service\r
29 @Slf4j\r
30 public class EngineWrapper {\r
31 \r
32     @Inject\r
33     private EngineService engineService;\r
34 \r
35     public String deployEngine(CorrelationDeployRule4Engine correlationRule) throws CorrelationException {\r
36         Response response;\r
37         try {\r
38             response = engineService.deploy(correlationRule);\r
39         } catch (Exception e) {\r
40             throw new CorrelationException("Failed to call the rule deployment RESTful API.", e);\r
41         }\r
42         if (response.getStatus() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
43             log.info("Succeeded in calling the rule deployment RESTful API from the engine management service.");\r
44             try {\r
45                 JSONObject json = JSONObject.fromObject(response.readEntity(String.class));\r
46                 return json.get(RuleMgtConstant.PACKAGE).toString();\r
47             } catch (Exception e) {\r
48                 throw new CorrelationException("Failed to parse the value returned by the engine management service.", e);\r
49             }\r
50         } else {\r
51             throw new CorrelationException("Failed to deploy the rule!");\r
52         }\r
53     }\r
54 \r
55     public boolean deleteRuleFromEngine(String packageName) throws CorrelationException {\r
56         Response response;\r
57         try {\r
58             response = engineService.delete(packageName);\r
59         } catch (Exception e) {\r
60             throw new CorrelationException("Failed to call the rule deleting RESTful API.", e);\r
61         }\r
62         if (response.getStatus() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
63             log.info("Succeeded in calling the rule deleting RESTful API from the engine management service.");\r
64             return true;\r
65         } else {\r
66             throw new CorrelationException("Failed to delete the rule!");\r
67         }\r
68     }\r
69 \r
70     public boolean checkRuleFromEngine(CorrelationCheckRule4Engine correlationCheckRule4Engine)\r
71             throws CorrelationException {\r
72         log.info("Rule Contents: " + correlationCheckRule4Engine.getContent());\r
73         Response response;\r
74         try {\r
75             response = engineService.check(correlationCheckRule4Engine);\r
76         } catch (Exception e) {\r
77             throw new CorrelationException("Failed to call the rule verification RESTful API.", e);\r
78         }\r
79         if (response.getStatus() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
80             log.info("Succeeded in calling the rule verification RESTful API from the engine management service.");\r
81             return true;\r
82         } else {\r
83             log.info(response.getStatus() + " " + response.getStatusInfo() + " " + response.getEntity());\r
84             throw new CorrelationException("Failed to verify the rule. The contents of the rule are invalid.");\r
85         }\r
86     }\r
87 }\r