Browse Source
* replace DataSourceController API with string JSON and cast to DTO * add DataSourceProcessorManager and DataSourceProcessorProvider to datasource processor functions3.1.0-release
Tq
3 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