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 }