Browse Source
* replace DataSourceController API with string JSON and cast to DTO * add DataSourceProcessorManager and DataSourceProcessorProvider to datasource processor functions3.1.0-release
Tq
2 years ago
committed by
GitHub
18 changed files with 347 additions and 132 deletions
@ -0,0 +1,63 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.api.plugin; |
||||
|
||||
import org.apache.dolphinscheduler.plugin.datasource.api.datasource.DataSourceProcessor; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
|
||||
import java.util.Collections; |
||||
import java.util.HashSet; |
||||
import java.util.Map; |
||||
import java.util.ServiceLoader; |
||||
import java.util.Set; |
||||
import java.util.concurrent.ConcurrentHashMap; |
||||
|
||||
import static java.lang.String.format; |
||||
|
||||
public class DataSourceProcessorManager { |
||||
private static final Logger logger = LoggerFactory.getLogger(DataSourceProcessorManager.class); |
||||
|
||||
private static final Map<String, DataSourceProcessor> dataSourceProcessorMap = new ConcurrentHashMap<>(); |
||||
|
||||
public Map<String, DataSourceProcessor> getDataSourceProcessorMap() { |
||||
return Collections.unmodifiableMap(dataSourceProcessorMap); |
||||
} |
||||
|
||||
public void installProcessor() { |
||||
final Set<String> names = new HashSet<>(); |
||||
|
||||
ServiceLoader.load(DataSourceProcessor.class).forEach(factory -> { |
||||
final String name = factory.getDbType().name(); |
||||
|
||||
logger.info("start register processor: {}", name); |
||||
if (!names.add(name)) { |
||||
throw new IllegalStateException(format("Duplicate datasource plugins named '%s'", name)); |
||||
} |
||||
loadDatasourceClient(factory); |
||||
|
||||
logger.info("done register processor: {}", name); |
||||
|
||||
}); |
||||
} |
||||
|
||||
private void loadDatasourceClient(DataSourceProcessor processor) { |
||||
DataSourceProcessor instance = processor.create(); |
||||
dataSourceProcessorMap.put(processor.getDbType().name(), instance); |
||||
} |
||||
} |
||||
@ -0,0 +1,52 @@
|
||||
/* |
||||
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||
* contributor license agreements. See the NOTICE file distributed with |
||||
* this work for additional information regarding copyright ownership. |
||||
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||
* (the "License"); you may not use this file except in compliance with |
||||
* the License. You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.apache.dolphinscheduler.plugin.datasource.api.plugin; |
||||
|
||||
import org.apache.dolphinscheduler.plugin.datasource.api.datasource.DataSourceProcessor; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
|
||||
import java.util.Map; |
||||
|
||||
|
||||
public class DataSourceProcessorProvider { |
||||
private static final Logger logger = LoggerFactory.getLogger(DataSourceProcessorProvider.class); |
||||
|
||||
private DataSourceProcessorManager dataSourcePluginManager; |
||||
|
||||
private DataSourceProcessorProvider() { |
||||
initDataSourceProcessorPlugin(); |
||||
} |
||||
|
||||
private static class DataSourceClientProviderHolder { |
||||
private static final DataSourceProcessorProvider INSTANCE = new DataSourceProcessorProvider(); |
||||
} |
||||
|
||||
public static DataSourceProcessorProvider getInstance() { |
||||
return DataSourceClientProviderHolder.INSTANCE; |
||||
} |
||||
|
||||
public Map<String, DataSourceProcessor> getDataSourceProcessorMap() { |
||||
return dataSourcePluginManager.getDataSourceProcessorMap(); |
||||
} |
||||
|
||||
private void initDataSourceProcessorPlugin() { |
||||
dataSourcePluginManager = new DataSourceProcessorManager(); |
||||
dataSourcePluginManager.installProcessor(); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue