1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.data.pipeline.core.metadata;
19
20 import com.google.common.base.Strings;
21 import org.apache.shardingsphere.data.pipeline.core.context.PipelineContextKey;
22 import org.apache.shardingsphere.data.pipeline.core.job.api.PipelineAPIFactory;
23 import org.apache.shardingsphere.infra.datasource.pool.props.domain.DataSourcePoolProperties;
24 import org.apache.shardingsphere.infra.util.yaml.YamlEngine;
25 import org.apache.shardingsphere.infra.yaml.config.swapper.resource.YamlDataSourceConfigurationSwapper;
26
27 import java.util.Collections;
28 import java.util.LinkedHashMap;
29 import java.util.Map;
30 import java.util.Map.Entry;
31
32
33
34
35 public final class PipelineDataSourcePersistService implements PipelineMetaDataPersistService<Map<String, DataSourcePoolProperties>> {
36
37 private final YamlDataSourceConfigurationSwapper swapper = new YamlDataSourceConfigurationSwapper();
38
39 @Override
40 @SuppressWarnings("unchecked")
41 public Map<String, DataSourcePoolProperties> load(final PipelineContextKey contextKey, final String jobType) {
42 String dataSourcesProps = PipelineAPIFactory.getPipelineGovernanceFacade(contextKey).getMetaDataFacade().getDataSource().load(jobType);
43 if (Strings.isNullOrEmpty(dataSourcesProps)) {
44 return Collections.emptyMap();
45 }
46 Map<String, Map<String, Object>> yamlDataSources = YamlEngine.unmarshal(dataSourcesProps, Map.class);
47 Map<String, DataSourcePoolProperties> result = new LinkedHashMap<>(yamlDataSources.size(), 1F);
48 yamlDataSources.forEach((key, value) -> result.put(key, swapper.swapToDataSourcePoolProperties(value)));
49 return result;
50 }
51
52 @Override
53 public void persist(final PipelineContextKey contextKey, final String jobType, final Map<String, DataSourcePoolProperties> propsMap) {
54 Map<String, Map<String, Object>> dataSourceMap = new LinkedHashMap<>(propsMap.size(), 1F);
55 for (Entry<String, DataSourcePoolProperties> entry : propsMap.entrySet()) {
56 dataSourceMap.put(entry.getKey(), swapper.swapToMap(entry.getValue()));
57 }
58 PipelineAPIFactory.getPipelineGovernanceFacade(contextKey).getMetaDataFacade().getDataSource().persist(jobType, YamlEngine.marshal(dataSourceMap));
59 }
60 }