| ShardDataSourceTargetSource.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 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 public class ShardDataSourceTargetSource implements TargetSource {
39
40 public DataSource getDataSource() {
41 return _dataSourceThreadLocal.get();
42 }
43
44 public Object getTarget() throws Exception {
45 return getDataSource();
46 }
47
48 public Class<DataSource> getTargetClass() {
49 return DataSource.class;
50 }
51
52 public boolean isStatic() {
53 return false;
54 }
55
56 public void releaseTarget(Object target) throws Exception {
57 }
58
59 public void setDataSource(String shardName) {
60 _dataSourceThreadLocal.set(_dataSources.get(shardName));
61 }
62
63 public void setDataSources(Map<String, DataSource> dataSources) {
64 _dataSources = dataSources;
65 }
66
67 private static Map<String, DataSource> _dataSources;
68
69 private static ThreadLocal<DataSource> _dataSourceThreadLocal =
70 new ThreadLocal<DataSource>() {
71
72 protected DataSource initialValue() {
73 return _dataSources.get(PropsValues.SHARD_DEFAULT_NAME);
74 }
75
76 };
77
78 }