com.litongjava.tio.utils.environment.Prop Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tio-utils Show documentation
t-io is a aio framework for java
There is a newer version: 3.7.3.v202400213-RELEASE
package com.litongjava.tio.utils.environment;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Properties;

import lombok.extern.slf4j.Slf4j;

/**
 * Prop. Prop can load properties file from CLASSPATH or File object.
 */
@Slf4j
public class Prop {

  public static final Charset DEFAULT_ENCODING = StandardCharsets.UTF_8;

  protected Properties properties;

  /**
   * 支持 new Prop().appendIfExists(...)
   */
  public Prop() {
    properties = new Properties();
  }

  /**
   * Prop constructor.
   * @see #Prop(String, String)
   */
  public Prop(String fileName) {
    this(fileName, DEFAULT_ENCODING);
  }

  /**
   * Prop constructor
   * 
   * Example:

   * Prop prop = new Prop("my_config.txt", "UTF-8");

   * String userName = prop.get("userName");


   * 
   * prop = new Prop("com/jfinal/file_in_sub_path_of_classpath.txt", "UTF-8");

   * String value = prop.get("key");
   * 
   * @param fileName the properties file's name in classpath or the sub directory of classpath
   * @param encoding the encoding
   */
  public Prop(String fileName, Charset encoding) {
    encoding.toString();
    InputStream inputStream = null;
    properties = new Properties();
    try {
      inputStream = getClassLoader().getResourceAsStream(fileName); // properties.load(Prop.class.getResourceAsStream(fileName));
      if (inputStream != null) {
        properties.load(new InputStreamReader(inputStream, encoding));
      } else {
        File file = new File(fileName);
        if (!file.exists()) {
          if (file.createNewFile()) {
            log.info("file created successful：" + fileName);
            inputStream = new FileInputStream(file);
            properties.load(new InputStreamReader(inputStream, encoding));
          } else {
            log.info("failed to create file" + fileName);
          }
        } else {
          log.info("load file：" + fileName);
          inputStream = new FileInputStream(file);
          properties.load(new InputStreamReader(inputStream, encoding));
        }
      }
    } catch (IOException e) {
      throw new RuntimeException("Error loading properties file.", e);
    } finally {
      if (inputStream != null)
        try {
          inputStream.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
    }
  }

  private ClassLoader getClassLoader() {
    ClassLoader ret = Thread.currentThread().getContextClassLoader();
    return ret != null ? ret : getClass().getClassLoader();
  }

  /**
   * Prop constructor.
   * @see #Prop(File, String)
   */
  public Prop(File file) {
    this(file, DEFAULT_ENCODING);
  }

  /**
   * Prop constructor
   * 
   * Example:

   * Prop prop = new Prop(new File("/var/config/my_config.txt"), "UTF-8");

   * String userName = prop.get("userName");
   * 
   * @param file the properties File object
   * @param encoding the encoding
   */
  public Prop(File file, Charset encoding) {
    if (file == null) {
      throw new IllegalArgumentException("File can not be null.");
    }
    if (!file.isFile()) {
      throw new IllegalArgumentException("File not found : " + file.getName());
    }

    InputStream inputStream = null;
    try {
      inputStream = new FileInputStream(file);
      properties = new Properties();
      properties.load(new InputStreamReader(inputStream, encoding));
    } catch (IOException e) {
      throw new RuntimeException("Error loading properties file.", e);
    } finally {
      if (inputStream != null)
        try {
          inputStream.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
    }
  }

  public Prop append(Prop prop) {
    if (prop == null) {
      throw new IllegalArgumentException("prop can not be null");
    }
    properties.putAll(prop.getProperties());
    return this;
  }

  public Prop append(String fileName, Charset encoding) {
    return append(new Prop(fileName, encoding));
  }

  public Prop append(String fileName) {
    return append(fileName, DEFAULT_ENCODING);
  }

  public Prop appendIfExists(String fileName, Charset encoding) {
    try {
      return append(new Prop(fileName, encoding));
    } catch (Exception e) {
      return this;
    }
  }

  public Prop appendIfExists(String fileName) {
    return appendIfExists(fileName, DEFAULT_ENCODING);
  }

  public Prop append(File file, Charset encoding) {
    return append(new Prop(file, encoding));
  }

  public Prop append(File file) {
    return append(file, DEFAULT_ENCODING);
  }

  public Prop appendIfExists(File file, Charset encoding) {
    if (file.isFile()) {
      append(new Prop(file, encoding));
    }
    return this;
  }

  public Prop appendIfExists(File file) {
    return appendIfExists(file, DEFAULT_ENCODING);
  }

  public String get(String key) {
    // 下面这行代码只要 key 存在，就不会返回 null。未给定 value 或者给定一个或多个空格都将返回 ""
    String value = properties.getProperty(key);
    return value != null && value.length() != 0 ? value.trim() : null;
  }

  public String get(String key, String defaultValue) {
    String value = properties.getProperty(key);
    return value != null && value.length() != 0 ? value.trim() : defaultValue;
  }

  public Integer getInt(String key) {
    return getInt(key, null);
  }

  public Integer getInt(String key, Integer defaultValue) {
    String value = properties.getProperty(key);
    if (value != null) {
      return Integer.parseInt(value.trim());
    }
    return defaultValue;
  }

  public Long getLong(String key) {
    return getLong(key, null);
  }

  public Long getLong(String key, Long defaultValue) {
    String value = properties.getProperty(key);
    if (value != null) {
      return Long.parseLong(value.trim());
    }
    return defaultValue;
  }

  public Double getDouble(String key) {
    return getDouble(key, null);
  }

  public Double getDouble(String key, Double defaultValue) {
    String value = properties.getProperty(key);
    if (value != null) {
      return Double.parseDouble(value.trim());
    }
    return defaultValue;
  }

  public Boolean getBoolean(String key) {
    return getBoolean(key, null);
  }

  public Boolean getBoolean(String key, Boolean defaultValue) {
    String value = properties.getProperty(key);
    if (value != null) {
      value = value.toLowerCase().trim();
      if ("true".equals(value)) {
        return true;
      } else if ("false".equals(value)) {
        return false;
      }
      throw new RuntimeException("The value can not parse to Boolean : " + value);
    }
    return defaultValue;
  }

  public boolean containsKey(String key) {
    return properties.containsKey(key);
  }

  public boolean isEmpty() {
    return properties.isEmpty();
  }

  public boolean notEmpty() {
    return !properties.isEmpty();
  }

  public Properties getProperties() {
    return properties;
  }
}