org.noear.water.protocol.solution.MsgBrokerImpl 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.redisx.RedisClient;
import org.noear.water.WW;
import org.noear.water.model.ConfigM;
import org.noear.water.model.PropertiesM;
import org.noear.water.protocol.MsgBroker;
import org.noear.water.protocol.MsgQueue;
import org.noear.water.protocol.MsgSource;
import org.noear.water.utils.DsUtils;
import org.noear.water.utils.TextUtils;
import org.noear.wood.cache.ICacheServiceEx;
import org.noear.mongox.MgContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.util.Properties;

/**
 * @author noear 2021/11/1 created
 */
public class MsgBrokerImpl implements MsgBroker {
    private final String name;

    private MsgSource source;
    private MsgQueue queue;
    private Logger log = LoggerFactory.getLogger(WW.logger_water_log_msg);

    /**
     * 新的配置
     */
    public MsgBrokerImpl(String name, ConfigM cfg, ICacheServiceEx cache) {
        if (name == null) {
            this.name = "";
        } else {
            this.name = name;
        }

        PropertiesM prop = cfg.getProp();
        PropertiesM storeProp = prop.getProp("store");
        PropertiesM queueProp = prop.getProp("queue");

        if (storeProp.size() < 3) {
            init(cfg, prop, null, cache);
        } else {
            init(cfg, storeProp, queueProp, cache);
        }
    }

    private void init(ConfigM cfg, Properties sourceProp, Properties queueProp, ICacheServiceEx cache) {
        //for source
        if (cfg.value.contains("=mongodb")) {
            String schema = sourceProp.getProperty("schema");
            if (TextUtils.isEmpty(schema)) {
                schema = "water_msg_store";
            }
            source = new MsgSourceMongo(new MgContext(sourceProp, schema), cache, log);
        } else {
            source = new MsgSourceRdb(DsUtils.getDb(sourceProp, true), cache, log);
        }

        if (cfg.value.contains("=redis") && queueProp != null) {
            String queueName = queueProp.getProperty("name");

            if (TextUtils.isEmpty(queueName)) {
                throw new RuntimeException("MsgBroker::Missing name configuration");
            }

            queue = new MsgQueueRedis(queueName, new RedisClient(queueProp));
        } else {
            queue = MsgQueueLocal.getInstance();
        }
    }

    @Override
    public String getName() {
        return name;
    }

    @Override
    public MsgQueue getQueue() {
        return queue;
    }

    @Override
    public MsgSource getSource() {
        return source;
    }

    @Override
    public void close() throws IOException {
        try {
            source.close();
        } catch (Throwable e) {
            e.printStackTrace();
        }

        try {
            queue.close();
        } catch (Throwable e) {
            e.printStackTrace();
        }
    }
}