ERXThrashTest.java

package er.extensions.eof;

import java.util.LinkedList;
import java.util.List;

import com.webobjects.eoaccess.EOUtilities;
import com.webobjects.eocontrol.EOEditingContext;
import com.webobjects.eocontrol.EOEnterpriseObject;
import com.webobjects.foundation.NSArray;

import er.erxtest.Application;
import er.extensions.ERXExtensions;

public class ERXThrashTest {
  static List<ThrashTest> useScheme(String scheme) throws Throwable {
    // System.out.println("useScheme:: start, scheme = \""+scheme+"\"");

    List<ThrashTest> tests = new LinkedList<>();

    String[] parts = scheme.split(":");
    // System.out.println("useScheme:: parts # "+parts.length);

    String testClassName = parts[1];
    String parameter = parts[2];
    String plan = parts[3];

    System.out.println("testClassName = \"" + testClassName + "\"");
    System.out.println("parameter = \"" + parameter + "\"");
    System.out.println("plan = \"" + plan + "\"");

    Class<? extends ThrashTest> testClass = Class.forName(ERXThrashTest.class.getName() + "$" + testClassName).asSubclass(ThrashTest.class);

    String[] plans = plan.split(",");
    for (int planNum = 0; planNum < plans.length; planNum++) {
      String subPlan = plans[planNum];
      // System.out.println("subPlan # "+idx+" = \""+subPlan+"\"");

      String[] planParts = subPlan.split("-");
      if (planParts.length != 2) {
        throw new IllegalArgumentException("Cannot figure out part # " + planNum + " (\"" + planParts[planNum] + "\") in scheme: \"" + scheme + "\"");
      }

      int delay = Integer.parseInt(planParts[0]);
      int count = Integer.parseInt(planParts[1]);

      for (int testNum = 0; testNum < count; testNum++) {
        ThrashTest test = testClass.newInstance();
        test.setDelay(delay);
        test.setParam(parameter);
        tests.add(test);
      }
    }

    return tests;
  }

  public static boolean _useWonder = true;
  
  public static void main(String[] arg) throws Throwable {
    ERXExtensions.initApp(Application.class, arg);

    // System.out.println("ERXThrashTest:: Hear me roar!");

    // Enumeration props = System.getProperties().propertyNames();
    // while (props.hasMoreElements()) { System.out.println("props = "+props.nextElement()); }

    //List<ThrashTest> tests = useScheme(System.getProperty("thrash.scheme", "scheme:SimpleInsertTest::0-100,2-100,5-200,10-100,12-100,15-200,20-100,22-100,25-200"));
    long startTime = System.currentTimeMillis();
    //List<ThrashTest> tests = useScheme(System.getProperty("thrash.scheme", "scheme:SimpleFetchTest::0-100,2-100"));
    List<ThrashTest> tests = useScheme(System.getProperty("thrash.scheme", "scheme:SimpleInsertTest::0-100,2-100"));

    List<Thread> threads = new LinkedList<>();
    for (ThrashTest test : tests) {
      Thread thread = new Thread(test);
      thread.start();
      threads.add(thread);
    }

    for (Thread thread : threads) {
      thread.join();
    }
    System.out.println("ERXThrashTest.main: " + (System.currentTimeMillis() - startTime) + " total time");
  }

  public abstract static class ThrashTest implements Runnable {
    private String _param;
    private int _delay;
    private long _startTime;
    private long _endTime;

    public void setDelay(int delay) {
      _delay = delay;
    }

    public void setParam(String param) {
      _param = param;
    }

    public String param() {
      return _param;
    }

    public EOEditingContext editingContext() {
      EOEditingContext ec = _useWonder ? ERXEC.newEditingContext() : new EOEditingContext();
      return ec;
    }

    protected void start() {
      _startTime = System.currentTimeMillis();
    }

    protected void end() {
      _endTime = System.currentTimeMillis();
    }

    protected void print(String title, String str) {
      System.out.println(this + ": " + title + ": " + str);
    }

    public void run() {
      if (_delay > 0) {
        try {
          Thread.sleep(_delay * 1000L);
        }
        catch (java.lang.InterruptedException ie) {
        }
      }

      start();
      _run();
      if (_endTime == 0) {
        end();
      }

      print("TIME", "duration = " + (_endTime - _startTime));
    }

    protected abstract void _run();
  }

  public static class SimpleFetchTest extends ThrashTest {
    @Override
    protected void _run() {
      EOEditingContext ec = editingContext();
      ec.lock();
      NSArray rows = EOUtilities.objectsForEntityNamed(ec, "Employee");
      ec.unlock();
      print("RESULT", "row count = " + rows.count());
    }
  }

  public static class SimpleInsertTest extends ThrashTest {
    @Override
    protected void _run() {
      EOEditingContext ec = editingContext();

      ec.lock();
      EOEnterpriseObject company = EOUtilities.createAndInsertInstance(ec, "Company");
      company.takeValueForKey("SomeBody.com", "name");
      ec.saveChanges();
      ec.unlock();

      start();

      String lastName = Thread.currentThread().getName() + System.currentTimeMillis() + "Person";

      int iters = 10;

      for (int idx = 0; idx < iters; idx++) {
        ec.lock();
        EOEnterpriseObject eo = EOUtilities.createAndInsertInstance(ec, "Employee");
        eo.addObjectToBothSidesOfRelationshipWithKey(company, "company");
        eo.takeValueForKey(Boolean.TRUE, "manager");
        eo.takeValueForKey("Bob" + idx, "name");
        eo.takeValueForKey(lastName, "state");
        ec.unlock();
      }
      ec.lock();
      ec.saveChanges();
      ec.unlock();

      ec.lock();
      NSArray rowsInserted = EOUtilities.objectsMatchingKeyAndValue(ec, "Employee", "state", lastName);
      boolean insertOk = (rowsInserted.count() == iters);

      for (int idx = 0; idx < iters; idx++) {
        ec.deleteObject((EOEnterpriseObject) rowsInserted.get(idx));
      }
      ec.saveChanges();
      ec.unlock();

      ec.lock();
      NSArray rowsDeleted = EOUtilities.objectsMatchingKeyAndValue(ec, "Employee", "state", lastName);
      ec.unlock();

      boolean deleteOk = (rowsDeleted.count() == 0);

      ec.lock();
      ec.deleteObject(company);
      ec.saveChanges();
      ec.unlock();

      end();

      print("RESULT", ((insertOk && deleteOk) ? "OK" : "FAILED"));
    }
  }
}