summaryrefslogtreecommitdiffstats
path: root/policy-core/src
diff options
context:
space:
mode:
Diffstat (limited to 'policy-core/src')
-rw-r--r--policy-core/src/main/java/org/onap/policy/drools/util/KieUtils.java112
1 files changed, 112 insertions, 0 deletions
diff --git a/policy-core/src/main/java/org/onap/policy/drools/util/KieUtils.java b/policy-core/src/main/java/org/onap/policy/drools/util/KieUtils.java
new file mode 100644
index 00000000..59ee5119
--- /dev/null
+++ b/policy-core/src/main/java/org/onap/policy/drools/util/KieUtils.java
@@ -0,0 +1,112 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP
+ * ================================================================================
+ * Copyright (C) 2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * 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.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.drools.util;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.List;
+import org.drools.compiler.kie.builder.impl.InternalKieModule;
+import org.drools.compiler.kproject.models.KieModuleModelImpl;
+import org.kie.api.KieServices;
+import org.kie.api.builder.KieBuilder;
+import org.kie.api.builder.KieFileSystem;
+import org.kie.api.builder.Message;
+import org.kie.api.builder.ReleaseId;
+import org.kie.api.builder.model.KieModuleModel;
+import org.kie.scanner.MavenRepository;
+
+/**
+ * Kie related utilities
+ */
+public class KieUtils {
+
+ /**
+ * Installs a rules artifact in the local maven repository
+ *
+ * @param kmodule kmodule specification
+ * @param pom pom
+ * @param drlKJarPath path used in kjar drl to the actual drl
+ * @param drl rules in drl language
+ *
+ * @return releaseId result o a sucessful installation
+ * @throws IOException error accessing necessary resources
+ */
+ public static ReleaseId installArtifact(String kmodule, String pom, String drlKJarPath, String drl) throws IOException {
+ KieModuleModel kieModule = KieModuleModelImpl.fromXML(kmodule);
+
+ final KieFileSystem kieFileSystem = KieServices.Factory.get().newKieFileSystem();
+ kieFileSystem.writeKModuleXML(kieModule.toXML());
+ kieFileSystem.writePomXML(pom);
+ kieFileSystem.write(drlKJarPath, drl);
+
+ KieBuilder kieBuilder = kieBuild(kieFileSystem);
+
+ Path pomPath = Files.createTempFile("policy-core-", ".pom");
+ Files.write(pomPath, pom.getBytes(StandardCharsets.UTF_8));
+ File pomFile = pomPath.toFile();
+ pomFile.deleteOnExit();
+
+ ReleaseId releaseId = kieBuilder.getKieModule().getReleaseId();
+ MavenRepository.getMavenRepository().
+ installArtifact(releaseId,
+ (InternalKieModule) kieBuilder.getKieModule(),
+ pomFile);
+ return releaseId;
+ }
+
+ /**
+ * Installs a rules artifact in the local maven repository
+ *
+ * @param kmodule kmodule specification
+ * @param pom pom
+ * @param drlKJarPath path used in kjar drl to the actual drl
+ * @param drl rules in drl language
+ *
+ * @return releaseId result o a sucessful installation
+ * @throws IOException error accessing necessary resources
+ */
+ public static ReleaseId installArtifact(File kmodule, File pom, String drlKJarPath, File drl) throws IOException {
+ KieModuleModel kieModule = KieModuleModelImpl.fromXML(kmodule);
+
+ final KieFileSystem kieFileSystem = KieServices.Factory.get().newKieFileSystem();
+ kieFileSystem.writeKModuleXML(kieModule.toXML());
+ kieFileSystem.writePomXML(new String(Files.readAllBytes(pom.toPath())));
+ kieFileSystem.write(drlKJarPath, new String(Files.readAllBytes(drl.toPath())));
+
+ KieBuilder kieBuilder = kieBuild(kieFileSystem);
+
+ ReleaseId releaseId = kieBuilder.getKieModule().getReleaseId();
+ MavenRepository.getMavenRepository().
+ installArtifact(releaseId, (InternalKieModule) kieBuilder.getKieModule(), pom);
+ return releaseId;
+ }
+
+ private static KieBuilder kieBuild(KieFileSystem kieFileSystem) {
+ KieBuilder kieBuilder = KieServices.Factory.get().newKieBuilder(kieFileSystem);
+ List<Message> messages = kieBuilder.buildAll().getResults().getMessages();
+ if (messages != null && !messages.isEmpty())
+ throw new IllegalArgumentException(messages.toString());
+ return kieBuilder;
+ }
+}