org.noear.water.protocol.solution.LogSourceFactoryImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of water.protocol Show documentation
There is a newer version: 2.14.2
package org.noear.water.protocol.solution;

import org.noear.water.model.ConfigM;
import org.noear.water.protocol.*;
import org.noear.water.protocol.model.log.LoggerEntity;
import org.noear.water.protocol.model.log.LoggerMeta;
import org.noear.water.utils.TextUtils;
import org.noear.water.utils.ext.Fun1;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

public class LogSourceFactoryImpl implements LogSourceFactory {
    private static String _lock = "";

    private LoggerEntity _def;
    private Fun1 _loggerGetter;
    private Map _loggerMap = new HashMap<>();

    public LogSourceFactoryImpl(ConfigM def, Fun1 loggerGetter) {
        _def = new LoggerEntity(createLogSource(def), def);
        _loggerGetter = loggerGetter;
    }

    @Override
    public void updateSource(String logger) throws IOException {
        if (TextUtils.isEmpty(logger)) {
            return;
        }

        LoggerEntity entity = _loggerMap.get(logger);
        if (entity == null) {
            //如果没有，则不用更新
            getSource(logger);
            return;
        }

        LoggerMeta meta = getLoggerMeta(logger);

        if (meta == null) {
            //找不到元信息，不用更新
            return;
        }


        LogSource source;
        ConfigM cfg;

        if (TextUtils.isEmpty(meta.getSource())) {
            //如果是空配置，是否为默认？
            if (entity.source == _def.source) {
                return;//说明没变
            }

            cfg = _def.sourceConfig;
            source = _def.source;
        } else {
            cfg = ProtocolHub.config.getByTagKey(meta.getSource());

            if (entity.sourceConfig.value.equals(cfg.value)) {
                return;//说明没变
            }

            source = createLogSource(cfg);
        }

        if (source != null) {
            LogSource oldSource = entity.source;

            entity.source = source;
            entity.sourceConfig = cfg;

            //如果与默认的是同一个源，则不关闭
            if (oldSource != _def.source) {
                oldSource.close();
            }
        }
    }


    @Override
    public LogSource getSource(String logger) {
        if (TextUtils.isEmpty(logger)) {
            return _def.source;
        }

        LoggerEntity entity = _loggerMap.get(logger);

        if (entity == null) {
            synchronized (_lock) {
                entity = _loggerMap.get(logger);

                if (entity == null) {
                    LoggerMeta model = getLoggerMeta(logger);

                    if (model != null && TextUtils.isEmpty(model.getSource()) == false) {
                        ConfigM cfg = ProtocolHub.config.getByTagKey(model.getSource());
                        entity = new LoggerEntity(createLogSource(cfg), cfg);
                    }
                }

                if (entity == null) {
                    entity = new LoggerEntity(_def.source, _def.sourceConfig);
                }

                _loggerMap.put(logger, entity);
            }
        }

        return entity.source;
    }

    @Override
    public LoggerMeta getLoggerMeta(String logger) {
        return _loggerGetter.run(logger);
    }


    public static LogSource createLogSource(ConfigM cfg) {
        if (cfg == null || TextUtils.isEmpty(cfg.value)) {
            return null;
        }

        if (cfg.value.contains("=mongodb")) {
            String schema = cfg.getProp().getProperty("schema");
            if (TextUtils.isEmpty(schema)) {
                schema = "water_log_store";
            }

            return new LogSourceMongo(cfg.getMg(schema));
        } else if (cfg.value.contains("=elasticsearch")) {
            return new LogSourceElasticsearch(cfg.getEs());
        } else {
            return new LogSourceRdb(cfg.getDb(true));
        }
    }
}