org.frameworkset.persitent.datasource.RandomBalanceDatasource Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bboss-persistent Show documentation

bboss is a j2ee framework include aop/ioc,mvc,persistent,taglib,rpc,event ,bean-xml serializable and so on.http://www.bbossgroups.com

The newest version!

package org.frameworkset.persitent.datasource;
/**
 * Copyright 2023 bboss
 * 
 * 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.
 */

import com.frameworkset.commons.dbcp.BasicDataSourceFactory;
import com.frameworkset.orm.adapter.DB;
import org.frameworkset.balance.RoundRobinList;

import javax.sql.DataSource;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.*;
import java.util.logging.Logger;

/**
 * 
Description: 
 * 
 *
 * @author biaoping.yin
 * @Date 2023/12/26
 */
public class RandomBalanceDatasource  extends BalanceDatasource {

    private final ThreadLocal randomThreadLocal = new ThreadLocal<>();
    public RandomBalanceDatasource(String url, DB db, Properties properties)  {
        super(  url,   db,   properties);
    }
    @Override
    protected void buildAddress() throws Exception{
        DataSource currentDatasource = null;
        for(String node: nodes){

            currentDatasource = BasicDataSourceFactory.createDBCP2DataSource(node,properties);
            balanceDatasources.put(node,currentDatasource);
            if(this.currentDatasource == null){
                this.currentDatasource = currentDatasource;
            }
        }

    }
    @Override
    protected String getAnyUrl() throws SQLException {
        List localEnabledUrls = nodes;
        if (localEnabledUrls.isEmpty()) {
            throw new SQLException("Unable to get connection: there are no enabled urls");
        }
        Random random = this.randomThreadLocal.get();
        if (random == null) {
            this.randomThreadLocal.set(new Random());
            random = this.randomThreadLocal.get();
        }

        int index = random.nextInt(localEnabledUrls.size());
        return localEnabledUrls.get(index);
    }
}