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  }