Refactored config file resource loading to make it easier to load config files from class path when embedding.

2016-12-31 11:46:32 +01:00
parent c36ebc648b
commit ac93f04dde
12 changed files with 373 additions and 167 deletions
@@ -0,0 +1,76 @@
+/*
+ * Copyright (c) 2012-2015 The original author or authors
+ * ------------------------------------------------------
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * and Apache License v2.0 which accompanies this distribution.
+ *
+ * The Eclipse Public License is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * The Apache License v2.0 is available at
+ * http://www.opensource.org/licenses/apache2.0.php
+ *
+ * You may elect to redistribute this code under either of these licenses.
+ */
+package io.moquette.server.config;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.Reader;
+import java.text.ParseException;
+import java.util.Properties;
+
+/**
+ * Configuration that loads config stream from a {@link IResourceLoader} instance.
+ */
+public class ResourceLoaderConfig extends IConfig {
+
+    private static final Logger LOG = LoggerFactory.getLogger(ResourceLoaderConfig.class);
+
+    private final Properties m_properties;
+    private final IResourceLoader resourceLoader;
+
+    public ResourceLoaderConfig(IResourceLoader resourceLoader) {
+        this(resourceLoader, null);
+    }
+
+    public ResourceLoaderConfig(IResourceLoader resourceLoader, String configName) {
+        this.resourceLoader = resourceLoader;
+        Reader configReader = configName != null ? resourceLoader.loadResource(configName) : resourceLoader.loadDefaultResource();
+        if (configReader == null) {
+            throw new IllegalArgumentException(
+              "Can't locate " + resourceLoader.getName() + " \"" + configName + "\"");
+        }
+        ConfigurationParser confParser = new ConfigurationParser();
+        m_properties = confParser.getProperties();
+        assignDefaults();
+        try {
+            confParser.parse(configReader);
+        } catch (ParseException pex) {
+            LOG.warn("An error occurred in parsing configuration, fallback on default configuration", pex);
+        }
+    }
+
+    @Override
+    public void setProperty(String name, String value) {
+        m_properties.setProperty(name, value);
+    }
+
+    @Override
+    public String getProperty(String name) {
+        return m_properties.getProperty(name);
+    }
+
+    @Override
+    public String getProperty(String name, String defaultValue) {
+        return m_properties.getProperty(name, defaultValue);
+    }
+
+    @Override
+    public IResourceLoader getResourceLoader() {
+        return resourceLoader;
+    }
+
+}