Examples with KieContainerImpl org.drools.compiler.kie.builder.impl.KieContainerImpl used on opensource projects

Example 16 with KieContainerImpl

use of org.drools.compiler.kie.builder.impl.KieContainerImpl in project drools by kiegroup.

the class GeneratedClassNamesTest method testGeneratedClassNames.

@Test
public void testGeneratedClassNames() {
    String str = "import " + Person.class.getCanonicalName() + ";" + "rule R when\n" + "  $p : Person(name == \"Mario\")\n" + "then\n" + "  System.out.println(\"hello\");\n" + "end";
    KieServices ks = KieServices.get();
    ReleaseId releaseId = ks.newReleaseId("org.kie", "kjar-test-" + UUID.randomUUID(), "1.0");
    createKieBuilder(ks, getDefaultKieModuleModel(ks), releaseId, toKieFiles(new String[] { str }));
    KieContainer kcontainer = ks.newKieContainer(releaseId);
    KieModule kieModule = ((KieContainerImpl) kcontainer).getKieModuleForKBase("kbase");
    assertTrue(kieModule instanceof CanonicalKieModule);
    KieSession ksession = kcontainer.newKieSession();
    Set<String> generatedClassNames = ((CanonicalKieModule) kieModule).getGeneratedClassNames();
    assertGeneratedClassNames(generatedClassNames);
    Person me = new Person("Mario", 40);
    ksession.insert(me);
    int fired = ksession.fireAllRules();
    assertEquals(1, fired);
}

Example 17 with KieContainerImpl

use of org.drools.compiler.kie.builder.impl.KieContainerImpl in project drools by kiegroup.

the class GeneratedClassNamesTest method testModuleWithDep.

private void testModuleWithDep(ProjectClassLoader projectClassLoader) throws Exception {
    KieServices ks = KieServices.get();
    ReleaseId releaseIdDep = ks.newReleaseId("org.example", "depRule", "1.0.0");
    ReleaseId releaseIdMain = ks.newReleaseId("org.example", "mainRule", "1.0.0");
    String depStr = "package org.example.dep\n" + "import " + Person.class.getCanonicalName() + ";\n" + "rule R_dep when\n" + "  $p : Person(name == \"Mario\")\n" + "then\n" + "  System.out.println(\"hello dep rule\");\n" + "end";
    String mainStr = "package org.example.main\n" + "import " + Person.class.getCanonicalName() + ";\n" + "rule R_main when\n" + "  $p : Person(name == \"Luca\")\n" + "then\n" + "  System.out.println(\"hello main rule\");\n" + "end";
    String mainPom = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + "<project xmlns=\"http://maven.apache.org/POM/4.0.0\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n" + "         xsi:schemaLocation=\"http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd\">\n" + "  <modelVersion>4.0.0</modelVersion>\n" + "\n" + "  <groupId>" + releaseIdMain.getGroupId() + "</groupId>\n" + "  <artifactId>" + releaseIdMain.getArtifactId() + "</artifactId>\n" + "  <version>" + releaseIdMain.getVersion() + "</version>\n" + "  <dependencies>\n" + "    <dependency>\n" + "      <groupId>" + releaseIdDep.getGroupId() + "</groupId>\n" + "      <artifactId>" + releaseIdDep.getArtifactId() + "</artifactId>\n" + "      <version>" + releaseIdDep.getVersion() + "</version>\n" + "    </dependency>\n" + "  </dependencies>" + "</project>";
    // create dep
    KieFileSystem depKfs = ks.newKieFileSystem();
    KieModuleModel depModuleModel = ks.newKieModuleModel();
    KieBaseModel depKieBaseModel = depModuleModel.newKieBaseModel("dep").setDefault(false);
    depKieBaseModel.addPackage("*");
    depKfs.writeKModuleXML(depModuleModel.toXML());
    depKfs.generateAndWritePomXML(releaseIdDep);
    Resource depResource = ResourceFactory.newByteArrayResource(depStr.getBytes());
    depResource.setSourcePath("src/main/resources/org/example/dep/r1.drl");
    depKfs.write(depResource);
    KieBuilder depKieBuilder = ks.newKieBuilder(depKfs);
    depKieBuilder.buildAll(ExecutableModelProject.class);
    CanonicalKieModule depKieModule = (CanonicalKieModule) depKieBuilder.getKieModule();
    MemoryKieModule depMemoryKieModule = (MemoryKieModule) depKieModule.getInternalKieModule();
    MemoryFileSystem depMfs = depMemoryKieModule.getMemoryFileSystem();
    byte[] depBinary = depMfs.writeAsBytes();
    Resource depJarResource = ks.getResources().newByteArrayResource(depBinary);
    // create main
    KieFileSystem mainKfs = ks.newKieFileSystem();
    KieModuleModel mainModuleModel = ks.newKieModuleModel();
    KieBaseModel mainKieBaseModel = mainModuleModel.newKieBaseModel("main").setDefault(true);
    mainKieBaseModel.addPackage("*");
    mainKieBaseModel.addInclude("dep");
    Resource mainResource = ResourceFactory.newByteArrayResource(mainStr.getBytes());
    mainResource.setSourcePath("src/main/resources/org/example/main/r2.drl");
    mainKfs.write(mainResource);
    mainKfs.writeKModuleXML(mainModuleModel.toXML());
    mainKfs.write("pom.xml", mainPom);
    KieBuilder mainKieBuilder = ks.newKieBuilder(mainKfs);
    mainKieBuilder.setDependencies(depJarResource);
    mainKieBuilder.buildAll(ExecutableModelProject.class);
    CanonicalKieModule mainKieModule = (CanonicalKieModule) mainKieBuilder.getKieModule();
    MemoryKieModule mainMemoryKieModule = (MemoryKieModule) mainKieModule.getInternalKieModule();
    MemoryFileSystem mainMfs = mainMemoryKieModule.getMemoryFileSystem();
    byte[] mainBinary = mainMfs.writeAsBytes();
    Resource mainJarResource = ks.getResources().newByteArrayResource(mainBinary);
    // cleanup
    ks.getRepository().removeKieModule(releaseIdMain);
    ks.getRepository().removeKieModule(releaseIdDep);
    ks.getRepository().addKieModule(mainJarResource, depJarResource);
    KieContainerImpl kieContainer;
    if (projectClassLoader == null) {
        kieContainer = (KieContainerImpl) ks.newKieContainer(releaseIdMain);
    } else {
        kieContainer = (KieContainerImpl) ks.newKieContainer(releaseIdMain, projectClassLoader);
    }
    CanonicalKieModule mainLoadedModule = (CanonicalKieModule) kieContainer.getKieModuleForKBase("main");
    KieBase kbase = kieContainer.getKieBase();
    Set<String> mainGeneratedClassNames = mainLoadedModule.getGeneratedClassNames();
    assertGeneratedClassNamesWithDep(mainGeneratedClassNames);
    KieSession ksession = kbase.newKieSession();
    ksession.insert(new Person("Mario"));
    ksession.insert(new Person("Luca"));
    int fired = ksession.fireAllRules();
    assertEquals(2, fired);
    ksession.dispose();
}

Example 18 with KieContainerImpl

use of org.drools.compiler.kie.builder.impl.KieContainerImpl in project drools by kiegroup.

the class KieCDIExtension method addKSessionBean.

public void addKSessionBean(AfterBeanDiscovery abd, KieCDIEntry entry) {
    ReleaseId releaseId = entry.getReleaseId();
    // default to classpath, but allow it to be overriden
    KieContainerImpl kieContainer = classpathKContainer;
    if (releaseId != null) {
        kieContainer = (KieContainerImpl) gavs.get(releaseId);
        if (kieContainer == null) {
            log.error("Unable to create KSession({}), could not retrieve KieContainer for ReleaseId {}", entry.getValue(), releaseId.toString());
            return;
        }
    }
    KieProject kProject = kieContainer.getKieProject();
    String kSessionName = entry.getValue();
    KieSessionModel kSessionModel = null;
    if (StringUtils.isEmpty(kSessionName)) {
        kSessionModel = (entry.getType() == KieSession.class) ? kProject.getDefaultKieSession() : kProject.getDefaultStatelessKieSession();
    } else {
        kSessionModel = kProject.getKieSessionModel(kSessionName);
    }
    if (kSessionModel == null) {
        log.error("Annotation @KSession({}) found, but no KieSessionModel exists.\nEither the required kmodule.xml does not exist, is corrupted, or is missing the KieBase entry", kSessionName);
        return;
    }
    if (kSessionModel.getScope() != null && !kSessionModel.getScope().trim().equals(entry.getScope().getClass().getName())) {
        try {
            if (kSessionModel.getScope().indexOf('.') >= 0) {
                entry.setScope((Class<? extends Annotation>) Class.forName(kSessionModel.getScope()));
            } else {
                entry.setScope((Class<? extends Annotation>) Class.forName("javax.enterprise.context." + kSessionModel.getScope()));
            }
        } catch (ClassNotFoundException e) {
            log.error("KieBaseModule {} overrides default annotation, but it was not able to find it {}\n{}", new String[] { kSessionName, kSessionModel.getScope(), e.getMessage() });
        }
    }
    if (KieSessionType.STATELESS.equals(kSessionModel.getType())) {
        if (log.isDebugEnabled()) {
            InternalKieModule kModule = kProject.getKieModuleForKBase(((KieSessionModelImpl) kSessionModel).getKieBaseModel().getName());
            log.debug("Added Bean for Stateless @KSession({}) from: {}", kSessionName, kModule);
        }
        abd.addBean(new StatelessKSessionBean(kSessionModel, kieContainer, entry.getKReleaseId(), entry.getScope(), entry.getName(), entry.getInjectionPoints()));
    } else {
        InternalKieModule kModule = kProject.getKieModuleForKBase(((KieSessionModelImpl) kSessionModel).getKieBaseModel().getName());
        log.debug("Added Bean for Stateful @KSession({})  from: {}", kSessionName, kModule);
        abd.addBean(new StatefulKSessionBean(kSessionName, kSessionModel, kieContainer, entry.getKReleaseId(), entry.getScope(), entry.getName(), entry.getInjectionPoints()));
    }
}

Example 19 with KieContainerImpl

use of org.drools.compiler.kie.builder.impl.KieContainerImpl in project drools by kiegroup.

the class ClassLoaderLeakTest method buildKjarsInALoop.

private void buildKjarsInALoop(boolean withExecModel) {
    KieServices ks = KieServices.Factory.get();
    KieRepository kr = ks.getRepository();
    KieContainer kieContainer = null;
    int oldSize = Integer.MAX_VALUE;
    for (int i = 0; i < 3; i++) {
        KieModule kModule = buildKieModule("1.0." + i, RULES, withExecModel);
        if (kieContainer == null) {
            kieContainer = ks.newKieContainer(kModule.getReleaseId());
        } else {
            kieContainer.updateToVersion(kModule.getReleaseId());
        }
        KieProject kp = ((KieContainerImpl) kieContainer).getKieProject();
        ProjectClassLoader cl = (ProjectClassLoader) kp.getClassLoader();
        System.out.println(String.format("ProjectClassLoader.store size: %d, " + "retained bytes: %d.", cl.getStore().size(), cl.getStore().values().stream().map(b -> b.length).reduce(0, Integer::sum)));
        assertTrue(cl.getStore().size() <= oldSize);
        oldSize = cl.getStore().size();
        kr.removeKieModule(kModule.getReleaseId());
    }
}

Example 20 with KieContainerImpl

use of org.drools.compiler.kie.builder.impl.KieContainerImpl in project drools by kiegroup.

the class DMNUpdateTest method test_as_kie_wb_common_services_backend_Builder.

@Test
public void test_as_kie_wb_common_services_backend_Builder() throws Exception {
    final KieServices ks = KieServices.Factory.get();
    final ReleaseId v100 = ks.newReleaseId("org.kie", "dmn-test", "1.0.0");
    final KieModule kieModule = KieHelper.createAndDeployJar(ks, v100, ks.getResources().newClassPathResource("0001-input-data-string.dmn", this.getClass()));
    final KieContainer kieContainer = ks.newKieContainer(v100);
    final KieSession kieSession = kieContainer.newKieSession();
    final DMNRuntime runtime = kieSession.getKieRuntime(DMNRuntime.class);
    Assert.assertNotNull(runtime);
    assertThat(runtime.getModels(), hasSize(1));
    check0001_input_data_string(runtime);
    // the below is performed by the WB at: https://github.com/kiegroup/kie-wb-common/blob/9e6b6da145e61ac8f5a9f7c0259d44aa9d090a2b/kie-wb-common-services/kie-wb-common-services-backend/src/main/java/org/kie/workbench/common/services/backend/builder/core/Builder.java#L592-L620
    final KieProject kieProject = new KieModuleKieProject((InternalKieModule) kieModule, null);
    final KieContainer kieContainer2 = new KieContainerImpl(kieProject, ks.getRepository(), v100);
    // exhibit the issue.
    final KieSession kieSession2 = kieContainer2.newKieSession();
    final DMNRuntime runtime2 = kieSession2.getKieRuntime(DMNRuntime.class);
    Assert.assertNotNull(runtime2);
    assertThat(runtime2.getModels(), hasSize(1));
    check0001_input_data_string(runtime2);
}