private static void loadRuleFile() { try { ClassLoader classLoader = Steward.class.getClassLoader(); PackageBuilderConfiguration configuration = new PackageBuilderConfiguration(classLoader); PackageBuilder builder = new PackageBuilder(configuration); Reader rulesFile = new InputStreamReader(Steward.class.getResourceAsStream("../Steward.drl")); builder.addPackageFromDrl(rulesFile); Reader ruleFlowFile1 = new InputStreamReader(Steward.class.getResourceAsStream("../ActionGeneration.rfm")); Reader ruleFlowFile2 = new InputStreamReader(Steward.class.getResourceAsStream("../ActionSelection.rfm")); builder.addRuleFlow(ruleFlowFile1); builder.addRuleFlow(ruleFlowFile2); if (builder.hasErrors()) { System.out.println(builder.getErrors().toString()); throw new RuntimeException("Unable to compile rule file."); } RuleBaseConfiguration ruleBaseConfiguration = new RuleBaseConfiguration(classLoader); ruleBase = RuleBaseFactory.newRuleBase(ruleBaseConfiguration); ruleBase.addPackage(builder.getPackage()); } catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); } }
@Test public void testLogicalInsertionsWithModify() throws Exception { final PackageBuilder builder = new PackageBuilder(); builder.addPackageFromDrl( new InputStreamReader( getClass().getResourceAsStream("test_LogicalInsertionsWithUpdate.drl"))); if (builder.hasErrors()) { fail(builder.getErrors().toString()); } final Package pkg = builder.getPackage(); RuleBase ruleBase = getRuleBase(); ruleBase.addPackage(pkg); ruleBase = SerializationHelper.serializeObject(ruleBase); StatefulSession workingMemory = ruleBase.newStatefulSession(); List l; final Person p = new Person("person"); p.setAge(2); FactHandle h = workingMemory.insert(p); assertEquals(1, IteratorToList.convert(workingMemory.iterateObjects()).size()); workingMemory.fireAllRules(); workingMemory = getSerialisedStatefulSession(workingMemory); assertEquals(2, IteratorToList.convert(workingMemory.iterateObjects()).size()); l = IteratorToList.convert( workingMemory.iterateObjects(new ClassObjectFilter(CheeseEqual.class))); assertEquals(1, l.size()); assertEquals(2, ((CheeseEqual) l.get(0)).getPrice()); h = getFactHandle(h, workingMemory); workingMemory.retract(h); workingMemory = getSerialisedStatefulSession(workingMemory); assertEquals(0, IteratorToList.convert(workingMemory.iterateObjects()).size()); TruthMaintenanceSystem tms = ((InternalWorkingMemory) workingMemory).getTruthMaintenanceSystem(); final java.lang.reflect.Field field = tms.getClass().getDeclaredField("assertMap"); field.setAccessible(true); final ObjectHashMap m = (ObjectHashMap) field.get(tms); field.setAccessible(false); assertEquals("assertMap should be empty", 0, m.size()); }
public void testNotWithBindings() throws Exception { final PackageBuilder builder = new PackageBuilder(); builder.addPackageFromDrl( new InputStreamReader(getClass().getResourceAsStream("not_with_bindings_rule_test.drl"))); final Package pkg = builder.getPackage(); final Rule rule = pkg.getRules()[0]; assertTrue(rule.isValid()); assertEquals(0, builder.getErrors().getErrors().length); RuleBase ruleBase = getRuleBase(); ruleBase.addPackage(pkg); ruleBase = serializeRuleBase(ruleBase); StatefulSession workingMemory = ruleBase.newStatefulSession(); List list = new ArrayList(); workingMemory.setGlobal("list", list); final Cheese stilton = new Cheese("stilton", 5); FactHandle stiltonHandle = workingMemory.insert(stilton); final Cheese cheddar = new Cheese("cheddar", 7); FactHandle cheddarHandle = workingMemory.insert(cheddar); final PersonInterface paul = new Person("paul", "stilton", 12); workingMemory.insert(paul); ruleBase = serializeRuleBase(ruleBase); workingMemory = serializeWorkingMemory(ruleBase, workingMemory); list = (List) workingMemory.getGlobal("list"); stiltonHandle = updateHandle(workingMemory, stiltonHandle); cheddarHandle = updateHandle(workingMemory, cheddarHandle); workingMemory.fireAllRules(); assertEquals(0, list.size()); workingMemory.retract(stiltonHandle); workingMemory.fireAllRules(); assertEquals(1, list.size()); }
/** * Compiles the given rule strings into a single rulebase * * @param rules * @return * @throws Exception */ public static RuleBase compile(List<String> rules) throws Exception { Properties properties = new Properties(); properties.setProperty("drools.dialect.java.compiler", "JANINO"); properties.setProperty("drools.dialect.default", "java"); properties.setProperty("drools.dialect.java.compiler.lnglevel", "1.5"); // properties.setProperty("drools.dump.dir", "/tmp"); // log.debug("drools.dump.dir value is set to " + dumpDir); PackageBuilderConfiguration conf = new PackageBuilderConfiguration(properties); PackageBuilder builder = new PackageBuilder(conf); RuleBase ruleBase = RuleBaseFactory.newRuleBase(); for (String rule : rules) { StringReader drl = new StringReader(rule); builder.addPackageFromDrl(drl); PackageBuilderErrors errors = builder.getErrors(); for (DroolsError error : errors.getErrors()) { log.warn(error); } } ruleBase.addPackage(builder.getPackage()); return ruleBase; }
public void execute() throws MojoExecutionException { // find all the rules items and load them into a package try { // Need to load the build classpath @SuppressWarnings("unchecked") List<Dependency> dependencies = project.getDependencies(); List<URL> url = new ArrayList<URL>(); url.add(outputDirectory.toURI().toURL()); for (Dependency d : dependencies) { String scope = d.getScope(); if (!Artifact.SCOPE_TEST.equals(scope)) { Artifact artifact = getArtifact( d.getGroupId(), d.getArtifactId(), d.getVersion(), d.getType(), d.getClassifier()); url.add(artifact.getFile().toURI().toURL()); } } URL[] classpath = url.toArray(new URL[url.size()]); URLClassLoader uc = new URLClassLoader(classpath, this.getClass().getClassLoader()) { @Override public Class<?> loadClass(String name) throws ClassNotFoundException { getLog().debug("Loading Class for compile [" + name + "]"); Class<?> c = super.loadClass(name); getLog().debug("Loading Class for compile [" + name + "] found [" + c + "]"); return c; } @Override protected Class<?> findClass(String name) throws ClassNotFoundException { getLog().debug("Finding Class for compile [" + name + "]"); Class<?> c = super.findClass(name); getLog().debug("Finding Class for compile [" + name + "] found [" + c + "]"); return c; } }; URLClassLoader uc2 = new URLClassLoader(classpath, this.getClass().getClassLoader()) { @Override public Class<?> loadClass(String name) throws ClassNotFoundException { getLog().debug("Loading Class for runtime [" + name + "]"); Class<?> c = super.loadClass(name); getLog().debug("Loading Class for runtime [" + name + "] found [" + c + "]"); return c; } @Override protected Class<?> findClass(String name) throws ClassNotFoundException { getLog().debug("Finding Class for runtime [" + name + "]"); Class<?> c = super.findClass(name); getLog().debug("Finding Class for runtime [" + name + "] found [" + c + "]"); return c; } }; getLog().info("Package Class loader is using classpath " + Arrays.toString(uc.getURLs())); listClassloader(" ", uc); PackageBuilderConfiguration packageBuilderConfiguration = new PackageBuilderConfiguration(uc); PackageBuilder pb = new PackageBuilder(packageBuilderConfiguration); DirectoryScanner ds = new DirectoryScanner(); ds.setIncludes(includes); ds.setExcludes(excludes); ds.setBasedir(rulesdir); ds.setCaseSensitive(true); ds.scan(); String[] files = ds.getIncludedFiles(); for (String file : files) { File f = new File(rulesdir, file); Reader reader = new FileReader(f); try { if (file.endsWith(".drl")) { getLog().info("Adding Rules " + f); pb.addPackageFromDrl(reader); } else if (file.endsWith(".xml")) { getLog().info("Adding Package definition " + f); pb.addPackageFromXml(reader); } else if (file.endsWith(".rf")) { getLog().info("Adding Rule Flow " + f); pb.addRuleFlow(reader); } else { getLog().info("Ignored Resource " + f); } } finally { reader.close(); } } pb.compileAll(); PackageBuilderErrors errors = pb.getErrors(); if (errors.size() > 0) { for (KnowledgeBuilderError kberr : errors) { getLog().error(kberr.toString()); } throw new MojoExecutionException("Package is not valid"); } org.drools.rule.Package p = pb.getPackage(); if (!p.isValid()) { getLog().error("Package is not valid "); throw new MojoExecutionException("Package is not valid"); } File outputFile = getOutputFile(); getLog().info("Saving compiled package to " + outputFile.getPath()); outputFile.getParentFile().mkdirs(); FileOutputStream fout = new FileOutputStream(outputFile); DroolsStreamUtils.streamOut(fout, p); fout.close(); getLog().info("Testing Compiled package " + outputFile.getPath()); File inputFile = getOutputFile(); FileInputStream fin = new FileInputStream(inputFile); RuleBaseConfiguration config = new RuleBaseConfiguration(uc2); RuleBase ruleBase = RuleBaseFactory.newRuleBase(config); Object o = DroolsStreamUtils.streamIn(fin, uc); ruleBase.addPackage((Package) o); KnowledgeBase kb = new KnowledgeBaseImpl(ruleBase); @SuppressWarnings("unused") StatelessKnowledgeSession session = kb.newStatelessKnowledgeSession(); getLog().info("Testing passed "); } catch (Exception e) { getLog().error(e); throw new MojoExecutionException(e.getMessage()); } }