001
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
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 }