Menu
▾
▴
[Jboss-svn-commits] JBL Code SVN: r3928 - labs/jbossrules/trunk/drools-examples/src/test/java/com/sample/benchmark
|
From: <jbo...@li...> - 2006-04-23 18:39:23
|
Author: woolfel
Date: 2006-04-23 14:39:20 -0400 (Sun, 23 Apr 2006)
New Revision: 3928
Added:
labs/jbossrules/trunk/drools-examples/src/test/java/com/sample/benchmark/SharedNode2CEPerformanceTest.java
Log:
adding a new unit test for rules with 2 CE and shared nodes
peter
Added: labs/jbossrules/trunk/drools-examples/src/test/java/com/sample/benchmark/SharedNode2CEPerformanceTest.java
===================================================================
--- labs/jbossrules/trunk/drools-examples/src/test/java/com/sample/benchmark/SharedNode2CEPerformanceTest.java 2006-04-23 18:38:45 UTC (rev 3927)
+++ labs/jbossrules/trunk/drools-examples/src/test/java/com/sample/benchmark/SharedNode2CEPerformanceTest.java 2006-04-23 18:39:20 UTC (rev 3928)
@@ -0,0 +1,486 @@
+package com.sample.benchmark;
+
+import junit.framework.TestCase;
+
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.Reader;
+import java.util.ArrayList;
+import java.util.Iterator;
+
+import org.drools.PackageIntegrationException;
+import org.drools.RuleBase;
+import org.drools.RuleIntegrationException;
+import org.drools.WorkingMemory;
+import org.drools.compiler.DrlParser;
+import org.drools.compiler.DroolsParserException;
+import org.drools.compiler.PackageBuilder;
+import org.drools.lang.descr.PackageDescr;
+import org.drools.reteoo.RuleBaseImpl;
+import org.drools.rule.InvalidPatternException;
+import org.drools.rule.Package;
+
+import com.sample.benchmark.models.Account;
+import com.sample.benchmark.models.Address;
+
+/**
+ * @author Peter Lin
+ *
+ */
+public class SharedNode2CEPerformanceTest extends TestCase {
+ public SharedNode2CEPerformanceTest() {
+ super();
+ }
+
+ public void setUp() {
+
+ }
+
+ public void tearDown() {
+
+ }
+
+ private static RuleBase readRule(String file) throws IOException, DroolsParserException, RuleIntegrationException, PackageIntegrationException, InvalidPatternException {
+ //read in the source
+ Reader reader =
+ new InputStreamReader(
+ SharedNode2CEPerformanceTest.class.getResourceAsStream( file ) );
+ DrlParser parser = new DrlParser();
+ PackageDescr packageDescr = parser.parse( reader );
+
+ //pre build the package
+ PackageBuilder builder = new PackageBuilder();
+ builder.addPackage( packageDescr );
+ Package pkg = builder.getPackage();
+
+ //add the package to a rulebase
+ RuleBaseImpl ruleBase = new RuleBaseImpl();
+ ruleBase.addPackage( pkg );
+ return ruleBase;
+ }
+
+ /**
+ * test the performance with 50 facts and 5 rules. The test measures
+ * the load, assert and fire time.
+ */
+ public void testFiveSharedNodes() {
+ try {
+ int factCount = 5000;
+ String file = "20rules_2CE_5shared.drl";
+ int loop = 5;
+ long totalload = 0;
+ long totalassert = 0;
+ long totalfire = 0;
+ long aveloadmem = 0;
+ long aveassertmem = 0;
+ Runtime rt = Runtime.getRuntime();
+ for (int c=0; c < loop; c++) {
+ rt.gc();
+ long memt1 = rt.totalMemory();
+ long memf1 = rt.freeMemory();
+ long used1 = memt1 - memf1;
+ long loadStart = System.currentTimeMillis();
+ RuleBase ruleBase = readRule(file);
+ long loadEnd = System.currentTimeMillis();
+ long memt2 = rt.totalMemory();
+ long memf2 = rt.freeMemory();
+ long used2 = memt2 - memf2;
+ long loadet = loadEnd - loadStart;
+ rt.gc();
+ totalload += loadet;
+ aveloadmem += (used2 - used1);
+ System.out.println("time to load " + file +
+ " " + loadet + "ms");
+ System.out.println("load memory used " + ((used2-used1)/1024) + " kb");
+ WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+ ArrayList objects = new ArrayList();
+ Account acc = new Account();
+ acc.setStatus("standard");
+ objects.add(acc);
+ // create the objects
+ for (int idx=0; idx < factCount; idx++) {
+ Address addr = new Address();
+ addr.setCity("boston");
+ addr.setState("ma");
+ addr.setHouseType("single family");
+ addr.setStatus("not listed");
+ addr.setCountry("usa");
+ addr.setAccountId("acc" + idx);
+ objects.add(addr);
+ }
+ Iterator itr = objects.iterator();
+ long memt3 = rt.totalMemory();
+ long memf3 = rt.freeMemory();
+ long used3 = memt3 - memf3;
+ long assertStart = System.currentTimeMillis();
+ while( itr.hasNext() ) {
+ workingMemory.assertObject(itr.next());
+ }
+ long assertEnd = System.currentTimeMillis();
+ long memt4 = rt.totalMemory();
+ long memf4 = rt.freeMemory();
+ long used4 = memt4 - memf4;
+ long assertet = assertEnd - assertStart;
+ totalassert += assertet;
+ aveassertmem += (used4 - used3);
+ System.out.println("time to assert " + assertet +
+ " ms");
+ System.out.println("assert memory used " + ((used4-used3)/1024) + " kb");
+ long fireStart = System.currentTimeMillis();
+ workingMemory.fireAllRules();
+ long fireEnd = System.currentTimeMillis();
+ long fireet = fireEnd - fireStart;
+ totalfire += fireet;
+ System.out.println("time to fireAllRules " + fireet +
+ " ms");
+ workingMemory.dispose();
+ rt.gc();
+ System.out.println("");
+ }
+ System.out.println(file);
+ System.out.println("number of objects asserted " + factCount);
+ System.out.println("average load " + (totalload/loop) + " ms");
+ System.out.println("average load mem " + (aveloadmem/1024/loop) + " kb");
+ System.out.println("average assert " + (totalassert/loop) + " ms");
+ System.out.println("average assert mem " + (aveassertmem/1024/loop) + " kb");
+ System.out.println("average fire " + (totalfire/loop) + " ms");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ /**
+ *
+ *
+ */
+ public void testFourSharedNodes() {
+ try {
+ int factCount = 5000;
+ String file = "20rules_2CE_4shared.drl";
+ int loop = 5;
+ long totalload = 0;
+ long totalassert = 0;
+ long totalfire = 0;
+ Runtime rt = Runtime.getRuntime();
+ for (int c=0; c < loop; c++) {
+ rt.gc();
+ long loadStart = System.currentTimeMillis();
+ RuleBase ruleBase = readRule(file);
+ long loadEnd = System.currentTimeMillis();
+ long loadet = loadEnd - loadStart;
+ totalload += loadet;
+ rt.gc();
+ System.out.println("time to load " + file +
+ " " + loadet + "ms");
+ WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+ ArrayList objects = new ArrayList();
+ Account acc = new Account();
+ acc.setStatus("standard");
+ objects.add(acc);
+ // create the objects
+ for (int idx=0; idx < factCount; idx++) {
+ Address addr = new Address();
+ addr.setCity("boston");
+ addr.setState("ma");
+ addr.setHouseType("single family");
+ addr.setStatus("not listed");
+ addr.setCountry("usa" + idx);
+ objects.add(addr);
+ }
+ Iterator itr = objects.iterator();
+ long assertStart = System.currentTimeMillis();
+ while( itr.hasNext() ) {
+ workingMemory.assertObject(itr.next());
+ }
+ long assertEnd = System.currentTimeMillis();
+ long assertet = assertEnd - assertStart;
+ totalassert += assertet;
+ rt.gc();
+ System.out.println("time to assert " + assertet +
+ " ms");
+ long fireStart = System.currentTimeMillis();
+ workingMemory.fireAllRules();
+ long fireEnd = System.currentTimeMillis();
+ long fireet = fireEnd - fireStart;
+ totalfire += fireet;
+ System.out.println("time to fireAllRules " + fireet +
+ " ms");
+ workingMemory.dispose();
+ System.out.println("");
+ rt.gc();
+ }
+ System.out.println(file);
+ System.out.println("number of objects asserted " + factCount);
+ System.out.println("average load " + (totalload/loop) + " ms");
+ System.out.println("average assert " + (totalassert/loop) + " ms");
+ System.out.println("average fire " + (totalfire/loop) + " ms");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ /**
+ *
+ *
+ */
+ public void testThreeSharedNodes() {
+ try {
+ int factCount = 5000;
+ String file = "20rules_2CE_3shared.drl";
+ int loop = 5;
+ long totalload = 0;
+ long totalassert = 0;
+ long totalfire = 0;
+ Runtime rt = Runtime.getRuntime();
+ for (int c=0; c < loop; c++) {
+ long loadStart = System.currentTimeMillis();
+ RuleBase ruleBase = readRule(file);
+ long loadEnd = System.currentTimeMillis();
+ long loadet = loadEnd - loadStart;
+ totalload += loadet;
+ rt.gc();
+ System.out.println("time to load " + file +
+ " " + loadet + "ms");
+ WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+ ArrayList objects = new ArrayList();
+ Account acc = new Account();
+ acc.setStatus("standard");
+ objects.add(acc);
+ // create the objects
+ for (int idx=0; idx < factCount; idx++) {
+ Address addr = new Address();
+ addr.setCity("boston");
+ addr.setState("ma");
+ addr.setHouseType("single family");
+ addr.setStatus("not listed" + idx);
+ addr.setCountry("usa");
+ objects.add(addr);
+ }
+ Iterator itr = objects.iterator();
+ long assertStart = System.currentTimeMillis();
+ while( itr.hasNext() ) {
+ workingMemory.assertObject(itr.next());
+ }
+ long assertEnd = System.currentTimeMillis();
+ long assertet = assertEnd - assertStart;
+ totalassert += assertet;
+ rt.gc();
+ System.out.println("time to assert " + assertet +
+ " ms");
+ long fireStart = System.currentTimeMillis();
+ workingMemory.fireAllRules();
+ long fireEnd = System.currentTimeMillis();
+ long fireet = fireEnd - fireStart;
+ totalfire += fireet;
+ System.out.println("time to fireAllRules " + fireet +
+ " ms");
+ workingMemory.dispose();
+ rt.gc();
+ }
+ System.out.println(file);
+ System.out.println("number of objects asserted " + factCount);
+ System.out.println("average load " + (totalload/loop) + " ms");
+ System.out.println("average assert " + (totalassert/loop) + " ms");
+ System.out.println("average fire " + (totalfire/loop) + " ms");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ /**
+ *
+ *
+ */
+ public void testTwoSharedNodes() {
+ try {
+ int factCount = 5000;
+ String file = "20rules_2CE_2shared.drl";
+ int loop = 5;
+ long totalload = 0;
+ long totalassert = 0;
+ long totalfire = 0;
+ Runtime rt = Runtime.getRuntime();
+ for (int c=0; c < loop; c++) {
+ long loadStart = System.currentTimeMillis();
+ RuleBase ruleBase = readRule(file);
+ long loadEnd = System.currentTimeMillis();
+ long loadet = loadEnd - loadStart;
+ totalload += loadet;
+ rt.gc();
+ System.out.println("time to load " + file +
+ " " + loadet + "ms");
+ WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+ ArrayList objects = new ArrayList();
+ Account acc = new Account();
+ acc.setStatus("standard");
+ objects.add(acc);
+ // create the objects
+ for (int idx=0; idx < factCount; idx++) {
+ Address addr = new Address();
+ addr.setCity("boston");
+ addr.setState("ma");
+ addr.setHouseType("single family" + idx);
+ addr.setStatus("not listed");
+ addr.setCountry("usa");
+ objects.add(addr);
+ }
+ Iterator itr = objects.iterator();
+ long assertStart = System.currentTimeMillis();
+ while( itr.hasNext() ) {
+ workingMemory.assertObject(itr.next());
+ }
+ long assertEnd = System.currentTimeMillis();
+ long assertet = assertEnd - assertStart;
+ totalassert += assertet;
+ rt.gc();
+ System.out.println("time to assert " + assertet +
+ " ms");
+ long fireStart = System.currentTimeMillis();
+ workingMemory.fireAllRules();
+ long fireEnd = System.currentTimeMillis();
+ long fireet = fireEnd - fireStart;
+ totalfire += fireet;
+ System.out.println("time to fireAllRules " + fireet +
+ " ms");
+ workingMemory.dispose();
+ rt.gc();
+ }
+ System.out.println(file);
+ System.out.println("number of objects asserted " + factCount);
+ System.out.println("average load " + (totalload/loop) + " ms");
+ System.out.println("average assert " + (totalassert/loop) + " ms");
+ System.out.println("average fire " + (totalfire/loop) + " ms");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ public void testOneSharedNodes() {
+ try {
+ int factCount = 5000;
+ String file = "20rules_2CE_1shared.drl";
+ int loop = 5;
+ long totalload = 0;
+ long totalassert = 0;
+ long totalfire = 0;
+ Runtime rt = Runtime.getRuntime();
+ for (int c=0; c < loop; c++) {
+ long loadStart = System.currentTimeMillis();
+ RuleBase ruleBase = readRule(file);
+ long loadEnd = System.currentTimeMillis();
+ long loadet = loadEnd - loadStart;
+ totalload += loadet;
+ rt.gc();
+ System.out.println("time to load " + file +
+ " " + loadet + "ms");
+ WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+ ArrayList objects = new ArrayList();
+ Account acc = new Account();
+ acc.setStatus("standard");
+ objects.add(acc);
+ // create the objects
+ for (int idx=0; idx < factCount; idx++) {
+ Address addr = new Address();
+ addr.setCity("boston");
+ addr.setState("ma" + idx);
+ addr.setHouseType("single family");
+ addr.setStatus("not listed");
+ addr.setCountry("usa");
+ objects.add(addr);
+ }
+ Iterator itr = objects.iterator();
+ long assertStart = System.currentTimeMillis();
+ while( itr.hasNext() ) {
+ workingMemory.assertObject(itr.next());
+ }
+ long assertEnd = System.currentTimeMillis();
+ long assertet = assertEnd - assertStart;
+ totalassert += assertet;
+ rt.gc();
+ System.out.println("time to assert " + assertet +
+ " ms");
+ long fireStart = System.currentTimeMillis();
+ workingMemory.fireAllRules();
+ long fireEnd = System.currentTimeMillis();
+ long fireet = fireEnd - fireStart;
+ totalfire += fireet;
+ System.out.println("time to fireAllRules " + fireet +
+ " ms");
+ workingMemory.dispose();
+ rt.gc();
+ }
+ System.out.println(file);
+ System.out.println("number of objects asserted " + factCount);
+ System.out.println("average load " + (totalload/loop) + " ms");
+ System.out.println("average assert " + (totalassert/loop) + " ms");
+ System.out.println("average fire " + (totalfire/loop) + " ms");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+
+ public void testNoneSharedNodes() {
+ try {
+ int factCount = 5000;
+ String file = "20rules_2CE_0shared.drl";
+ int loop = 5;
+ long totalload = 0;
+ long totalassert = 0;
+ long totalfire = 0;
+ Runtime rt = Runtime.getRuntime();
+ for (int c=0; c < loop; c++) {
+ long loadStart = System.currentTimeMillis();
+ RuleBase ruleBase = readRule(file);
+ long loadEnd = System.currentTimeMillis();
+ long loadet = loadEnd - loadStart;
+ totalload += loadet;
+ rt.gc();
+ System.out.println("time to load " + file +
+ " " + loadet + "ms");
+ WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+ ArrayList objects = new ArrayList();
+ Account acc = new Account();
+ acc.setStatus("standard");
+ objects.add(acc);
+ // create the objects
+ for (int idx=0; idx < factCount; idx++) {
+ Address addr = new Address();
+ addr.setAccountId("acc" + idx);
+ addr.setCity("boston");
+ addr.setState("ma");
+ addr.setHouseType("single family");
+ addr.setStatus("not listed");
+ addr.setCountry("usa");
+ objects.add(addr);
+ }
+ Iterator itr = objects.iterator();
+ long assertStart = System.currentTimeMillis();
+ while( itr.hasNext() ) {
+ workingMemory.assertObject(itr.next());
+ }
+ long assertEnd = System.currentTimeMillis();
+ long assertet = assertEnd - assertStart;
+ totalassert += assertet;
+ rt.gc();
+ System.out.println("time to assert " + assertet +
+ " ms");
+ long fireStart = System.currentTimeMillis();
+ workingMemory.fireAllRules();
+ long fireEnd = System.currentTimeMillis();
+ long fireet = fireEnd - fireStart;
+ totalfire += fireet;
+ System.out.println("time to fireAllRules " + fireet +
+ " ms");
+ workingMemory.dispose();
+ rt.gc();
+ }
+ System.out.println(file);
+ System.out.println("number of objects asserted " + factCount);
+ System.out.println("average load " + (totalload/loop) + " ms");
+ System.out.println("average assert " + (totalassert/loop) + " ms");
+ System.out.println("average fire " + (totalfire/loop) + " ms");
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+}
|
