2 * Copyright (c) 2015 Brocade Communications Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.cluster.datastore.config;
10 import com.typesafe.config.Config;
11 import com.typesafe.config.ConfigFactory;
12 import com.typesafe.config.ConfigObject;
14 import java.util.HashMap;
15 import java.util.List;
17 import java.util.stream.Collectors;
18 import org.opendaylight.controller.cluster.access.concepts.MemberName;
19 import org.opendaylight.controller.cluster.datastore.shardstrategy.ShardStrategyFactory;
20 import org.slf4j.Logger;
21 import org.slf4j.LoggerFactory;
24 * Implementation of ModuleShardConfigProvider that reads the module and shard configuration from files.
26 * @author Thomas Pantelis
28 public class FileModuleShardConfigProvider implements ModuleShardConfigProvider {
29 private static final Logger LOG = LoggerFactory.getLogger(FileModuleShardConfigProvider.class);
31 private final String moduleShardsConfigPath;
32 private final String modulesConfigPath;
34 public FileModuleShardConfigProvider(final String moduleShardsConfigPath, final String modulesConfigPath) {
35 this.moduleShardsConfigPath = moduleShardsConfigPath;
36 this.modulesConfigPath = modulesConfigPath;
40 public Map<String, ModuleConfig.Builder> retrieveModuleConfigs(final Configuration configuration) {
41 final File moduleShardsFile = new File(moduleShardsConfigPath);
42 final File modulesFile = new File(modulesConfigPath);
44 Config moduleShardsConfig = null;
45 if (moduleShardsFile.exists()) {
46 LOG.info("module shards config file exists - reading config from it");
47 moduleShardsConfig = ConfigFactory.parseFile(moduleShardsFile);
49 LOG.warn("module shards configuration read from resource");
50 moduleShardsConfig = ConfigFactory.load(moduleShardsFile.getName());
53 Config modulesConfig = null;
54 if (modulesFile.exists()) {
55 LOG.info("modules config file exists - reading config from it");
56 modulesConfig = ConfigFactory.parseFile(modulesFile);
58 LOG.warn("modules configuration read from resource");
59 modulesConfig = ConfigFactory.load(modulesFile.getName());
62 final Map<String, ModuleConfig.Builder> moduleConfigMap = readModuleShardsConfig(moduleShardsConfig);
63 readModulesConfig(modulesConfig, moduleConfigMap, configuration);
65 return moduleConfigMap;
68 private static void readModulesConfig(final Config modulesConfig,
69 final Map<String, ModuleConfig.Builder> moduleConfigMap, final Configuration configuration) {
70 final List<? extends ConfigObject> modulesConfigObjectList = modulesConfig.getObjectList("modules");
72 for (final ConfigObject o : modulesConfigObjectList) {
73 final ConfigObjectWrapper wrapper = new ConfigObjectWrapper(o);
75 final String moduleName = wrapper.stringValue("name");
76 final ModuleConfig.Builder builder = moduleConfigMap.computeIfAbsent(moduleName, ModuleConfig::builder);
78 builder.nameSpace(wrapper.stringValue("namespace"));
79 builder.shardStrategy(ShardStrategyFactory.newShardStrategyInstance(moduleName,
80 wrapper.stringValue("shard-strategy"), configuration));
84 private static Map<String, ModuleConfig.Builder> readModuleShardsConfig(final Config moduleShardsConfig) {
85 final List<? extends ConfigObject> moduleShardsConfigObjectList =
86 moduleShardsConfig.getObjectList("module-shards");
88 final Map<String, ModuleConfig.Builder> moduleConfigMap = new HashMap<>();
89 for (final ConfigObject moduleShardConfigObject : moduleShardsConfigObjectList) {
90 final String moduleName = moduleShardConfigObject.get("name").unwrapped().toString();
91 final ModuleConfig.Builder builder = ModuleConfig.builder(moduleName);
93 final List<? extends ConfigObject> shardsConfigObjectList =
94 moduleShardConfigObject.toConfig().getObjectList("shards");
96 for (final ConfigObject shard : shardsConfigObjectList) {
97 final String shardName = shard.get("name").unwrapped().toString();
98 final List<MemberName> replicas = shard.toConfig().getStringList("replicas").stream()
99 .map(MemberName::forName).collect(Collectors.toList());
100 builder.shardConfig(shardName, replicas);
103 moduleConfigMap.put(moduleName, builder);
106 return moduleConfigMap;
109 private static class ConfigObjectWrapper {
111 private final ConfigObject configObject;
113 ConfigObjectWrapper(final ConfigObject configObject) {
114 this.configObject = configObject;
117 public String stringValue(final String name) {
118 return configObject.get(name).unwrapped().toString();