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.getName());
  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.getName());
  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 }