ShardSessionFactoryTargetSource.java |
1 /** 2 * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved. 3 * 4 * Permission is hereby granted, free of charge, to any person obtaining a copy 5 * of this software and associated documentation files (the "Software"), to deal 6 * in the Software without restriction, including without limitation the rights 7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 8 * copies of the Software, and to permit persons to whom the Software is 9 * furnished to do so, subject to the following conditions: 10 * 11 * The above copyright notice and this permission notice shall be included in 12 * all copies or substantial portions 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 */ 40 public class ShardSessionFactoryTargetSource implements TargetSource { 41 42 public SessionFactory getSessionFactory() { 43 return _sessionFactoryThreadLocal.get(); 44 } 45 46 public Object getTarget() throws Exception { 47 return getSessionFactory(); 48 } 49 50 public Class<?> getTargetClass() { 51 return _sessionFactories.get(PropsValues.SHARD_DEFAULT_NAME).getClass(); 52 } 53 54 public boolean isStatic() { 55 return false; 56 } 57 58 public void releaseTarget(Object target) throws Exception { 59 } 60 61 public void setSessionFactory(String shardName) { 62 _sessionFactoryThreadLocal.set(_sessionFactories.get(shardName)); 63 } 64 65 public void setSessionFactories( 66 Map<String, SessionFactory> sessionFactories) { 67 68 _sessionFactories = sessionFactories; 69 } 70 71 private static Map<String, SessionFactory> _sessionFactories; 72 73 private static ThreadLocal<SessionFactory> _sessionFactoryThreadLocal = 74 new ThreadLocal<SessionFactory>() { 75 76 protected SessionFactory initialValue() { 77 return _sessionFactories.get(PropsValues.SHARD_DEFAULT_NAME); 78 } 79 80 }; 81 82 }