MannersBenchmark.java

package org.drools.integrationtests.manners;

import org.drools.RuleBase;
import org.drools.RuleBaseFactory;
import org.drools.StatefulSession;
import org.drools.common.InternalRuleBase;
import org.drools.compiler.PackageBuilder;
import org.drools.reteoo.ObjectTypeNode;
import org.drools.reteoo.compiled.CompiledNetwork;
import org.drools.reteoo.compiled.ObjectTypeNodeCompiler;
import org.drools.rule.Package;

import java.io.*;
import java.util.*;

public class MannersBenchmark {
    /**
     * Number of guests at the dinner (default: 16).
     */
    private int numGuests = 16;

    /**
     * Number of seats at the table (default: 16).
     */
    private int numSeats = 16;

    /**
     * Minimum number of hobbies each guest should have (defaJavaCompilerFactoryult: 2).
     */
    private int minHobbies = 2;

    /**
     * Maximun number of hobbies each guest should have (default: 3).
     */
    private int maxHobbies = 3;

    public static void main(final String[] args) throws Exception {
        PackageBuilder builder = new PackageBuilder();
        builder.addPackageFromDrl(new InputStreamReader(MannersBenchmark.class.getResourceAsStream("manners.drl")));
        Package pkg = builder.getPackage();

        // add the package to a rulebase
        final RuleBase ruleBase = RuleBaseFactory.newRuleBase();
        ruleBase.addPackage(pkg);

        for (ObjectTypeNode node : ((InternalRuleBase) ruleBase).getRete().getObjectTypeNodes()) {
            CompiledNetwork compiledNetwork = ObjectTypeNodeCompiler.compile(builder, node);
            node.setCompiledNetwork(compiledNetwork);
        }

        String filename;
        if (args.length != 0) {
            String arg = args[0];
            filename = arg;
        } else {
            filename = "manners128.dat";
        }

        for (int i = 0; i < 3; ++i) {
            InputStream is = MannersBenchmark.class.getResourceAsStream(filename);
            List list = getInputObjects(is);
            StatefulSession session = ruleBase.newStatefulSession();

            for (Iterator it = list.iterator(); it.hasNext();) {
                Object object = it.next();
                session.insert(object);
            }

            session.insert(new Count(1));

            long start = System.currentTimeMillis();
            session.fireAllRules();
            System.err.println(System.currentTimeMillis() - start);
            session.dispose();
        }
    }

    /**
     * Convert the facts from the <code>InputStream</code> to a list of
     * objects.
     */
    protected static List getInputObjects(InputStream inputStream) throws IOException {
        List list = new ArrayList();

        BufferedReader br = new BufferedReader(new InputStreamReader(inputStream));

        String line;
        while ((line = br.readLine()) != null) {
            if (line.trim().length() == 0 || line.trim().startsWith(";")) {
                continue;
            }
            StringTokenizer st = new StringTokenizer(line,
                    "() ");
            String type = st.nextToken();

            if ("guest".equals(type)) {
                if (!"name".equals(st.nextToken())) {
                    throw new IOException("expected 'name' in: " + line);
                }
                String name = st.nextToken();
                if (!"sex".equals(st.nextToken())) {
                    throw new IOException("expected 'sex' in: " + line);
                }
                String sex = st.nextToken();
                if (!"hobby".equals(st.nextToken())) {
                    throw new IOException("expected 'hobby' in: " + line);
                }
                String hobby = st.nextToken();

                Guest guest = new Guest(name,
                        Sex.resolve(sex),
                        Hobby.resolve(hobby));

                list.add(guest);
            }

            if ("last_seat".equals(type)) {
                if (!"seat".equals(st.nextToken())) {
                    throw new IOException("expected 'seat' in: " + line);
                }
                list.add(new LastSeat(new Integer(st.nextToken()).intValue()));
            }

            if ("context".equals(type)) {
                if (!"state".equals(st.nextToken())) {
                    throw new IOException("expected 'state' in: " + line);
                }
                list.add(new Context(st.nextToken()));
            }
        }
        inputStream.close();

        return list;
    }

    private InputStream generateData() {
        final String LINE_SEPARATOR = System.getProperty("line.separator");

        StringWriter writer = new StringWriter();

        int maxMale = numGuests / 2;
        int maxFemale = numGuests / 2;

        int maleCount = 0;
        int femaleCount = 0;

        // init hobbies
        List hobbyList = new ArrayList();
        for (int i = 1; i <= maxHobbies; i++) {
            hobbyList.add("h" + i);
        }

        Random rnd = new Random();
        for (int i = 1; i <= numGuests; i++) {
            char sex = rnd.nextBoolean() ? 'm' : 'f';
            if (sex == 'm' && maleCount == maxMale) {
                sex = 'f';
            }
            if (sex == 'f' && femaleCount == maxFemale) {
                sex = 'm';
            }
            if (sex == 'm') {
                maleCount++;
            }
            if (sex == 'f') {
                femaleCount++;
            }

            List guestHobbies = new ArrayList(hobbyList);

            int numHobbies = minHobbies + rnd.nextInt(maxHobbies - minHobbies + 1);
            for (int j = 0; j < numHobbies; j++) {
                int hobbyIndex = rnd.nextInt(guestHobbies.size());
                String hobby = (String) guestHobbies.get(hobbyIndex);
                writer.write("(guest (name n" + i + ") (sex " + sex + ") (hobby " + hobby + "))" + LINE_SEPARATOR);
                guestHobbies.remove(hobbyIndex);
            }
        }
        writer.write("(last_seat (seat " + numSeats + "))" + LINE_SEPARATOR);

        writer.write(LINE_SEPARATOR);
        writer.write("(context (state start))" + LINE_SEPARATOR);

        return new ByteArrayInputStream(writer.getBuffer().toString().getBytes());
    }

}