src/org/objectweb/cjdbc/controller/scheduler/raidb1/RAIDb1QueryLevelScheduler.java

説明を見る。

package org.objectweb.cjdbc.controller.scheduler.raidb1;

import java.sql.SQLException;

import org.objectweb.cjdbc.common.sql.AbstractWriteRequest;
import org.objectweb.cjdbc.common.sql.ParsingGranularities;
import org.objectweb.cjdbc.common.sql.SelectRequest;
import org.objectweb.cjdbc.common.xml.DatabasesXmlTags;
import org.objectweb.cjdbc.controller.requestmanager.RAIDbLevels;
import org.objectweb.cjdbc.controller.scheduler.AbstractScheduler;

public class RAIDb1QueryLevelScheduler extends AbstractScheduler
{

  //
  // How the code is organized ?
  //
  // 1. Member variables
  // 2. Constructor
  // 3. Request handling
  // 4. Transaction management
  // 5. Debug/Monitoring
  //

  private long   requestId;
  private int    pendingReads;

  // We have to distinguish read and write to wake up only
  // waiting reads or writes according to the situation
  private Object readSync;    // to synchronize on reads completion
  private Object writeSync;   // to synchronize on writes completion

  //
  // Constructor
  //

  public RAIDb1QueryLevelScheduler()
  {
    super(RAIDbLevels.RAIDb1, ParsingGranularities.NO_PARSING);
    requestId = 0;
    pendingReads = 0;
    readSync = new Object();
    writeSync = new Object();
  }

  //
  // Request Handling
  //

  public void scheduleReadRequest(SelectRequest request) throws SQLException
  {
    // Now deal with synchronization
    synchronized (this.writeSync)
    {
      if (getPendingWrites() == 0)
      { // No writes pending, go ahead !
        synchronized (this.readSync)
        {
          request.setId(requestId++);
          pendingReads++;
          if (logger.isDebugEnabled())
            logger.debug("Request "
                + request.getId()
                + (request.isAutoCommit() ? "" : " transaction "
                    + request.getTransactionId()) + " scheduled for read ("
                + pendingReads + " pending reads)");
          return;
        }
      }

      // Wait for the writes completion
      try
      {
        if (logger.isDebugEnabled())
          logger.debug("Request " + request.getId() + " waiting for "
              + getPendingWrites() + " pending writes)");

        int timeout = request.getTimeout();
        if (timeout > 0)
        {
          long start = System.currentTimeMillis();
          // Convert seconds to milliseconds for wait call
          long lTimeout = timeout * 1000;
          this.writeSync.wait(lTimeout);
          long end = System.currentTimeMillis();
          int remaining = (int) (lTimeout - (end - start));
          if (remaining > 0)
            request.setTimeout(remaining);
          else
          {
            String msg = "Timeout (" + request.getTimeout() + ") for request: "
                + request.getId();
            logger.warn(msg);
            throw new SQLException(msg);
          }
        }
        else
          this.writeSync.wait();

        synchronized (this.readSync)
        {
          request.setId(requestId++);
          pendingReads++;
          if (logger.isDebugEnabled())
            logger.debug("Request " + request.getId() + " scheduled for read ("
                + pendingReads + " pending reads)");
          return; // Ok, write completed before timeout
        }
      }
      catch (InterruptedException e)
      {
        // Timeout
        if (logger.isWarnEnabled())
          logger.warn("Request " + request.getId() + " timed out ("
              + request.getTimeout() + " s)");
        throw new SQLException("Timeout (" + request.getTimeout()
            + ") for request: " + request.getId());
      }
    }
  }

  public final void readCompletedNotify(SelectRequest request)
  {
    synchronized (this.readSync)
    {
      pendingReads--;
      if (logger.isDebugEnabled())
        logger.debug("Read request " + request.getId() + " completed - "
            + pendingReads + " pending reads");
      if (pendingReads == 0)
      {
        if (logger.isDebugEnabled())
          logger.debug("Last read completed, notifying writes");
        readSync.notifyAll(); // Wakes up any waiting write query
      }
    }
  }

  public void scheduleNonSuspendedWriteRequest(AbstractWriteRequest request)
      throws SQLException
  {
    // We have to take the locks in the same order as reads else
    // we could have a deadlock
    synchronized (this.writeSync)
    {
      synchronized (this.readSync)
      {
        if (pendingReads == 0)
        { // No read pending, go ahead
          request.setId(requestId++);
          if (logger.isDebugEnabled())
            logger.debug("Request "
                + request.getId()
                + (request.isAutoCommit() ? "" : " transaction "
                    + request.getTransactionId()) + " scheduled for write ("
                + getPendingWrites() + " pending writes)");
          return;
        }
      }
    }

    waitForReadCompletion(request);
    scheduleNonSuspendedWriteRequest(request);
  }

  private void waitForReadCompletion(AbstractWriteRequest request)
      throws SQLException
  {
    synchronized (this.readSync)
    {
      // Wait for the reads completion
      try
      {
        if (logger.isDebugEnabled())
          logger.debug("Request " + request.getId() + " waiting for "
              + pendingReads + " pending reads)");

        int timeout = request.getTimeout();
        if (timeout > 0)
        {
          long start = System.currentTimeMillis();
          // Convert seconds to milliseconds for wait call
          long lTimeout = timeout * 1000;
          this.readSync.wait(lTimeout);
          long end = System.currentTimeMillis();
          int remaining = (int) (lTimeout - (end - start));
          if (remaining > 0)
            request.setTimeout(remaining);
          else
          {
            String msg = "Timeout (" + request.getTimeout() + ") for request: "
                + request.getId();
            logger.warn(msg);
            throw new SQLException(msg);
          }
        }
        else
          this.readSync.wait();
      }
      catch (InterruptedException e)
      {
        // Timeout
        if (logger.isWarnEnabled())
          logger.warn("Request " + request.getId() + " timed out ("
              + request.getTimeout() + " ms)");
        throw new SQLException("Timeout (" + request.getTimeout()
            + ") for request: " + request.getId());
      }
    }
  }

  public final synchronized void notifyWriteCompleted(
      AbstractWriteRequest request)
  {
    synchronized (this.writeSync)
    {
      if (logger.isDebugEnabled())
        logger.debug("Request " + request.getId() + " completed - "
            + getPendingWrites() + " pending writes");
      if (getPendingWrites() == 0)
      {
        if (logger.isDebugEnabled())
          logger.debug("Last write completed, notifying reads");
        writeSync.notifyAll(); // Wakes up all waiting read queries
      }
    }
  }

  //
  // Transaction Management
  //

  protected final void commitTransaction(long transactionId)
  {
  }

  protected final void rollbackTransaction(long transactionId)
  {
  }

  //
  // Debug/Monitoring
  //
  public String getXmlImpl()
  {
    return "<" + DatabasesXmlTags.ELT_RAIDb1Scheduler + " "
        + DatabasesXmlTags.ATT_level + "=\"" + DatabasesXmlTags.VAL_query
        + "\"/>";
  }

}

---