Add engine

add engine

Change-Id: If66801ece7f9107486100626a9eec81ba75c5054
Issue-ID:HOLMES-19
Signed-off-by: FengLiang <feng.liang1@zte.com.cn>

12 files changed, 855 insertions, 0 deletions

diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/EngineActiveApp.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/EngineActiveApp.java
new file mode 100644
index 0000000..ddaed97
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/EngineActiveApp.java
@@ -0,0 +1,25 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt;
+
+import org.openo.dropwizard.ioc.bundle.IOCApplication;
+
+public class EngineActiveApp extends IOCApplication<EngineAppConfig> {
+
+    public static void main(String[] args) throws Exception {
+        new EngineActiveApp().run(args);
+    }
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/EngineAppConfig.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/EngineAppConfig.java
new file mode 100644
index 0000000..b01d173
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/EngineAppConfig.java
@@ -0,0 +1,68 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import io.dropwizard.Configuration;
+import io.dropwizard.db.DataSourceFactory;
+import javax.validation.Valid;
+import javax.validation.constraints.NotNull;
+import lombok.Getter;
+import org.hibernate.validator.constraints.NotEmpty;
+import org.openo.holmes.common.config.MQConfig;
+
+public class EngineAppConfig extends Configuration {
+
+    @NotEmpty
+    private String defaultName = "Correlation-Rule";
+
+    @NotEmpty
+    private String apidescription = "Holmes rule management rest API";
+
+    @JsonProperty
+    @NotNull
+    @Valid
+    private MQConfig mqConfig;
+    @Valid
+    @NotNull
+    private DataSourceFactory database = new DataSourceFactory();
+
+    public MQConfig getMqConfig() {
+        return mqConfig;
+    }
+
+    public void setMqConfig(MQConfig mqConfig) {
+        this.mqConfig = mqConfig;
+    }
+
+    @JsonProperty("database")
+    public DataSourceFactory getDataSourceFactory() {
+        return database;
+    }
+
+    @JsonProperty("database")
+    public void setDataSourceFactory(DataSourceFactory dataSourceFactory) {
+        this.database = dataSourceFactory;
+    }
+
+    public String getApidescription() {
+        return apidescription;
+    }
+
+    public void setApidescription(String apidescription) {
+        this.apidescription = apidescription;
+    }
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/db/CorrelationRuleDao.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/db/CorrelationRuleDao.java
new file mode 100644
index 0000000..2393417
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/db/CorrelationRuleDao.java
@@ -0,0 +1,37 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.db;
+
+
+import java.util.List;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.openo.holmes.enginemgt.db.mapper.CorrelationRuleMapper;
+import org.skife.jdbi.v2.sqlobject.Bind;
+import org.skife.jdbi.v2.sqlobject.SqlQuery;
+import org.skife.jdbi.v2.sqlobject.customizers.RegisterMapper;
+
+@RegisterMapper(CorrelationRuleMapper.class)
+public abstract class CorrelationRuleDao {
+
+
+    @SqlQuery("SELECT * FROM APLUS_RULE WHERE enable=:enable")
+    public abstract List<CorrelationRule> queryRuleByEnable(@Bind("enable") int enable);
+
+    public List<CorrelationRule> queryRuleByRuleEnable(int enable) {
+        return queryRuleByEnable(enable);
+    }
+}
+
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/db/mapper/CorrelationRuleMapper.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/db/mapper/CorrelationRuleMapper.java
new file mode 100644
index 0000000..7d9ed97
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/db/mapper/CorrelationRuleMapper.java
@@ -0,0 +1,50 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.db.mapper;
+
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.Properties;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.skife.jdbi.v2.StatementContext;
+import org.skife.jdbi.v2.tweak.ResultSetMapper;
+
+public class CorrelationRuleMapper implements ResultSetMapper<CorrelationRule> {
+
+    public CorrelationRule map(int i, ResultSet resultSet, StatementContext statementContext)
+        throws SQLException {
+        CorrelationRule correlationRule = new CorrelationRule();
+        correlationRule.setName(resultSet.getString("name"));
+        correlationRule.setRid(resultSet.getString("rid"));
+        correlationRule.setDescription(resultSet.getString("description"));
+        correlationRule.setEnabled(resultSet.getInt("enable"));
+        correlationRule.setTemplateID(resultSet.getInt("templateID"));
+        correlationRule.setEngineId(resultSet.getString("engineID"));
+        correlationRule.setEngineType(resultSet.getString("engineType"));
+        correlationRule.setCreator(resultSet.getString("creator"));
+        correlationRule.setCreateTime(resultSet.getDate("createTime"));
+        correlationRule.setModifier(resultSet.getString("updator"));
+        correlationRule.setUpdateTime(resultSet.getDate("updateTime"));
+        correlationRule.setParams((Properties) resultSet.getObject("params"));
+        correlationRule.setDomain(resultSet.getString("domain"));
+        correlationRule.setContent(resultSet.getString("content"));
+        correlationRule.setIsManual(resultSet.getInt("isManual"));
+        correlationRule.setVendor(resultSet.getString("vendor"));
+        correlationRule.setPackageName(resultSet.getString("package"));
+        return correlationRule;
+    }
+
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/listener/AlarmMqMessageListener.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/listener/AlarmMqMessageListener.java
new file mode 100644
index 0000000..f1bc622
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/listener/AlarmMqMessageListener.java
@@ -0,0 +1,83 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.listener;
+
+
+import javax.annotation.PostConstruct;
+import javax.inject.Inject;
+import javax.jms.Connection;
+import javax.jms.ConnectionFactory;
+import javax.jms.Destination;
+import javax.jms.JMSException;
+import javax.jms.MessageConsumer;
+import javax.jms.ObjectMessage;
+import javax.jms.Session;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.activemq.ActiveMQConnectionFactory;
+import org.glassfish.hk2.api.IterableProvider;
+import org.jvnet.hk2.annotations.Service;
+import org.openo.holmes.common.api.stat.Alarm;
+import org.openo.holmes.common.config.MQConfig;
+import org.openo.holmes.common.constant.AlarmConst;
+import org.openo.holmes.enginemgt.manager.DroolsEngine;
+
+@Service
+@Slf4j
+public class AlarmMqMessageListener implements Runnable {
+
+    @Inject
+    private static IterableProvider<MQConfig> mqConfigProvider;
+    @Inject
+    DroolsEngine droolsEngine;
+    private ConnectionFactory connectionFactory;
+
+    @PostConstruct
+    public void init() {
+
+        String brokerURL =
+            "tcp://" + mqConfigProvider.get().brokerIp + ":" + mqConfigProvider.get().brokerPort;
+        connectionFactory = new ActiveMQConnectionFactory(mqConfigProvider.get().brokerUsername,
+            mqConfigProvider.get().brokerPassword, brokerURL);
+    }
+
+
+    public void run() {
+        Connection connection;
+        Session session;
+        Destination destination;
+        MessageConsumer messageConsumer;
+
+        try {
+            connection = connectionFactory.createConnection();
+            connection.start();
+            session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
+            destination = session.createTopic(AlarmConst.MQ_TOPIC_NAME_ALARM);
+            messageConsumer = session.createConsumer(destination);
+
+            while (true) {
+                ObjectMessage objMessage = (ObjectMessage) messageConsumer.receive(100000);
+                if (objMessage != null) {
+                    droolsEngine.putRaisedIntoStream((Alarm) objMessage.getObject());
+                } else {
+                    break;
+                }
+            }
+        } catch (JMSException e) {
+            log.debug("Receive alarm failure" + e.getMessage());
+        }
+
+    }
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/manager/DroolsEngine.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/manager/DroolsEngine.java
new file mode 100644
index 0000000..49b2042
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/manager/DroolsEngine.java
@@ -0,0 +1,215 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.manager;
+
+
+import java.io.StringReader;
+import java.util.List;
+import java.util.Locale;
+import javax.annotation.PostConstruct;
+import javax.inject.Inject;
+import lombok.extern.slf4j.Slf4j;
+import org.drools.KnowledgeBase;
+import org.drools.KnowledgeBaseConfiguration;
+import org.drools.KnowledgeBaseFactory;
+import org.drools.builder.KnowledgeBuilder;
+import org.drools.builder.KnowledgeBuilderFactory;
+import org.drools.builder.ResourceType;
+import org.drools.conf.EventProcessingOption;
+import org.drools.definition.KnowledgePackage;
+import org.drools.io.Resource;
+import org.drools.io.ResourceFactory;
+import org.drools.runtime.StatefulKnowledgeSession;
+import org.drools.runtime.rule.FactHandle;
+import org.jvnet.hk2.annotations.Service;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.openo.holmes.common.api.stat.Alarm;
+import org.openo.holmes.common.exception.DbException;
+import org.openo.holmes.common.exception.EngineException;
+import org.openo.holmes.common.exception.RuleIllegalityException;
+import org.openo.holmes.common.utils.ExceptionUtil;
+import org.openo.holmes.common.utils.I18nProxy;
+import org.openo.holmes.enginemgt.listener.AlarmMqMessageListener;
+import org.openo.holmes.enginemgt.request.DeployRuleRequest;
+import org.openo.holmes.enginemgt.wrapper.RuleMgtWrapper;
+
+@Slf4j
+@Service
+public class DroolsEngine {
+
+    private final static String CORRELATION_RULE = "CORRELATION_RULE";
+    private final static String CORRELATION_ALARM = "CORRELATION_ALARM";
+    private final static int ENABLE = 1;
+    @Inject
+    private RuleMgtWrapper ruleMgtWrapper;
+    @Inject
+    private AlarmMqMessageListener mqRegister;
+    private KnowledgeBase kbase;
+    private KnowledgeBaseConfiguration kconf;
+    private StatefulKnowledgeSession ksession;
+    private KnowledgeBuilder kbuilder;
+
+    @PostConstruct
+    private void init() {
+        registerAlarmTopicListener();
+        try {
+            start();
+        } catch (Exception e) {
+            log.error("Start service failed: " + e.getMessage());
+            throw ExceptionUtil.buildExceptionResponse("Start service failed!");
+        }
+    }
+
+    private void registerAlarmTopicListener() {
+        Thread thread = new Thread(mqRegister);
+        thread.start();
+    }
+
+
+    private void start() throws EngineException, RuleIllegalityException, DbException {
+        log.info("Drools Egine Initialize Begining ... ");
+
+        initEngineParameter();
+        initDeployRule();
+
+        log.info("Business Rule Egine Initialize Successfully ");
+    }
+
+    public void stop() throws Exception {
+        this.ksession.dispose();
+    }
+
+    private void initEngineParameter() throws EngineException {
+        this.kconf = KnowledgeBaseFactory.newKnowledgeBaseConfiguration();
+
+        this.kconf.setOption(EventProcessingOption.STREAM);
+
+        this.kconf.setProperty("drools.assertBehaviour", "equality");
+
+        this.kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+
+        this.kbase = KnowledgeBaseFactory.newKnowledgeBase("D-ENGINE", this.kconf);
+
+        this.ksession = kbase.newStatefulKnowledgeSession();
+    }
+
+    private void initDeployRule() throws RuleIllegalityException, EngineException, DbException {
+        List<CorrelationRule> rules = ruleMgtWrapper.queryRuleByEnable(ENABLE);
+
+        if (rules.size() > 0) {
+            for (CorrelationRule rule : rules) {
+                if (rule.getContent() != null) {
+                    deployRuleFromCache(rule.getContent());
+                }
+            }
+        }
+    }
+
+    private void deployRuleFromCache(String ruleContent) throws EngineException {
+        StringReader reader = new StringReader(ruleContent);
+        Resource res = ResourceFactory.newReaderResource(reader);
+
+        kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+
+        kbuilder.add(res, ResourceType.DRL);
+
+        try {
+
+            kbase.addKnowledgePackages(kbuilder.getKnowledgePackages());
+        } catch (Exception e) {
+            throw new EngineException(e);
+        }
+
+        kbuilder = null;
+
+        ksession.fireAllRules();
+    }
+
+    public synchronized String deployRule(DeployRuleRequest rule, Locale locale)
+        throws RuleIllegalityException, EngineException {
+        StringReader reader = new StringReader(rule.getContent());
+        Resource res = ResourceFactory.newReaderResource(reader);
+
+        kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+
+        kbuilder.add(res, ResourceType.DRL);
+
+        if (kbuilder.hasErrors()) {
+            throw new RuleIllegalityException(kbuilder.getErrors().toString());
+        }
+
+        String packageName = kbuilder.getKnowledgePackages().iterator().next().getName();
+
+        if (kbase.getKnowledgePackages().contains(packageName)) {
+            throw new RuleIllegalityException(
+                I18nProxy.getInstance().getValue(locale, I18nProxy.ENGINE_CONTAINS_PACKAGE));
+        }
+        try {
+
+            kbase.addKnowledgePackages(kbuilder.getKnowledgePackages());
+        } catch (Exception e) {
+            throw new EngineException(e);
+        }
+
+        kbuilder = null;
+
+        ksession.fireAllRules();
+        return packageName;
+    }
+
+
+    public synchronized void undeployRule(String packageName)
+        throws RuleIllegalityException, EngineException {
+
+        KnowledgePackage pkg = kbase.getKnowledgePackage(packageName);
+
+        if (null == pkg) {
+            throw new RuleIllegalityException(packageName);
+        }
+
+        try {
+
+            kbase.removeKnowledgePackage(pkg.getName());
+        } catch (Exception e) {
+
+            throw new EngineException(e);
+        }
+    }
+
+    public void compileRule(String content, Locale locale)
+        throws RuleIllegalityException {
+        StringReader reader = new StringReader(content);
+        Resource res = ResourceFactory.newReaderResource(reader);
+
+        kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
+
+        kbuilder.add(res, ResourceType.DRL);
+
+        if (kbuilder.hasErrors()) {
+            throw new RuleIllegalityException(kbuilder.getErrors().toString());
+        }
+        kbuilder = null;
+    }
+
+    public void putRaisedIntoStream(Alarm raiseAlarm) {
+        FactHandle factHandle = this.ksession.getFactHandle(raiseAlarm);
+        if (factHandle != null) {
+            this.ksession.retract(factHandle);
+        }
+        this.ksession.insert(raiseAlarm);
+        this.ksession.fireAllRules();
+    }
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/request/CompileRuleRequest.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/request/CompileRuleRequest.java
new file mode 100644
index 0000000..bf056be
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/request/CompileRuleRequest.java
@@ -0,0 +1,30 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.request;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import javax.validation.constraints.NotNull;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class CompileRuleRequest {
+
+    @JsonProperty(value = "content")
+    @NotNull
+    private String content;
+}
\ No newline at end of file
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/request/DeployRuleRequest.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/request/DeployRuleRequest.java
new file mode 100644
index 0000000..b1d9346
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/request/DeployRuleRequest.java
@@ -0,0 +1,33 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.request;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import javax.validation.constraints.NotNull;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class DeployRuleRequest {
+
+    @JsonProperty(value = "content")
+    @NotNull
+    private String content;
+
+    @JsonProperty(value = "engineid")
+    private String engineId;
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/resources/EngineResources.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/resources/EngineResources.java
new file mode 100644
index 0000000..59fb9e3
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/resources/EngineResources.java
@@ -0,0 +1,140 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.resources;
+
+
+import com.codahale.metrics.annotation.Timed;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import java.util.Locale;
+import javax.inject.Inject;
+import javax.servlet.http.HttpServletRequest;
+import javax.ws.rs.DELETE;
+import javax.ws.rs.POST;
+import javax.ws.rs.PUT;
+import javax.ws.rs.Path;
+import javax.ws.rs.PathParam;
+import javax.ws.rs.Produces;
+import javax.ws.rs.core.Context;
+import javax.ws.rs.core.MediaType;
+import lombok.extern.slf4j.Slf4j;
+import org.jvnet.hk2.annotations.Service;
+import org.openo.holmes.common.exception.EngineException;
+import org.openo.holmes.common.exception.RuleIllegalityException;
+import org.openo.holmes.common.utils.ExceptionUtil;
+import org.openo.holmes.common.utils.I18nProxy;
+import org.openo.holmes.common.utils.LanguageUtil;
+import org.openo.holmes.enginemgt.manager.DroolsEngine;
+import org.openo.holmes.enginemgt.request.CompileRuleRequest;
+import org.openo.holmes.enginemgt.request.DeployRuleRequest;
+import org.openo.holmes.enginemgt.response.CorrelationRuleResponse;
+
+@Service
+@Path("/rule")
+@Api(tags = {"Engine Manager"})
+@Produces(MediaType.APPLICATION_JSON)
+@Slf4j
+public class EngineResources {
+
+    @Inject
+    DroolsEngine droolsEngine;
+
+    @PUT
+    @ApiOperation(value = "Add rule to Engine and Cache", response = CorrelationRuleResponse.class)
+    @Produces(MediaType.APPLICATION_JSON)
+    @Timed
+    public CorrelationRuleResponse deployRule(DeployRuleRequest deployRuleRequest,
+        @Context HttpServletRequest httpRequest) {
+
+        CorrelationRuleResponse crResponse = new CorrelationRuleResponse();
+        Locale locale = LanguageUtil.getLocale(httpRequest);
+        try {
+
+            String packageName = droolsEngine.deployRule(deployRuleRequest, locale);
+            crResponse.setPackageName(packageName);
+
+        } catch (RuleIllegalityException ruleIllegalityException) {
+
+            String errorMsg = I18nProxy.getInstance().getValueByArgs(locale,
+                I18nProxy.ENGINE_CONTENT_ILLEGALITY,
+                new String[]{ruleIllegalityException.getMessage()});
+            log.error(errorMsg);
+            throw ExceptionUtil.buildExceptionResponse(errorMsg);
+        } catch (EngineException e) {
+
+            String errorMsg =
+                I18nProxy.getInstance().getValue(locale, I18nProxy.ENGINE_DEPLOY_RULE_FAILED);
+            log.error(errorMsg + ":" + e.getMessage());
+            throw ExceptionUtil.buildExceptionResponse(errorMsg);
+        }
+
+        return crResponse;
+    }
+
+    @DELETE
+    @ApiOperation(value = "delete rule")
+    @Produces(MediaType.APPLICATION_JSON)
+    @Timed
+    @Path("/{packageName}")
+    public boolean undeployRule(@PathParam("packageName") String packageName,
+        @Context HttpServletRequest httpRequest) {
+
+        Locale locale = LanguageUtil.getLocale(httpRequest);
+
+        try {
+
+            droolsEngine.undeployRule(packageName);
+
+        } catch (RuleIllegalityException ruleIllegalityException) {
+
+            String errorMsg = I18nProxy.getInstance().getValueByArgs(locale,
+                I18nProxy.ENGINE_DELETE_RULE_NULL,
+                new String[]{ruleIllegalityException.getMessage()});
+            log.error(errorMsg);
+            throw ExceptionUtil.buildExceptionResponse(errorMsg);
+        } catch (EngineException e) {
+
+            String errorMsg = I18nProxy.getInstance().getValueByArgs(locale,
+                I18nProxy.ENGINE_DELETE_RULE_FAILED, new String[]{packageName});
+            log.error(errorMsg + e.getMessage());
+            throw ExceptionUtil.buildExceptionResponse(errorMsg);
+        }
+        return true;
+    }
+
+
+    @POST
+    @ApiOperation(value = "compile rule")
+    @Produces(MediaType.APPLICATION_JSON)
+    @Timed
+    public boolean compileRule(CompileRuleRequest compileRuleRequest,
+        @Context HttpServletRequest httpRequest) {
+
+        Locale locale = LanguageUtil.getLocale(httpRequest);
+
+        try {
+            droolsEngine.compileRule(compileRuleRequest.getContent(), locale);
+        } catch (RuleIllegalityException ruleIllegalityException) {
+
+            String errorMsg = I18nProxy.getInstance().getValueByArgs(locale,
+                I18nProxy.ENGINE_CONTENT_ILLEGALITY,
+                new String[]{ruleIllegalityException.getMessage()});
+            log.error(errorMsg);
+            throw ExceptionUtil.buildExceptionResponse(errorMsg);
+        }
+        return true;
+    }
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/response/CorrelationRuleResponse.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/response/CorrelationRuleResponse.java
new file mode 100644
index 0000000..6c9842c
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/response/CorrelationRuleResponse.java
@@ -0,0 +1,28 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.response;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class CorrelationRuleResponse {
+
+    @JsonProperty(value = "package")
+    private String packageName;
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/utils/AlarmUtil.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/utils/AlarmUtil.java
new file mode 100644
index 0000000..2d5b5ca
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/utils/AlarmUtil.java
@@ -0,0 +1,99 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.utils;
+
+import java.util.HashMap;
+import java.util.Map;
+import org.jvnet.hk2.annotations.Service;
+import org.openo.holmes.common.api.stat.Alarm;
+import org.openo.holmes.common.producer.MQProducer;
+
+@Service
+public class AlarmUtil {
+
+    private final static AlarmUtil alarmUtil = new AlarmUtil();
+    /**
+     * Map<ruleId, <ProbableCause-EquipType, priority>>
+     */
+    private final Map<String, Map<String, Integer>> rootPriorityMap =
+        new HashMap<String, Map<String, Integer>>();
+    /**
+     * Map<rule, ProbableCause+EquipType+priority>
+     */
+    private final Map<String, String> saveRuleMsg = new HashMap<String, String>();
+
+    private AlarmUtil() {
+    }
+
+    public static AlarmUtil getInstance() {
+        return alarmUtil;
+    }
+
+    public boolean equipTypeFilter(String probableCauseStr, String equipType, Alarm alarm) {
+        if ("null".equals(probableCauseStr)) {
+            return true;
+        }
+        String[] equipTypes = equipType.replace(" ", "").split(",");
+        String[] probableCauseStrs = probableCauseStr.replace(" ", "").split(",");
+        for (int i = 0; i < probableCauseStrs.length; i++) {
+            if (alarm.getProbableCause() == Long.parseLong(probableCauseStrs[i])
+                && alarm.getEquipType().equals(equipTypes[i])) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public Integer getPriority(String ruleId, String probableCauseStr, String rootAlarmFeatureStr,
+        String equipTypeStr, Alarm alarm) {
+        if (rootPriorityMap.containsKey(ruleId)) {
+            if (!saveRuleMsg.get(ruleId)
+                .equals(probableCauseStr + equipTypeStr + rootAlarmFeatureStr)) {
+                setPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr);
+            }
+        } else {
+            setPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr);
+        }
+
+        Integer priority =
+            rootPriorityMap.get(ruleId).get(alarm.getProbableCause() + "-" + alarm.getEquipType());
+        if (priority == null) {
+            priority = 0;
+        }
+        return priority;
+    }
+
+    private void setPriority(String ruleId, String probableCauseStr, String rootAlarmFeatureStr,
+        String equipTypeStr) {
+        saveRuleMsg.put(ruleId, probableCauseStr + equipTypeStr + rootAlarmFeatureStr);
+
+        Map<String, Integer> map = new HashMap<String, Integer>();
+        String[] probableCauseStrs = probableCauseStr.replace(" ", "").split(",");
+        String[] rootAlarmFeatureStrs = rootAlarmFeatureStr.replace(" ", "").split(",");
+        String[] equipTypes = equipTypeStr.replace(" ", "").split(",");
+        for (int i = 0; i < rootAlarmFeatureStrs.length; i++) {
+            map.put(probableCauseStrs[i] + "-" + equipTypes[i],
+                Integer.parseInt(rootAlarmFeatureStrs[i]));
+        }
+
+        rootPriorityMap.put(ruleId, map);
+    }
+
+    public MQProducer getMqProducer() {
+        MQProducer mqProducer = new MQProducer();
+        return mqProducer;
+    }
+}
diff --git a/engine-d/src/main/java/org/openo/holmes/enginemgt/wrapper/RuleMgtWrapper.java b/engine-d/src/main/java/org/openo/holmes/enginemgt/wrapper/RuleMgtWrapper.java
new file mode 100644
index 0000000..35890ca
--- /dev/null
+++ b/engine-d/src/main/java/org/openo/holmes/enginemgt/wrapper/RuleMgtWrapper.java
@@ -0,0 +1,47 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.openo.holmes.enginemgt.wrapper;
+
+import java.util.List;
+import javax.inject.Inject;
+import javax.inject.Singleton;
+import lombok.extern.slf4j.Slf4j;
+import org.jvnet.hk2.annotations.Service;
+import org.openo.holmes.common.api.entity.CorrelationRule;
+import org.openo.holmes.common.exception.DbException;
+import org.openo.holmes.common.utils.DbDaoUtil;
+import org.openo.holmes.common.utils.I18nProxy;
+import org.openo.holmes.enginemgt.db.CorrelationRuleDao;
+
+
+@Service
+@Singleton
+@Slf4j
+public class RuleMgtWrapper {
+
+    @Inject
+    private DbDaoUtil daoUtil;
+
+    public List<CorrelationRule> queryRuleByEnable(int enable) throws DbException {
+
+        List<CorrelationRule> ruleTemp = daoUtil.getJdbiDaoByOnDemand(CorrelationRuleDao.class)
+            .queryRuleByRuleEnable(enable);
+        if (ruleTemp != null) {
+            throw new DbException(I18nProxy.RULE_MANAGEMENT_REPEAT_RULE_NAME);
+        }
+        return ruleTemp;
+    }
+}