package com.taover.repository.shardingsphere;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.Resource;

import org.springframework.stereotype.Service;

import com.taover.repository.autoconfigure.ShardingSphereKeyGeneratorConfiguration;

@Service
public class ShardingSphereService {
	private Map<Long, Map<String, String>> CACHED_TABLE_SUFFIX_BY_TENANT = null;
	private Map<String, ShardingKeyGeneratorExt> GENERATOR_HOLDER = new HashMap<String, ShardingKeyGeneratorExt>();
	
	@Resource
	private ShardingInfoRepository shardingInfoRepository;
	private ShardingSphereKeyGeneratorConfiguration config;

	public ShardingSphereService(ShardingSphereKeyGeneratorConfiguration config) {
		this.config = config;
	}

	public List<Long> generateKeyList(String tableName, int number){
		if(!GENERATOR_HOLDER.containsKey(tableName)) {
			loadShardingKeyGenerator(tableName);
		}
		return GENERATOR_HOLDER.get(tableName).generateKeyList(number);	
	}

	public Map<String, String> getShardingInfoMapByTenantId(Long tenantId) {
		if(CACHED_TABLE_SUFFIX_BY_TENANT == null) {
			loadCacheTableShardingInfoByTenantId();
		}
		return CACHED_TABLE_SUFFIX_BY_TENANT.get(tenantId);
	}
	
	private synchronized void loadCacheTableShardingInfoByTenantId() {
		if(CACHED_TABLE_SUFFIX_BY_TENANT != null) {
			return;
		}		
		List<ShardingInfoEntity> dataList = this.shardingInfoRepository.findListBySql("1=1");
		Map<Long, Map<String, String>> tempData = new HashMap<Long, Map<String, String>>();
		for(ShardingInfoEntity item: dataList) {
			Map<String, String> tempItem = tempData.getOrDefault(item.getTenantId(), new HashMap<String, String>());
			tempItem.put(item.getTableName(), item.getShardingSuffix());
			tempData.put(item.getTenantId(), tempItem);
		}
		CACHED_TABLE_SUFFIX_BY_TENANT = tempData;
	}

	private synchronized void loadShardingKeyGenerator(String tableName) {
		if(GENERATOR_HOLDER.containsKey(tableName)) {
			return;
		}
		ShardingKeyGeneratorExt generator = new ShardingKeyGeneratorImpl();
		generator.loalConfig(config);
		GENERATOR_HOLDER.put(tableName, generator);
	}
}