[Jboss-svn-commits] JBL Code SVN: r3784 - labs/jbossrules/trunk/drools-examples/src/java/com/sample/benchmark

[<jbo...@li...>]

Author: woolfel
Date: 2006-04-19 00:10:08 -0400 (Wed, 19 Apr 2006)
New Revision: 3784

Added:
   labs/jbossrules/trunk/drools-examples/src/java/com/sample/benchmark/RuleSetBenchmark.java
Log:
adding a simple unit test to measure how long it takes to load a large rule file

peter

Added: labs/jbossrules/trunk/drools-examples/src/java/com/sample/benchmark/RuleSetBenchmark.java
===================================================================
--- labs/jbossrules/trunk/drools-examples/src/java/com/sample/benchmark/RuleSetBenchmark.java	2006-04-19 03:38:47 UTC (rev 3783)
+++ labs/jbossrules/trunk/drools-examples/src/java/com/sample/benchmark/RuleSetBenchmark.java	2006-04-19 04:10:08 UTC (rev 3784)
@@ -0,0 +1,67 @@
+package com.sample.benchmark;
+
+import junit.framework.TestCase;
+
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.Reader;
+
+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;
+
+/**
+ * @author Peter Lin
+ *
+ */
+public class RuleSetBenchmark extends TestCase {
+	public RuleSetBenchmark() {
+		super();
+	}
+	
+	public void setUp() {
+		
+	}
+	
+	public void tearDown() {
+		
+	}
+
+    private static RuleBase readRule() throws IOException, DroolsParserException, RuleIntegrationException, PackageIntegrationException, InvalidPatternException {
+        //read in the source
+        Reader reader = new InputStreamReader( RuleSetBenchmark.class.getResourceAsStream( "1000_rules.drl" ) );
+        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;
+    }
+	
+	public void testOneThousand() {
+		try {
+			long loadStart = System.currentTimeMillis();
+	        RuleBase ruleBase = readRule();
+	        long loadEnd = System.currentTimeMillis();
+	        System.out.println("time to load 1000 rules " +
+	        		(loadEnd - loadStart) + "ms");
+	        WorkingMemory workingMemory = ruleBase.newWorkingMemory();
+		} catch (Exception e) {
+			e.printStackTrace();
+		}
+	}
+}