package com.walker.store.strategy;

import com.walker.connector.Address;
import com.walker.dbmeta.util.DatabaseUtils;

import java.util.List;

/**
 * 默认的多库单表存储策略。</p>
 * 该策略适用于：
 * <pre>
 * 1、输入表不区分业务，只有一批，如：100个表
 * 2、想把这些表存储到多个数据库中，便于维护
 * </pre>
 * @author shikeying
 * @date 2015年12月18日
 *
 */
public class OnlyMultiDatabaseStrategy extends DatabaseStoreStrategy {

	private int maxTablesInDatabase = 512;
	
	private int maxDatabaseInHost = 16;
	
	/**
	 * 返回每个主机能存放数据库的最大数量
	 * @return
	 */
	public int getMaxDatabaseInHost() {
		return maxDatabaseInHost;
	}

	public void setMaxDatabaseInHost(int maxDatabaseInHost) {
		this.maxDatabaseInHost = maxDatabaseInHost;
	}

	/**
	 * 返回系统设置的每个库存放表的最大值
	 * @return
	 */
	public int getMaxTablesInDatabase() {
		return maxTablesInDatabase;
	}

	public void setMaxTablesInDatabase(int maxTablesInDatabase) {
		this.maxTablesInDatabase = maxTablesInDatabase;
	}

	@Override
	protected Address doQueryAddress(String storeId, List<Address> addresses, Object parameter) {
		// 从现有的库中选择一个，
		if(this.getDefineAddressSize() > 1){
			logger.debug("多库策略：存在多个地址库，查询表数量未超过的库");
			Address usingAddr = this.getMetaDataEngine().getUsingAddress(getStoreId());
			if(usingAddr == null){
				usingAddr = this.getDefineAddressList().get(0);
				logger.debug("还没有使用任何库，需要选择第一个: " + usingAddr);
				return createFirstAddress(usingAddr);
			} else {
				logger.debug("已经使用了数据库：" + usingAddr);
				int tableSize = this.getMetaDataEngine().getTableSize(usingAddr);
				if(tableSize >= getMaxTablesInDatabase()){
					logger.debug("表超过数量，准备切换数据库,tableSize = " + tableSize);
					// 如果存在多个主机，还要确定使用那个服务器
					if(this.getMetaDataEngine().getDatabaseSize(usingAddr) >= getMaxDatabaseInHost()){
						return switchHostServer(usingAddr);
					} else {
						return DatabaseUtils.copyAddress(usingAddr, true, usingAddr.getSuffixIndex()+1, this.defineDbName);
					}
				} else {
					return usingAddr;
				}
			}
			
		} else {
			logger.debug("MultiDbSingleTableStrategy：只有一个地址库");
			Address usingAddr = this.getMetaDataEngine().getUsingAddress(getStoreId());
			if(usingAddr == null){
				logger.debug("还没有使用任何库，需要选择第一个");
				usingAddr = this.getDefineAddressList().get(0);
				return createFirstAddress(usingAddr);
			} else {
				logger.debug("已经使用了数据库：" + usingAddr);
				int tableSize = this.getMetaDataEngine().getTableSize(usingAddr);
				logger.debug("数据库 '" + usingAddr.getServiceName() + "' 表数量 = " + tableSize);
				if(tableSize >= getMaxTablesInDatabase()){
					Address changedAddr = DatabaseUtils.copyAddress(usingAddr, true, usingAddr.getSuffixIndex()+1, this.defineDbName);
					logger.info("表数量超过最大值，更换数据库：" + changedAddr);
					return changedAddr;
				} else {
					return usingAddr;
				}
			}
		}
	}
	
	private Address createFirstAddress(Address addr){
		return DatabaseUtils.copyAddress(addr, true, 1, this.defineDbName);
	}
	
	private Address switchHostServer(Address using){
		for(Address a : this.getDefineAddressList()){
			if(!using.getUrl().equals(a.getUrl()) 
					|| using.getPort() != a.getPort()){
				return DatabaseUtils.copyAddress(a, false, using.getSuffixIndex()+1, this.defineDbName);
			}
		}
		logger.warn("无法切换数据库，还使用目前的。请检查List<Address>配置");
		return DatabaseUtils.copyAddress(using, true, using.getSuffixIndex()+1, this.defineDbName);
	}

	@Override
	protected String doQueryTableName(Address address, String srcTable, Object parameter) {
		return srcTable;
	}

}