b0bd1f52aed9b287d851fd9c128b891cf82e071a
[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 com.alibaba.fastjson.JSON;\r
19 import com.alibaba.fastjson.JSONObject;\r
20 import javax.inject.Inject;\r
21 import lombok.extern.slf4j.Slf4j;\r
22 import org.apache.http.HttpResponse;\r
23 import org.jvnet.hk2.annotations.Service;\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 import org.onap.holmes.common.exception.CorrelationException;\r
29 \r
30 @Service\r
31 @Slf4j\r
32 public class EngineWrapper {\r
33 \r
34     @Inject\r
35     private EngineService engineService;\r
36 \r
37     public String deployEngine(CorrelationDeployRule4Engine correlationRule,String ip) throws CorrelationException {\r
38         HttpResponse response;\r
39         try {\r
40             response = engineService.deploy(correlationRule, ip);\r
41         } catch (Exception e) {\r
42             throw new CorrelationException("Failed to call the rule deployment RESTful API.", e);\r
43         }\r
44         if (response.getStatusLine().getStatusCode() == RuleMgtConstant.RESPONSE_STATUS_OK) {\r
45             log.info("Succeeded in calling the rule deployment RESTful API from the engine management service.");\r
46             try {\r
47                // JSONObject json = JSONObject.fromObject(HttpsUtils.extractResponseEntity(response));\r
48                 JSONObject json=  JSON.parseObject(HttpsUtils.extractResponseEntity(response));\r
49                 return json.get(RuleMgtConstant.PACKAGE).toString();\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