001    /**
002     * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.upgrade.dao.orm;
016    
017    import com.liferay.portal.kernel.dao.db.DBType;
018    import com.liferay.portal.kernel.upgrade.dao.orm.UpgradeOptimizedConnectionProvider;
019    import com.liferay.portal.kernel.upgrade.dao.orm.UpgradeOptimizedConnectionProviderRegistry;
020    
021    import java.util.HashMap;
022    import java.util.Map;
023    import java.util.ServiceLoader;
024    
025    /**
026     * @author Cristina González
027     */
028    public class UpgradeOptimizedConnectionProviderRegistryImpl
029            implements UpgradeOptimizedConnectionProviderRegistry {
030    
031            public UpgradeOptimizedConnectionProviderRegistryImpl() {
032                    ServiceLoader<UpgradeOptimizedConnectionProvider> serviceLoader =
033                            ServiceLoader.load(
034                                    UpgradeOptimizedConnectionProvider.class,
035                                    UpgradeOptimizedConnectionProviderRegistryImpl.class.
036                                            getClassLoader());
037    
038                    for (UpgradeOptimizedConnectionProvider
039                                    upgradeOptimizedConnectionProvider : serviceLoader) {
040    
041                            _upgradeOptimizedConnectionProviders.put(
042                                    upgradeOptimizedConnectionProvider.getDBType(),
043                                    upgradeOptimizedConnectionProvider);
044                    }
045            }
046    
047            @Override
048            public UpgradeOptimizedConnectionProvider
049                    getUpgradeOptimizedConnectionProvider(DBType dbType) {
050    
051                    return _upgradeOptimizedConnectionProviders.get(dbType);
052            }
053    
054            private final Map<DBType, UpgradeOptimizedConnectionProvider>
055                    _upgradeOptimizedConnectionProviders = new HashMap<>();
056    
057    }