Bug 9056 - Class FileModuleShardConfigProvider does not load
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / config / FileModuleShardConfigProvider.java
1 /*
2  * Copyright (c) 2015 Brocade Communications Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.controller.cluster.datastore.config;
9
10 import com.typesafe.config.Config;
11 import com.typesafe.config.ConfigFactory;
12 import com.typesafe.config.ConfigObject;
13 import java.io.File;
14 import java.util.HashMap;
15 import java.util.List;
16 import java.util.Map;
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;
22
23 /**
24  * Implementation of ModuleShardConfigProvider that reads the module and shard configuration from files.
25  *
26  * @author Thomas Pantelis
27  */
28 public class FileModuleShardConfigProvider implements ModuleShardConfigProvider {
29     private static final Logger LOG = LoggerFactory.getLogger(FileModuleShardConfigProvider.class);
30
31     private final String moduleShardsConfigPath;
32     private final String modulesConfigPath;
33
34     public FileModuleShardConfigProvider(final String moduleShardsConfigPath, final String modulesConfigPath) {
35         this.moduleShardsConfigPath = moduleShardsConfigPath;
36         this.modulesConfigPath = modulesConfigPath;
37     }
38
39     @Override
40     public Map<String, ModuleConfig.Builder> retrieveModuleConfigs(final Configuration configuration) {
41         final File moduleShardsFile = new File(moduleShardsConfigPath);
42         final File modulesFile = new File(modulesConfigPath);
43
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);
48         } else {
49             LOG.warn("module shards configuration read from resource");
50             moduleShardsConfig = ConfigFactory.load(moduleShardsFile.getPath());
51         }
52
53         Config modulesConfig = null;
54         if (modulesFile.exists()) {
55             LOG.info("modules config file exists - reading config from it");
56             modulesConfig = ConfigFactory.parseFile(modulesFile);
57         } else {
58             LOG.warn("modules configuration read from resource");
59             modulesConfig = ConfigFactory.load(modulesFile.getPath());
60         }
61
62         final Map<String, ModuleConfig.Builder> moduleConfigMap = readModuleShardsConfig(moduleShardsConfig);
63         readModulesConfig(modulesConfig, moduleConfigMap, configuration);
64
65         return moduleConfigMap;
66     }
67
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");
71
72         for (final ConfigObject o : modulesConfigObjectList) {
73             final ConfigObjectWrapper wrapper = new ConfigObjectWrapper(o);
74
75             final String moduleName = wrapper.stringValue("name");
76             final ModuleConfig.Builder builder = moduleConfigMap.computeIfAbsent(moduleName, ModuleConfig::builder);
77
78             builder.nameSpace(wrapper.stringValue("namespace"));
79             builder.shardStrategy(ShardStrategyFactory.newShardStrategyInstance(moduleName,
80                     wrapper.stringValue("shard-strategy"), configuration));
81         }
82     }
83
84     private static Map<String, ModuleConfig.Builder> readModuleShardsConfig(final Config moduleShardsConfig) {
85         final List<? extends ConfigObject> moduleShardsConfigObjectList =
86             moduleShardsConfig.getObjectList("module-shards");
87
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);
92
93             final List<? extends ConfigObject> shardsConfigObjectList =
94                 moduleShardConfigObject.toConfig().getObjectList("shards");
95
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);
101             }
102
103             moduleConfigMap.put(moduleName, builder);
104         }
105
106         return moduleConfigMap;
107     }
108
109     private static class ConfigObjectWrapper {
110
111         private final ConfigObject configObject;
112
113         ConfigObjectWrapper(final ConfigObject configObject) {
114             this.configObject = configObject;
115         }
116
117         public String stringValue(final String name) {
118             return configObject.get(name).unwrapped().toString();
119         }
120     }
121 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.