engine-d/src/main/java/org/openo/holmes/engine/resources/EngineResources.java

   1 /**
   2  * Copyright 2017 ZTE Corporation.
   3  *
   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
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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.
  15  */
  16 package org.openo.holmes.engine.resources;
  17
  18
  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;
  47
  48 @Service
  49 @Path("/rule")
  50 @Api(tags = {"Engine Manager"})
  51 @Produces(MediaType.APPLICATION_JSON)
  52 @Slf4j
  53 public class EngineResources {
  54
  55     @Inject
  56     DroolsEngine droolsEngine;
  57
  58     @Inject
  59     MSBRegisterUtil msbRegisterUtil;
  60
  61     @PostConstruct
  62     private void initMsbService() {
  63         try {
  64             msbRegisterUtil.register(initServiceEntity());
  65         } catch (IOException e) {
  66             log.warn("Failed register msb" + e.getMessage(), e);
  67         }
  68     }
  69
  70     @PUT
  71     @ApiOperation(value = "Add rule to Engine and Cache", response = CorrelationRuleResponse.class)
  72     @Produces(MediaType.APPLICATION_JSON)
  73     @Timed
  74     public CorrelationRuleResponse deployRule(DeployRuleRequest deployRuleRequest,
  75         @Context HttpServletRequest httpRequest) {
  76
  77         CorrelationRuleResponse crResponse = new CorrelationRuleResponse();
  78         Locale locale = LanguageUtil.getLocale(httpRequest);
  79         try {
  80
  81             String packageName = droolsEngine.deployRule(deployRuleRequest, locale);
  82             crResponse.setPackageName(packageName);
  83
  84         } catch (CorrelationException correlationException) {
  85             log.error(correlationException.getMessage(), correlationException);
  86             throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage());
  87         }
  88
  89         return crResponse;
  90     }
  91
  92     @DELETE
  93     @ApiOperation(value = "delete rule")
  94     @Produces(MediaType.APPLICATION_JSON)
  95     @Timed
  96     @Path("/{packageName}")
  97     public boolean undeployRule(@PathParam("packageName") String packageName,
  98         @Context HttpServletRequest httpRequest) {
  99
 100         Locale locale = LanguageUtil.getLocale(httpRequest);
 101
 102         try {
 103
 104             droolsEngine.undeployRule(packageName, locale);
 105
 106         } catch (CorrelationException correlationException) {
 107             log.error(correlationException.getMessage(), correlationException);
 108             throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage());
 109         }
 110         return true;
 111     }
 112
 113
 114     @POST
 115     @ApiOperation(value = "compile rule")
 116     @Produces(MediaType.APPLICATION_JSON)
 117     @Timed
 118     public boolean compileRule(CompileRuleRequest compileRuleRequest,
 119         @Context HttpServletRequest httpRequest) {
 120
 121         Locale locale = LanguageUtil.getLocale(httpRequest);
 122
 123         try {
 124             droolsEngine.compileRule(compileRuleRequest.getContent(), locale);
 125         } catch (CorrelationException correlationException) {
 126             log.error(correlationException.getMessage(), correlationException);
 127             throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage());
 128         }
 129         return true;
 130     }
 131
 132     private ServiceRegisterEntity initServiceEntity() {
 133         ServiceRegisterEntity serviceRegisterEntity = new ServiceRegisterEntity();
 134         serviceRegisterEntity.setServiceName("holmes-engine");
 135         serviceRegisterEntity.setProtocol("REST");
 136         serviceRegisterEntity.setVersion("v1");
 137         serviceRegisterEntity.setUrl("/api/holmes-engine/v1");
 138         serviceRegisterEntity.setSingleNode(MicroServiceConfig.getServiceIp(), "9102", 0);
 139         serviceRegisterEntity.setVisualRange("1");
 140         return serviceRegisterEntity;
 141     }
 142 }