com.gracerun.summermq.factory.MQClientInstance Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of summer-mq Show documentation
There is a newer version: 1.0.6

package com.gracerun.summermq.factory;

import com.gracerun.summermq.constant.ServiceState;
import com.gracerun.summermq.consumer.PullMessageService;
import com.gracerun.summermq.consumer.RedisMessagePullConsumer;
import com.gracerun.summermq.producer.RedisMessageProducer;
import lombok.Getter;
import lombok.extern.slf4j.Slf4j;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

@Slf4j
public class MQClientInstance {
    private static MQClientInstance instance = new MQClientInstance();
    private final ConcurrentMap producerTable = new ConcurrentHashMap<>();
    private final ConcurrentMap consumerTable = new ConcurrentHashMap<>();

    @Getter
    private final PullMessageService pullMessageService;

    private ServiceState serviceState = ServiceState.CREATE_JUST;

    private MQClientInstance() {
        this.pullMessageService = new PullMessageService(this);
    }

    public static MQClientInstance getInstance() {
        return instance;
    }

    public void start() {
        synchronized (this) {
            switch (this.serviceState) {
                case CREATE_JUST:
                    this.serviceState = ServiceState.START_FAILED;
                    this.pullMessageService.start();
                    this.serviceState = ServiceState.RUNNING;
                    break;
                case RUNNING:
                    break;
                case SHUTDOWN_ALREADY:
                    break;
                default:
                    break;
            }
        }
    }

    public boolean registerConsumer(final String namespace, final RedisMessagePullConsumer consumer) {
        if (null == namespace || null == consumer) {
            return false;
        }

        RedisMessagePullConsumer prev = this.consumerTable.putIfAbsent(namespace, consumer);
        if (prev != null) {
            log.warn("the consumer namespace[{}] exist already.", namespace);
            return false;
        }

        return true;
    }

    public boolean registerProducer(final String namespace, final RedisMessageProducer producer) {
        if (null == namespace || null == producer) {
            return false;
        }

        RedisMessageProducer prev = this.producerTable.putIfAbsent(namespace, producer);
        if (prev != null) {
            log.warn("the producer namespace[{}] exist already.", namespace);
            return false;
        }

        return true;
    }

    public RedisMessageProducer selectProducer(final String namespace) {
        return this.producerTable.get(namespace);
    }

    public RedisMessagePullConsumer selectConsumer(final String namespace) {
        return this.consumerTable.get(namespace);
    }
}