diff options
Diffstat (limited to 'engine-d')
-rw-r--r-- | engine-d/src/main/java/org/onap/holmes/engine/resources/EngineResources.java | 24 | ||||
-rw-r--r-- | engine-d/src/test/java/org/onap/holmes/engine/resources/EngineResourcesTest.java | 1 |
2 files changed, 22 insertions, 3 deletions
diff --git a/engine-d/src/main/java/org/onap/holmes/engine/resources/EngineResources.java b/engine-d/src/main/java/org/onap/holmes/engine/resources/EngineResources.java index 9724355..b4cb0b1 100644 --- a/engine-d/src/main/java/org/onap/holmes/engine/resources/EngineResources.java +++ b/engine-d/src/main/java/org/onap/holmes/engine/resources/EngineResources.java @@ -20,6 +20,9 @@ import com.codahale.metrics.annotation.Timed; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import io.swagger.annotations.ApiParam; +import java.io.BufferedReader; +import java.io.IOException; +import java.io.StringReader; import java.util.Locale; import javax.inject.Inject; import javax.servlet.http.HttpServletRequest; @@ -66,15 +69,25 @@ public class EngineResources { CorrelationRuleResponse crResponse = new CorrelationRuleResponse(); Locale locale = LanguageUtil.getLocale(httpRequest); try { - String packageName = droolsEngine.deployRule(deployRuleRequest, locale); + String packageName = getPackageName(deployRuleRequest.getContent()); DmaapService.loopControlNames .put(packageName, deployRuleRequest.getLoopControlName()); + String packageNameRet = droolsEngine.deployRule(deployRuleRequest, locale); + if (!packageName.equals(packageNameRet)) { + log.info("The parsed package name is different from that returned by the engine."); + DmaapService.loopControlNames.remove(packageName); + DmaapService.loopControlNames + .put(packageNameRet, deployRuleRequest.getLoopControlName()); + } log.info("Rule deployed. Package name: " + packageName); crResponse.setPackageName(packageName); } catch (CorrelationException correlationException) { log.error(correlationException.getMessage(), correlationException); throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage()); + } catch (RuntimeException e) { + log.error(e.getMessage(), e); + throw ExceptionUtil.buildExceptionResponse(e.getMessage()); } return crResponse; @@ -91,9 +104,8 @@ public class EngineResources { Locale locale = LanguageUtil.getLocale(httpRequest); try { - droolsEngine.undeployRule(packageName, locale); - + DmaapService.loopControlNames.remove(packageName); } catch (CorrelationException correlationException) { log.error(correlationException.getMessage(), correlationException); throw ExceptionUtil.buildExceptionResponse(correlationException.getMessage()); @@ -119,4 +131,10 @@ public class EngineResources { } return true; } + + private String getPackageName(String contents){ + String ret = contents.trim(); + ret = ret.substring(7, ret.indexOf("import")).trim(); + return ret.endsWith(";") ? ret.substring(0, ret.length() - 1) : ret; + } } diff --git a/engine-d/src/test/java/org/onap/holmes/engine/resources/EngineResourcesTest.java b/engine-d/src/test/java/org/onap/holmes/engine/resources/EngineResourcesTest.java index 98f4797..6be232f 100644 --- a/engine-d/src/test/java/org/onap/holmes/engine/resources/EngineResourcesTest.java +++ b/engine-d/src/test/java/org/onap/holmes/engine/resources/EngineResourcesTest.java @@ -66,6 +66,7 @@ public class EngineResourcesTest { @Test
public void deployRule_normal() throws CorrelationException {
DeployRuleRequest deployRuleRequest = new DeployRuleRequest();
+ deployRuleRequest.setContent("package packageName;\n\nimport xxx.xxx.xxx;");
deployRuleRequest.setLoopControlName("loopControlName");
HttpServletRequest httpRequest = PowerMock.createMock(HttpServletRequest.class);
|