ars.database.spring.DataSourceRouter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ars-database Show documentation

Ars framework database

There is a newer version: 2.0.0

package ars.database.spring;

import java.util.Map;
import java.util.HashMap;
import java.util.Map.Entry;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

import ars.util.Strings;
import ars.invoke.event.InvokeEvent;
import ars.invoke.event.InvokeListener;
import ars.invoke.event.InvokeBeforeEvent;
import ars.invoke.event.InvokeCompleteEvent;

/**
 * 数据源路由器实现
 * 
 * @author yongqiangwu
 * 
 */
public class DataSourceRouter extends AbstractRoutingDataSource implements
		InvokeListener {
	private Map routes = new HashMap();
	private ThreadLocal dataSource = new ThreadLocal();

	public Map getRoutes() {
		return routes;
	}

	public void setRoutes(Map routes) {
		this.routes = routes;
	}

	@Override
	protected Object determineCurrentLookupKey() {
		return this.dataSource.get();
	}

	@Override
	public void onInvokeEvent(InvokeEvent event) {
		if (event instanceof InvokeBeforeEvent) {
			String uri = event.getSource().getUri();
			for (Entry entry : this.routes.entrySet()) {
				if (Strings.matches(uri, entry.getKey())) {
					this.dataSource.set(entry.getValue());
					break;
				}
			}
		} else if (event instanceof InvokeCompleteEvent) {
			this.dataSource.remove();
		}
	}

}