nodece commented on code in PR #19208:
URL: https://github.com/apache/pulsar/pull/19208#discussion_r1071870686
##########
pulsar-metadata/src/main/java/org/apache/pulsar/metadata/impl/MetadataStoreFactoryImpl.java:
##########
@@ -45,19 +63,39 @@ public static MetadataStoreExtended createExtended(String
metadataURL, MetadataS
private static MetadataStore newInstance(String metadataURL,
MetadataStoreConfig metadataStoreConfig,
boolean enableSessionWatcher)
throws MetadataStoreException {
+ MetadataStoreProvider provider = findProvider(metadataURL);
+ return provider.create(metadataURL, metadataStoreConfig,
enableSessionWatcher);
+ }
+
+ static void loadProviders() {
+ providers = new HashMap<>();
+ providers.put(MEMORY_SCHEME_IDENTIFIER, new
MemeoryMetadataStoreProvider());
+ providers.put(ROCKSDB_SCHEME_IDENTIFIER, new
RocksdbMetadataStoreProvider());
+ providers.put(ETCD_SCHEME_IDENTIFIER, new EtcdMetadataStoreProvider());
+ providers.put(ZK_SCHEME_IDENTIFIER, new ZkMetadataStoreProvider());
+
+ String factoryClasses =
System.getProperty(METADATASTORE_PROVIDERS_PROPERTY, "");
+
+ for (String className :
Splitter.on(',').trimResults().omitEmptyStrings().split(factoryClasses)) {
+ try {
+ Class<? extends MetadataStoreProvider> clazz =
+ (Class<? extends MetadataStoreProvider>)
Class.forName(className);
+ MetadataStoreProvider provider =
clazz.getConstructor().newInstance();
+ String scheme = provider.urlScheme();
+ providers.put(scheme + ":", provider);
+ } catch (Exception e) {
+ log.warn("Failed to load metadata store provider class for
name '{}'", className, e);
+ }
+ }
+ }
- if
(metadataURL.startsWith(LocalMemoryMetadataStore.MEMORY_SCHEME_IDENTIFIER)) {
- return new LocalMemoryMetadataStore(metadataURL,
metadataStoreConfig);
- } else if
(metadataURL.startsWith(RocksdbMetadataStore.ROCKSDB_SCHEME_IDENTIFIER)) {
- return RocksdbMetadataStore.get(metadataURL, metadataStoreConfig);
- } else if
(metadataURL.startsWith(EtcdMetadataStore.ETCD_SCHEME_IDENTIFIER)) {
- return new EtcdMetadataStore(metadataURL, metadataStoreConfig,
enableSessionWatcher);
- } else if
(metadataURL.startsWith(ZKMetadataStore.ZK_SCHEME_IDENTIFIER)) {
- return new
ZKMetadataStore(metadataURL.substring(ZKMetadataStore.ZK_SCHEME_IDENTIFIER.length()),
- metadataStoreConfig, enableSessionWatcher);
- } else {
- return new ZKMetadataStore(metadataURL, metadataStoreConfig,
enableSessionWatcher);
+ private static MetadataStoreProvider findProvider(String metadataURL) {
+ for (Map.Entry<String, MetadataStoreProvider> entry :
providers.entrySet()) {
+ if (metadataURL.startsWith(entry.getKey())) {
Review Comment:
```suggestion
if (metadataURL.equals(entry.getKey())) {
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
