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