me.zhyd.hunter.config.HunterConfigTemplate Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of blog-hunter Show documentation

博客猎手，基于webMagic的博客爬取工具，支持慕课、csdn、iteye、cnblogs、掘金和V2EX等各大主流博客平台。博客千万篇，版权第一条。狩猎不规范，亲人两行泪。

The newest version!

package me.zhyd.hunter.config;

import cn.hutool.core.io.IoUtil;
import com.alibaba.fastjson.JSONObject;
import me.zhyd.hunter.consts.HunterConsts;
import me.zhyd.hunter.exception.HunterException;
import org.apache.commons.lang3.StringUtils;

import java.io.InputStream;
import java.nio.charset.Charset;

/**
 * @author yadong.zhang (yadong.zhang0415(a)gmail.com)
 * @version 1.0
 * @since 1.8
 */
public class HunterConfigTemplate {

    public static JSONObject configTemplate;

    static {
        HunterConfigTemplate configTemplate = new HunterConfigTemplate();
        configTemplate.init();
    }

    public static String getConfig(String platform) {
        if (configTemplate.containsKey(platform)) {
            return configTemplate.getString(platform);
        }
        throw new HunterException("暂不支持该平台[" + platform + "]");
    }

    private void init() {
        String configFileName = HunterConsts.CONFIG_FILE_NAME;
        String config = null;
        try {
            InputStream inputStream = this.getClass().getResourceAsStream(configFileName);
            if (null == inputStream) {
                throw new HunterException("请检查`src/main/resources`下是否存在" + configFileName);
            }
            config = IoUtil.read(inputStream, Charset.forName("UTF-8"));
            if (StringUtils.isEmpty(config)) {
                throw new HunterException("HunterConfig内容为空：" + configFileName);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }

        try {
            configTemplate = JSONObject.parseObject(config);
        } catch (Exception e) {
            throw new HunterException("HunterConfig配置文件格式错误");
        }

    }

}