src/org/objectweb/cjdbc/common/util/ControllerTask.java

説明を見る。

package org.objectweb.cjdbc.common.util;

import java.io.File;
import java.io.FilenameFilter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Arrays;

import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Task;

public class ControllerTask extends Task
{
  private String                configuration;
  private String                action;
  private String                controllerFile;
  private boolean               manageHsqldb           = true;
  private boolean               verbose                = false;
  private static final String   SINGLE_DB              = "SingleDB";
  private static final String   RAIDB0                 = "Raidb0";
  private static final String   RAIDB1                 = "Raidb1";
  private static final String   RAIDB2                 = "Raidb2";
  private static final String   DIRECT                 = "Direct";

  private static final String   HSQLDB_PORT_STANDALONE = "9000";
  private static final String   HSQLDB_PORT_DUAL_1     = "9001";
  private static final String   HSQLDB_PORT_DUAL_2     = "9002";

  private static final String[] FILES_TO_CLEAN         = new String[]{
      "cjdbc.log", "cjdbc.report", "request.log", "test1.script", "test1.data",
      "test2.script", "test2.data", "test1.properties", "test2.properties"};
  private static final String[] CONFIGS                = new String[]{
      SINGLE_DB, RAIDB0, RAIDB1, RAIDB2, DIRECT        };

  public void execute() throws BuildException
  {
    System.out.println("Manage hsql is set to " + manageHsqldb);
    if (action.equalsIgnoreCase("start"))
      startAction();
    else
      stopAction();
  }

  private void startAction() throws BuildException
  {
    System.out.println("Starting CJDBC Controller in mode:" + configuration);
    if (configuration == null)
      throw new BuildException(
          "Need a configuration file to start the controller");
    if (configuration.equals(DIRECT))
      startDirect();
    else if (configuration.equals(SINGLE_DB))
      startSingleDB();
    else if (configuration.equals(RAIDB0))
      startRaidb0();
    else if (configuration.equals(RAIDB1))
      startRaidb1();
    else if (configuration.equals(RAIDB2))
      startRaidb2();
    else
      throw new BuildException("Doing nothing ...");
  }

  private void stopAction() throws BuildException
  {
    System.out.println("Stopping CJDBC Controller from mode:" + configuration);
    if (configuration.equals(DIRECT))
      stopDirect();
    else if (configuration.equals(SINGLE_DB))
      stopSingleDB();
    else if (configuration.equals(RAIDB0))
      stopRaidb0();
    else if (configuration.equals(RAIDB1))
      stopRaidb1();
    else if (configuration.equals(RAIDB2))
      stopRaidb2();
    else
    {
      throw new BuildException("Doing nothing ...");
    }

    if (manageHsqldb)
    {
      cleanFiles();
    }
  }

  private void cleanFiles()
  {
    File f = new File(".");
    String[] list = f.list(new FilenameFilter()
    {
      public boolean accept(File dir, String name)
      {
        for (int i = 0; i < FILES_TO_CLEAN.length; i++)
        {
          if (name.equals(FILES_TO_CLEAN[i]))
            return true;
        }
        return false;
      }
    });
    for (int i = 0; i < list.length; i++)
    {
      if (verbose)
        System.out.println("Cleaning:" + list[i]);
      new File(list[i]).delete();
    }
  }

  private void startController() throws BuildException
  {
    System.out.println("Starting cjdbc with file:" + controllerFile);
    try
    {
      Runtime run = Runtime.getRuntime();
      String path = "loadtest/configfiles:3rdparty/hsqldb/lib/hsqldb.jar:lib/jakarta-regexp-1.3.jar:lib/xercesImpl.jar:lib/xmlutil.jar:lib/Octopus-2.2.jar:lib/OctopusGenerator.jar:lib/csvjdbc.jar:lib/c-jdbc-controller.jar:drivers/c-jdbc-driver.jar:3rdparty/hsqldb/lib/hsqldb.jar:drivers:xml:lib/crimson.jar:lib/jgroups-core.jar:lib/log4j.jar:lib/commons-cli.jar:lib/jmx/mx4j-tools.jar:lib/jmx/xsl:lib/jmx/xml-apis.jar:lib/jmx/mx4j-jmx.jar:lib/jmx/xalan.jar";
      run
          .exec("java -classpath build/classes:"
              + path
              + " -Dorg.xml.sax.driver=org.apache.crimson.parser.XMLReaderImpl org.objectweb.cjdbc.controller.core.Controller -f "
              + controllerFile);
    }
    catch (Exception e)
    {
      throw new BuildException(e.getMessage());
    }
  }

  private void stopController() throws BuildException
  {
    //ControllerShutdownClient.main(new String[]{"localhost", "25322", "3"});
  }

  private void startHsqldb(String port) throws BuildException
  {
    startHsqldb(port, "test");
  }

  private void startHsqldb(String port, String db) throws BuildException
  {
    System.out.println("Starting hsqldb on port:" + port);
    try
    {
      Runtime run = Runtime.getRuntime();
      run
          .exec("java -classpath 3rdparty/hsqldb/lib/hsqldb.jar org.hsqldb.Server -port "
              + port + " -database " + db);
      synchronized (this)
      {
        wait(3000);
      }
    }
    catch (Exception e)
    {
      throw new BuildException(e.getMessage());
    }
  }

  private void stopHsqldb(String port)
  {
    System.out.println("Stopping hsqldb on port:" + port);
    try
    {
      Class.forName("org.hsqldb.jdbcDriver");
      Connection c = DriverManager.getConnection(
          "jdbc:hsqldb:hsql://localhost:" + port, "TEST", "");
      Statement s = c.createStatement();
      s.execute("SHUTDOWN IMMEDIATELY;");
    }
    catch (Exception e)
    {
      // ignore throw new BuildException(e.getMessage());
      System.out.println(e.getMessage());
    }
  }

  private void startDirect()
  {
    startHsqldb(HSQLDB_PORT_STANDALONE, "test1");
  }

  private void stopDirect()
  {
    stopHsqldb(HSQLDB_PORT_STANDALONE);
  }

  private void startSingleDB()
  {
    startHsqldb(HSQLDB_PORT_STANDALONE);
  }

  private void stopSingleDB()
  {
    stopHsqldb(HSQLDB_PORT_STANDALONE);
  }

  private void startRaidb0()
  {
    startHsqldb(HSQLDB_PORT_DUAL_1);
    startHsqldb(HSQLDB_PORT_DUAL_2);
  }

  private void stopRaidb0()
  {
    stopHsqldb(HSQLDB_PORT_DUAL_1);
    stopHsqldb(HSQLDB_PORT_DUAL_2);
  }

  private void startRaidb1()
  {
    if (manageHsqldb)
    {
      startHsqldb(HSQLDB_PORT_DUAL_1, "test1");
      startHsqldb(HSQLDB_PORT_DUAL_2, "test2");
    }
    startController();
  }

  private void stopRaidb1()
  {
    stopController();
    if (manageHsqldb)
    {
      stopHsqldb(HSQLDB_PORT_DUAL_1);
      stopHsqldb(HSQLDB_PORT_DUAL_2);
    }
  }

  private void startRaidb2()
  {
    startHsqldb(HSQLDB_PORT_DUAL_1);
    startHsqldb(HSQLDB_PORT_DUAL_2);
  }

  private void stopRaidb2()
  {
    stopHsqldb(HSQLDB_PORT_DUAL_1);
    stopHsqldb(HSQLDB_PORT_DUAL_2);
  }

  public void setConfiguration(String config) throws BuildException
  {
    ArrayList possible = new ArrayList(Arrays.asList(CONFIGS));
    if (possible.contains(config) == false)
      throw new BuildException("Configuration " + config + " not supported");
    else
      this.configuration = config;
  }

  public void setAction(String action)
  {
    this.action = action;
  }

  public void setControllerfile(String file)
  {
    this.controllerFile = file;
  }

  public void setManageHsqldb(boolean manage)
  {
    this.manageHsqldb = manage;
  }
}

---