| ShardDataSourceTargetSource.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 javax.sql.DataSource;
30
31 import org.springframework.aop.TargetSource;
32
33 /**
34 * <a href="ShardDataSourceTargetSource.java.html"><b><i>View Source</i></b></a>
35 *
36 * @author Michael Young
37 *
38 */
39 public class ShardDataSourceTargetSource implements TargetSource {
40
41 public DataSource getDataSource() {
42 return _dataSourceThreadLocal.get();
43 }
44
45 public Object getTarget() throws Exception {
46 return getDataSource();
47 }
48
49 public Class<DataSource> getTargetClass() {
50 return DataSource.class;
51 }
52
53 public boolean isStatic() {
54 return false;
55 }
56
57 public void releaseTarget(Object target) throws Exception {
58 }
59
60 public void setDataSource(String shardName) {
61 _dataSourceThreadLocal.set(_dataSources.get(shardName));
62 }
63
64 public void setDataSources(Map<String, DataSource> dataSources) {
65 _dataSources = dataSources;
66 }
67
68 private static Map<String, DataSource> _dataSources;
69
70 private static ThreadLocal<DataSource> _dataSourceThreadLocal =
71 new ThreadLocal<DataSource>() {
72
73 protected DataSource initialValue() {
74 return _dataSources.get(PropsValues.SHARD_DEFAULT_NAME);
75 }
76
77 };
78
79 }