cn.echo.sharding.database.MasterSlaveRuleConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sharding-db Show documentation
sharding jdbc for db
The newest version!
package cn.echo.sharding.database;


import com.google.common.base.Preconditions;
import com.google.common.base.Strings;
import org.apache.shardingsphere.api.config.RuleConfiguration;
import org.apache.shardingsphere.api.config.masterslave.LoadBalanceStrategyConfiguration;
import org.apache.shardingsphere.api.config.masterslave.MasterSlaveRuleConfiguration;

import java.util.Collection;

/**
 * Master-slave rule configuration.
 * @author lonyee
 */
public final class MasterSlaveRuleConfig implements RuleConfiguration {

    private  String name;

    private String masterDataSourceName;

    private Collection slaveDataSourceNames;

    private String loadBalanceStrategy;

    public MasterSlaveRuleConfig() {}

    public MasterSlaveRuleConfig(final String name, final String masterDataSourceName, final Collection slaveDataSourceNames) {
        this(name, masterDataSourceName, slaveDataSourceNames, null);
    }

    public MasterSlaveRuleConfig(final String name,
                                        final String masterDataSourceName, final Collection slaveDataSourceNames, final String loadBalanceStrategy) {
        Preconditions.checkArgument(!Strings.isNullOrEmpty(name), "Name is required.");
        Preconditions.checkArgument(!Strings.isNullOrEmpty(masterDataSourceName), "MasterDataSourceName is required.");
        Preconditions.checkArgument(null != slaveDataSourceNames && !slaveDataSourceNames.isEmpty(), "SlaveDataSourceNames is required.");
        this.name = name;
        this.masterDataSourceName = masterDataSourceName;
        this.slaveDataSourceNames = slaveDataSourceNames;
        this.loadBalanceStrategy = loadBalanceStrategy;
    }

    /**
     * 转换成sharding的配置类 - 修复MasterSlaveRuleConfiguration不支持配置
     */
    public MasterSlaveRuleConfiguration parseMasterSlaveRule() {
        MasterSlaveRuleConfiguration config = new MasterSlaveRuleConfiguration(
                this.name,
                this.masterDataSourceName,
                this.slaveDataSourceNames,
                new LoadBalanceStrategyConfiguration(this.loadBalanceStrategy));
        return config;
    }

    public void setName(String name) {
        this.name = name;
    }

    public void setMasterDataSourceName(String masterDataSourceName) {
        this.masterDataSourceName = masterDataSourceName;
    }

    public void setSlaveDataSourceNames(Collection slaveDataSourceNames) {
        this.slaveDataSourceNames = slaveDataSourceNames;
    }

    public void setLoadBalanceStrategy(String loadBalanceStrategy) {
        this.loadBalanceStrategy = loadBalanceStrategy;
    }

    public String getName() {
        return name;
    }

    public String getMasterDataSourceName() {
        return masterDataSourceName;
    }

    public Collection getSlaveDataSourceNames() {
        return slaveDataSourceNames;
    }

    public String getLoadBalanceStrategy() {
        return loadBalanceStrategy;
    }
}