b2c / titan · Files

package com.diligrp.titan.common.datasource;

import org.springframework.util.Assert;

import java.util.Random;

/**
 * 数据源切换.
 * @author zhoumin@360buy.com
 * 2011-9-13 下午6:11:40
 */
public class DataSourceSwitcher {

	// 读库数量
	public static final String READ = "slave";
    public static final String WRITE = "master";

	public static final Random RANDOM = new Random(17);

	private static final ThreadLocal<String> CONTEX_THOLDER = new ThreadLocal<String>();

	public static void setDataSource(String dataSource) {
		Assert.notNull(dataSource, "dataSource cannot be null");
		CONTEX_THOLDER.set(dataSource);
	}

	public static String getDataSource() {
		return CONTEX_THOLDER.get();
	}
}