View Javadoc

    1   /*
    2    * Licensed to the Apache Software Foundation (ASF) under one or more
    3    * contributor license agreements. See the NOTICE file distributed with
    4    * this work for additional information regarding copyright ownership.
    5    * The ASF licenses this file to You under the Apache license, Version 2.0
    6    * (the "License"); you may not use this file except in compliance with
    7    * the License. You may obtain a copy of the License at
    8    *
    9    *      http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the license for the specific language governing permissions and
    15   * limitations under the license.
    16   */
    17  package org.apache.logging.log4j.core.config.yaml;
    18
    19  import java.io.IOException;
    20
    21  import org.apache.logging.log4j.core.config.Configuration;
    22  import org.apache.logging.log4j.core.config.ConfigurationSource;
    23  import org.apache.logging.log4j.core.config.json.JsonConfiguration;
    24
    25  import com.fasterxml.jackson.core.JsonParser;
    26  import com.fasterxml.jackson.databind.ObjectMapper;
    27  import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
    28
    29  public class YamlConfiguration extends JsonConfiguration {
    30
    31      private static final long serialVersionUID = 1L;
    32
    33      public YamlConfiguration(final ConfigurationSource configSource) {
    34          super(configSource);
    35      }
    36
    37      @Override
    38      protected ObjectMapper getObjectMapper() {
    39          return new ObjectMapper(new YAMLFactory()).configure(JsonParser.Feature.ALLOW_COMMENTS, true);
    40      }
    41
    42      @Override
    43      public Configuration reconfigure() {
    44          try {
    45              final ConfigurationSource source = getConfigurationSource().resetInputStream();
    46              if (source == null) {
    47                  return null;
    48              }
    49              return new YamlConfiguration(source);
    50          } catch (final IOException ex) {
    51              LOGGER.error("Cannot locate file {}", getConfigurationSource(), ex);
    52          }
    53          return null;
    54      }
    55  }