| ShardSessionFactoryTargetSource.java |
1 /**
2 * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3 *
4 *
5 *
6 *
7 * The contents of this file are subject to the terms of the Liferay Enterprise
8 * Subscription License ("License"). You may not use this file except in
9 * compliance with the License. You can obtain a copy of the License by
10 * contacting Liferay, Inc. See the License for the specific language governing
11 * permissions and limitations under the License, including but not limited to
12 * distribution rights of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20 * SOFTWARE.
21 */
22
23 package com.liferay.portal.dao.shard;
24
25 import com.liferay.portal.util.PropsValues;
26
27 import java.util.Map;
28
29 import org.hibernate.SessionFactory;
30
31 import org.springframework.aop.TargetSource;
32
33 /**
34 * <a href="ShardSessionFactoryTargetSource.java.html"><b><i>View Source</i></b>
35 * </a>
36 *
37 * @author Michael Young
38 */
39 public class ShardSessionFactoryTargetSource implements TargetSource {
40
41 public SessionFactory getSessionFactory() {
42 return _sessionFactoryThreadLocal.get();
43 }
44
45 public Object getTarget() throws Exception {
46 return getSessionFactory();
47 }
48
49 public Class<?> getTargetClass() {
50 return _sessionFactories.get(PropsValues.SHARD_DEFAULT_NAME).getClass();
51 }
52
53 public boolean isStatic() {
54 return false;
55 }
56
57 public void releaseTarget(Object target) throws Exception {
58 }
59
60 public void setSessionFactory(String shardName) {
61 _sessionFactoryThreadLocal.set(_sessionFactories.get(shardName));
62 }
63
64 public void setSessionFactories(
65 Map<String, SessionFactory> sessionFactories) {
66
67 _sessionFactories = sessionFactories;
68 }
69
70 private static Map<String, SessionFactory> _sessionFactories;
71
72 private static ThreadLocal<SessionFactory> _sessionFactoryThreadLocal =
73 new ThreadLocal<SessionFactory>() {
74
75 protected SessionFactory initialValue() {
76 return _sessionFactories.get(PropsValues.SHARD_DEFAULT_NAME);
77 }
78
79 };
80
81 }