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 }