2 * Copyright 2017 ZTE Corporation.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package org.openo.holmes.engine.resources;
19 import com.codahale.metrics.annotation.Timed;
20 import io.swagger.annotations.Api;
21 import io.swagger.annotations.ApiOperation;
22 import java.io.IOException;
23 import java.util.Locale;
24 import javax.annotation.PostConstruct;
25 import javax.inject.Inject;
26 import javax.servlet.http.HttpServletRequest;
27 import javax.ws.rs.DELETE;
28 import javax.ws.rs.POST;
29 import javax.ws.rs.PUT;
30 import javax.ws.rs.Path;
31 import javax.ws.rs.PathParam;
32 import javax.ws.rs.Produces;
33 import javax.ws.rs.core.Context;
34 import javax.ws.rs.core.MediaType;
35 import lombok.extern.slf4j.Slf4j;
36 import org.jvnet.hk2.annotations.Service;
37 import org.openo.holmes.common.api.entity.ServiceRegisterEntity;
38 import org.openo.holmes.common.config.MicroServiceConfig;
39 import org.openo.holmes.common.exception.CorrelationException;
40 import org.openo.holmes.common.utils.ExceptionUtil;
41 import org.openo.holmes.common.utils.LanguageUtil;
42 import org.openo.holmes.common.utils.MSBRegisterUtil;
43 import org.openo.holmes.engine.manager.DroolsEngine;
44 import org.openo.holmes.engine.request.CompileRuleRequest;
45 import org.openo.holmes.engine.request.DeployRuleRequest;
46 import org.openo.holmes.engine.response.CorrelationRuleResponse;
50 @Api(tags = {"Engine Manager"})
51 @Produces(MediaType.APPLICATION_JSON)
53 public class EngineResources {
56 DroolsEngine droolsEngine;
59 MSBRegisterUtil msbRegisterUtil;
62 private void initMsbService() {
64 msbRegisterUtil.register(initServiceEntity());
65 } catch (IOException e) {
66 log.warn("Failed register msb" + e.getMessage(), e);
71 @ApiOperation(value = "Add rule to Engine and Cache", response = CorrelationRuleResponse.class)
72 @Produces(MediaType.APPLICATION_JSON)
74 public CorrelationRuleResponse deployRule(DeployRuleRequest deployRuleRequest,
75 @Context HttpServletRequest httpRequest) {
77 CorrelationRuleResponse crResponse = new CorrelationRuleResponse();
78 Locale locale = LanguageUtil.getLocale(httpRequest);
81 String packageName = droolsEngine.deployRule(deployRuleRequest, locale);
82 crResponse.setPackageName(packageName);
84 } catch (CorrelationException correlationException) {
85 log.error(correlationException.getMessage(), correlationException);
86 throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage());
93 @ApiOperation(value = "delete rule")
94 @Produces(MediaType.APPLICATION_JSON)
96 @Path("/{packageName}")
97 public boolean undeployRule(@PathParam("packageName") String packageName,
98 @Context HttpServletRequest httpRequest) {
100 Locale locale = LanguageUtil.getLocale(httpRequest);
104 droolsEngine.undeployRule(packageName, locale);
106 } catch (CorrelationException correlationException) {
107 log.error(correlationException.getMessage(), correlationException);
108 throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage());
115 @ApiOperation(value = "compile rule")
116 @Produces(MediaType.APPLICATION_JSON)
118 public boolean compileRule(CompileRuleRequest compileRuleRequest,
119 @Context HttpServletRequest httpRequest) {
121 Locale locale = LanguageUtil.getLocale(httpRequest);
124 droolsEngine.compileRule(compileRuleRequest.getContent(), locale);
125 } catch (CorrelationException correlationException) {
126 log.error(correlationException.getMessage(), correlationException);
127 throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage());
132 private ServiceRegisterEntity initServiceEntity() {
133 ServiceRegisterEntity serviceRegisterEntity = new ServiceRegisterEntity();
134 serviceRegisterEntity.setServiceName("holmes");
135 serviceRegisterEntity.setProtocol("REST");
136 serviceRegisterEntity.setVersion("v1");
137 serviceRegisterEntity.setUrl("/api/holmes/v1");
138 serviceRegisterEntity.setSingleNode(MicroServiceConfig.getServiceIp(), "9102", 0);
139 serviceRegisterEntity.setVisualRange("1");
140 return serviceRegisterEntity;