LoadBalancer.java

/*
 * Agiato: A simple no frill Cassandra API
 * Author: Pranab Ghosh
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You may
 * obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0 
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

package agiato.cassandra.connect;

import agiato.cassandra.meta.Host;
import java.util.List;
import org.apache.commons.pool.impl.GenericObjectPool;

/**
 *
 * @author pranab
 */
public abstract class LoadBalancer {
    protected List<HostConnections> hostConnections;

    public LoadBalancer(){
    }
    


    public abstract HostConnections select();

    /**
     * @return the hostConnections
     */
    public List<HostConnections> getHostConnections() {
        return hostConnections;
    }

    /**
     * @param hostConnections the hostConnections to set
     */
    public void setHostConnections(List<HostConnections> hostConnections) {
        this.hostConnections = hostConnections;
    }
    
    public GenericObjectPool findConnectionPool(Host host){
        GenericObjectPool connPool = null;
        for (HostConnections hostConn : hostConnections){
            if (hostConn.getHost() == host){
                connPool = hostConn.getConnectionPool();
                break;
            }
        }
        
        return connPool;
    }
    
    public void associateKeySpace(String keySpace){
        for (HostConnections hostConn : hostConnections){
            hostConn.getConnectionFactory().setKeySpace(keySpace);
        }        
    }
}