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 org.hibernate.SessionFactory;
27
28 import org.springframework.aop.TargetSource;
29
30
37 public class ShardSessionFactoryTargetSource implements TargetSource {
38
39 public SessionFactory getSessionFactory() {
40 return _sessionFactoryThreadLocal.get();
41 }
42
43 public Object getTarget() throws Exception {
44 return getSessionFactory();
45 }
46
47 public Class<?> getTargetClass() {
48 return _sessionFactories.get(PropsValues.SHARD_DEFAULT_NAME).getClass();
49 }
50
51 public boolean isStatic() {
52 return false;
53 }
54
55 public void releaseTarget(Object target) throws Exception {
56 }
57
58 public void setSessionFactory(String shardName) {
59 _sessionFactoryThreadLocal.set(_sessionFactories.get(shardName));
60 }
61
62 public void setSessionFactories(
63 Map<String, SessionFactory> sessionFactories) {
64
65 _sessionFactories = sessionFactories;
66 }
67
68 private static Map<String, SessionFactory> _sessionFactories;
69
70 private static ThreadLocal<SessionFactory> _sessionFactoryThreadLocal =
71 new ThreadLocal<SessionFactory>() {
72
73 protected SessionFactory initialValue() {
74 return _sessionFactories.get(PropsValues.SHARD_DEFAULT_NAME);
75 }
76
77 };
78
79 }