src/org/objectweb/cjdbc/controller/loadbalancer/paralleldb/ParallelDB_LPRF.java

説明を見る。

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

import java.sql.SQLException;
import java.util.ArrayList;

import org.objectweb.cjdbc.common.i18n.Translate;
import org.objectweb.cjdbc.common.sql.AbstractRequest;
import org.objectweb.cjdbc.common.sql.AbstractWriteRequest;
import org.objectweb.cjdbc.common.xml.DatabasesXmlTags;
import org.objectweb.cjdbc.controller.backend.DatabaseBackend;
import org.objectweb.cjdbc.controller.virtualdatabase.VirtualDatabase;

public class ParallelDB_LPRF extends ParallelDB
{

  public ParallelDB_LPRF(VirtualDatabase vdb) throws SQLException
  {
    super(vdb);
  }

  public DatabaseBackend chooseBackendForReadRequest(AbstractRequest request)
      throws SQLException
  {
    // Choose a backend
    try
    {
      vdb.acquireReadLockBackendLists();
    }
    catch (InterruptedException e)
    {
      String msg = Translate.get(
          "loadbalancer.backendlist.acquire.readlock.failed", e);
      logger.error(msg);
      throw new SQLException(msg);
    }

    DatabaseBackend backend = null; // The backend that will execute the query

    // Note that vdb lock is released in the finally clause of this try/catch
    // block
    try
    {
      ArrayList backends = vdb.getBackends();
      int size = backends.size();

      if (size == 0)
        throw new SQLException(Translate.get(
            "loadbalancer.execute.no.backend.available", request.getId()));

      // Choose the backend that has the least pending requests
      int leastRequests = 0;
      for (int i = 0; i < size; i++)
      {
        DatabaseBackend b = (DatabaseBackend) backends.get(i);
        if (b.isReadEnabled())
        {
          int pending = b.getPendingRequests().size();
          if ((backend == null) || (pending < leastRequests))
          {
            backend = b;
            if (pending == 0)
              break; // Stop here we will never find a less loaded node
            else
              leastRequests = pending;
          }
        }
      }

      if (backend == null)
        throw new SQLException(Translate.get(
            "loadbalancer.execute.no.backend.enabled", request.getId()));
    }
    catch (RuntimeException e)
    {
      String msg = Translate.get("loadbalancer.execute.find.backend.failed",
          new String[]{request.getSQLShortForm(vdb.getSQLShortFormLength()),
              e.getMessage()});
      logger.error(msg, e);
      throw new SQLException(msg);
    }
    finally
    {
      vdb.releaseReadLockBackendLists();
    }

    return backend;
  }

  public DatabaseBackend chooseBackendForWriteRequest(
      AbstractWriteRequest request) throws SQLException
  {
    // Choose a backend
    try
    {
      vdb.acquireReadLockBackendLists();
    }
    catch (InterruptedException e)
    {
      String msg = Translate.get(
          "loadbalancer.backendlist.acquire.readlock.failed", e);
      logger.error(msg);
      throw new SQLException(msg);
    }

    DatabaseBackend backend = null; // The backend that will execute the query

    // Note that vdb lock is released in the finally clause of this try/catch
    // block
    try
    {
      ArrayList backends = vdb.getBackends();
      int size = backends.size();

      if (size == 0)
        throw new SQLException(Translate.get(
            "loadbalancer.execute.no.backend.available", request.getId()));

      // Choose the backend that has the least pending requests
      int leastRequests = 0;
      for (int i = 0; i < size; i++)
      {
        DatabaseBackend b = (DatabaseBackend) backends.get(i);
        if (b.isWriteEnabled() && !b.isDisabling())
        {
          int pending = b.getPendingRequests().size();
          if ((backend == null) || (pending < leastRequests))
          {
            backend = b;
            if (pending == 0)
              break; // Stop here we will never find a less loaded node
            else
              leastRequests = pending;
          }
        }
      }

      if (backend == null)
      {
        throw new SQLException(Translate.get(
            "loadbalancer.execute.no.backend.enabled", request.getId()));
      }
    }
    catch (RuntimeException e)
    {
      String msg = Translate.get("loadbalancer.execute.find.backend.failed",
          new String[]{request.getSQLShortForm(vdb.getSQLShortFormLength()),
              e.getMessage()});
      logger.error(msg, e);
      throw new SQLException(msg);
    }
    finally
    {
      vdb.releaseReadLockBackendLists();
    }

    return backend;
  }

  public String getInformation()
  {
    // We don't lock since we don't need a top accurate value
    int size = vdb.getBackends().size();

    if (size == 0)
      return "ParallelDB Least Pending Request First Request load balancer: !!!Warning!!! No backend nodes found\n";
    else
      return "ParallelDB Least Pending Request First Request load balancer ("
          + size + " backends)\n";
  }

  public String getParallelDBXml()
  {
    return "<" + DatabasesXmlTags.ELT_ParallelDB_LeastPendingRequestsFirst
        + "/>";
  }
}

---