001    /**
002     * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
003     *
004     * The contents of this file are subject to the terms of the Liferay Enterprise
005     * Subscription License ("License"). You may not use this file except in
006     * compliance with the License. You can obtain a copy of the License by
007     * contacting Liferay, Inc. See the License for the specific language governing
008     * permissions and limitations under the License, including but not limited to
009     * distribution rights of the Software.
010     *
011     *
012     *
013     */
014    
015    package com.liferay.portal.dao.shard;
016    
017    import com.liferay.portal.dao.jdbc.spring.MappingSqlQueryImpl;
018    import com.liferay.portal.kernel.dao.jdbc.RowMapper;
019    import com.liferay.portal.kernel.dao.shard.ShardUtil;
020    
021    import java.util.List;
022    
023    import javax.sql.DataSource;
024    
025    /**
026     * @author Alexander Chow
027     */
028    public class ShardMappingSqlQueryImpl<T> extends MappingSqlQueryImpl<T> {
029    
030            public ShardMappingSqlQueryImpl(
031                    DataSource dataSource, String sql, int[] types,
032                    RowMapper<T> rowMapper) {
033    
034                    super(dataSource, sql, types, rowMapper);
035            }
036    
037            public List<T> execute(Object... params) {
038                    setDataSource(ShardUtil.getDataSource());
039    
040                    return super.execute(params);
041            }
042    
043    }