1
19
20 package com.liferay.portal.dao.shard;
21
22 import com.liferay.portal.util.PropsValues;
23
24 import java.util.Map;
25
26 import javax.sql.DataSource;
27
28 import org.springframework.aop.TargetSource;
29
30
36 public class ShardDataSourceTargetSource implements TargetSource {
37
38 public DataSource getDataSource() {
39 return _dataSourceThreadLocal.get();
40 }
41
42 public Object getTarget() throws Exception {
43 return getDataSource();
44 }
45
46 public Class<DataSource> getTargetClass() {
47 return DataSource.class;
48 }
49
50 public boolean isStatic() {
51 return false;
52 }
53
54 public void releaseTarget(Object target) throws Exception {
55 }
56
57 public void setDataSource(String shardName) {
58 _dataSourceThreadLocal.set(_dataSources.get(shardName));
59 }
60
61 public void setDataSources(Map<String, DataSource> dataSources) {
62 _dataSources = dataSources;
63 }
64
65 private static Map<String, DataSource> _dataSources;
66
67 private static ThreadLocal<DataSource> _dataSourceThreadLocal =
68 new ThreadLocal<DataSource>() {
69
70 protected DataSource initialValue() {
71 return _dataSources.get(PropsValues.SHARD_DEFAULT_NAME);
72 }
73
74 };
75
76 }