src/org/objectweb/cjdbc/controller/loadbalancer/tasks/WriteRequestWithKeysTask.java

説明を見る。

package org.objectweb.cjdbc.controller.loadbalancer.tasks;

import java.sql.Connection;
import java.sql.SQLException;

import org.objectweb.cjdbc.common.exceptions.UnreachableBackendException;
import org.objectweb.cjdbc.common.i18n.Translate;
import org.objectweb.cjdbc.common.sql.AbstractWriteRequest;
import org.objectweb.cjdbc.common.sql.CreateRequest;
import org.objectweb.cjdbc.common.sql.schema.DatabaseSchema;
import org.objectweb.cjdbc.common.sql.schema.DatabaseTable;
import org.objectweb.cjdbc.controller.backend.DatabaseBackend;
import org.objectweb.cjdbc.controller.cache.metadata.MetadataCache;
import org.objectweb.cjdbc.controller.connection.AbstractConnectionManager;
import org.objectweb.cjdbc.controller.loadbalancer.AbstractLoadBalancer;
import org.objectweb.cjdbc.controller.loadbalancer.BackendWorkerThread;
import org.objectweb.cjdbc.controller.virtualdatabase.ControllerResultSet;

public class WriteRequestWithKeysTask extends AbstractTask
{
  private AbstractWriteRequest request;
  private ControllerResultSet  result;
  private MetadataCache        metadataCache;

  public WriteRequestWithKeysTask(int nbToComplete, int totalNb,
      AbstractWriteRequest request, MetadataCache metadataCache)
  {
    super(nbToComplete, totalNb);
    this.request = request;
    this.metadataCache = metadataCache;
  }

  public void execute(BackendWorkerThread backendThread) throws SQLException
  {
    DatabaseBackend backend = backendThread.getBackend();

    if (!backend.getDriverCompliance().supportGetGeneratedKeys())
      throw new SQLException(Translate.get(
          "loadbalancer.backend.autogeneratedkeys.unsupported", backend
              .getName()));

    AbstractConnectionManager cm = backend.getConnectionManager(request
        .getLogin());
    if (cm == null)
    {
      SQLException se = new SQLException(
          "No Connection Manager for Virtual Login:" + request.getLogin());
      try
      {
        notifyFailure(backendThread, 1, se);
      }
      catch (SQLException ignore)
      {

      }
      throw se;
    }

    if (request.isAutoCommit())
    {
      if (backend.isDisabling())
      {
        // Backend is disabling, we do not execute queries except the one in the
        // transaction we already started. Just notify the completion for the
        // others.
        notifyCompletion();
        return;
      }

      // Use a connection just for this request
      Connection c = null;
      try
      {
        c = cm.getConnection();
      }
      catch (UnreachableBackendException e1)
      {
        SQLException se = new SQLException("Backend " + backend.getName()
            + " is no more reachable.");
        try
        {
          notifyFailure(backendThread, 1, se);
        }
        catch (SQLException ignore)
        {
        }
        backendThread.getLogger().error(
            "Disabling backend " + backend.getName()
                + " because it is no more reachable.");
        backend.disable();
        throw se;
      }

      // Sanity check
      if (c == null)
      {
        SQLException se = new SQLException("No more connections");
        try
        { // All backends failed, just ignore
          if (!notifyFailure(backendThread, (long) request.getTimeout() * 1000,
              se))
            return;
        }
        catch (SQLException ignore)
        {
        }
        // Disable this backend (it is no more in sync) by killing the backend
        // thread
        backendThread.kill();
        String msg = "Request '"
            + request.getSQLShortForm(backend.getSQLShortFormLength())
            + "' failed on backend " + backend.getName() + " but "
            + getSuccess() + " succeeded (" + se + ")";
        backendThread.getLogger().error(msg);
        throw new SQLException(msg);
      }

      // Execute Query
      try
      {
        result = new ControllerResultSet(request, AbstractLoadBalancer
            .executeUpdateRequestOnBackendWithKeys(request, backend, c),
            metadataCache);

        // Update schema
        if (request.isCreate())
        { // Add the table to the schema
          DatabaseSchema dbs = backend.getDatabaseSchema();
          if (dbs != null)
          {
            DatabaseTable t = ((CreateRequest) request).getDatabaseTable();
            if (t != null)
            {
              dbs.addTable(t);
              if (backendThread.getLogger().isDebugEnabled())
                backendThread.getLogger().debug(
                    "Added table '" + request.getTableName()
                        + "' to backend database schema");
            }
          }
        }
        else if (request.isDrop())
        { // Delete the table from the schema
          DatabaseSchema dbs = backend.getDatabaseSchema();
          if (dbs != null)
          {
            DatabaseTable t = dbs.getTable(request.getTableName());
            if (t != null)
            {
              dbs.removeTable(t);
              if (backendThread.getLogger().isDebugEnabled())
                backendThread.getLogger().debug(
                    "Removed table '" + request.getTableName()
                        + "' from backend database schema");
            }
          }
        }
      }
      catch (Exception e)
      {
        try
        { // All backends failed, just ignore
          if (!notifyFailure(backendThread, (long) request.getTimeout() * 1000,
              e))
            return;
        }
        catch (SQLException ignore)
        {
        }
        // Disable this backend (it is no more in sync) by killing the backend
        // thread
        backendThread.kill();
        String msg = "Request '"
            + request.getSQLShortForm(backend.getSQLShortFormLength())
            + "' failed on backend " + backend.getName() + " but "
            + getSuccess() + " succeeded (" + e + ")";
        backendThread.getLogger().error(msg);
        throw new SQLException(msg);
      }
      finally
      {
        cm.releaseConnection(c);
      }
    }
    else
    { // Re-use the connection used by this transaction
      Connection c;
      long tid = request.getTransactionId();
      Long lTid = new Long(tid);

      if (!backend.isStartedTransaction(lTid))
      {
        if (backend.isDisabling())
        {
          // Backend is disabling, we do not start new transactions, just notify
          // the completion for the others
          notifyCompletion();
          return;
        }

        // Transaction has not been started yet, this is a lazy begin
        try
        {
          c = cm.getConnection(tid);
        }
        catch (UnreachableBackendException e1)
        {
          SQLException se = new SQLException("Backend " + backend.getName()
              + " is no more reachable.");
          try
          {
            notifyFailure(backendThread, 1, se);
          }
          catch (SQLException ignore)
          {
          }
          backendThread.getLogger().error(
              "Disabling backend " + backend.getName()
                  + " because it is no more reachable.");
          backend.disable();
          throw se;
        }

        // Sanity check
        if (c == null)
        { // Bad connection
          SQLException se = new SQLException(
              "Unable to get connection for transaction " + tid);
          try
          { // All backends failed, just ignore
            if (!notifyFailure(backendThread,
                (long) request.getTimeout() * 1000, se))
              return;
          }
          catch (SQLException ignore)
          {
          }
          // Disable this backend (it is no more in sync) by killing the
          // backend thread
          backendThread.kill();
          String msg = "Request '"
              + request.getSQLShortForm(backend.getSQLShortFormLength())
              + "' failed on backend " + backend.getName() + " but "
              + getSuccess() + " succeeded (" + se + ")";
          backendThread.getLogger().error(msg);
          throw new SQLException(msg);
        }

        // begin transaction
        backend.startTransaction(lTid);
        c.setAutoCommit(false);
      }
      else
      { // Transaction has already been started, retrieve connection
        c = cm.retrieveConnection(tid);

        // Sanity check
        if (c == null)
        { // Bad connection
          SQLException se = new SQLException(
              "Unable to retrieve connection for transaction " + tid);
          try
          { // All backends failed, just ignore
            if (!notifyFailure(backendThread,
                (long) request.getTimeout() * 1000, se))
              return;
          }
          catch (SQLException ignore)
          {
          }
          // Disable this backend (it is no more in sync) by killing the
          // backend thread
          backendThread.kill();
          String msg = "Request '"
              + request.getSQLShortForm(backend.getSQLShortFormLength())
              + "' failed on backend " + backend.getName() + " but "
              + getSuccess() + " succeeded (" + se + ")";
          backendThread.getLogger().error(msg);
          throw new SQLException(msg);
        }
      }

      // Execute Query
      try
      {
        result = new ControllerResultSet(request, AbstractLoadBalancer
            .executeUpdateRequestOnBackendWithKeys(request, backend, c),
            metadataCache);

        // Update schema
        if (request.isCreate())
        { // Add the table to the schema
          DatabaseSchema dbs = backend.getDatabaseSchema();
          if (dbs != null)
          {
            DatabaseTable t = ((CreateRequest) request).getDatabaseTable();
            if (t != null)
            {
              dbs.addTable(t);
              if (backendThread.getLogger().isDebugEnabled())
                backendThread.getLogger().debug(
                    "Added table '" + request.getTableName()
                        + "' to backend database schema");
            }
          }
        }
        else if (request.isDrop())
        { // Delete the table from the schema
          DatabaseSchema dbs = backend.getDatabaseSchema();
          if (dbs != null)
          {
            DatabaseTable t = dbs.getTable(request.getTableName());
            if (t != null)
            {
              dbs.removeTable(t);
              if (backendThread.getLogger().isDebugEnabled())
                backendThread.getLogger().debug(
                    "Removed table '" + request.getTableName()
                        + "' from backend database schema");
            }
          }
        }
      }
      catch (Exception e)
      {
        try
        { // All backends failed, just ignore
          if (!notifyFailure(backendThread, (long) request.getTimeout() * 1000,
              e))
            return;
        }
        catch (SQLException ignore)
        {
        }
        // Disable this backend (it is no more in sync) by killing the backend
        // thread
        backendThread.kill();
        String msg = "Request '"
            + request.getSQLShortForm(backend.getSQLShortFormLength())
            + "' failed on backend " + backend.getName() + " but "
            + getSuccess() + " succeeded (" + e + ")";
        backendThread.getLogger().error(msg);
        throw new SQLException(msg);
      }
    }
    notifySuccess();
  }

  public ControllerResultSet getResult()
  {
    return result;
  }

  public String toString()
  {
    if (request.isAutoCommit())
      return "WriteWithKeys  Autocommit Task (" + request.getSQL() + ")";
    else
      return "WriteWithKeys Task from transaction:"
          + request.getTransactionId() + "(" + request.getSQL() + ")";
  }

}

---