sharednodeperformancetest.java

jboss规则引擎

package org.drools.benchmark;
/*
 * Copyright 2005 JBoss Inc
 * 
 * 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.
 */
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.RuleBaseFactory;
import org.drools.RuleIntegrationException;
import org.drools.WorkingMemory;
import org.drools.benchmark.models.Address;
import org.drools.compiler.DrlParser;
import org.drools.compiler.DroolsParserException;
import org.drools.compiler.PackageBuilder;
import org.drools.lang.descr.PackageDescr;
import org.drools.rule.InvalidPatternException;
import org.drools.rule.Package;

/**
 * @author Peter Lin
 *
 */
public class SharedNodePerformanceTest {
    
    public static final void main(final String[] args) {
        SharedNodePerformanceTest test = new SharedNodePerformanceTest();
        test.testFiveSharedNodes();
        test.testFourSharedNodes();
        test.testThreeSharedNodes();
        test.testTwoSharedNodes();
        test.testOneSharedNodes();
        test.testNoneSharedNodes();        
    }
    
    private static RuleBase readRule(String file) throws Exception {
        //read in the source
        Reader reader = new InputStreamReader( SharedNodePerformanceTest.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
        RuleBase ruleBase = RuleBaseFactory.newRuleBase();
        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_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();
                // 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_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();
                // 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();
                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_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++ ) {
                rt.gc();
                long loadStart = System.currentTimeMillis();