wz2cool / local-queue / #32

Committed 01 Feb 2025 03:28AM UTC coverage: 90.877% (+0.8%) from 90.052%

Build # #32

Build Type

push

Committed by

Commit Message

multi close

Run Details

0 of 1 new or added line in 1 file covered. (0.0%)

518 of 570 relevant lines covered (90.88%)

0.91 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.88

/src/main/java/com/github/wz2cool/localqueue/impl/SimpleQueue.java

package com.github.wz2cool.localqueue.impl;

import com.github.wz2cool.localqueue.IConsumer;
import com.github.wz2cool.localqueue.IQueue;
import com.github.wz2cool.localqueue.event.CloseListener;
import com.github.wz2cool.localqueue.model.config.SimpleConsumerConfig;
import com.github.wz2cool.localqueue.model.config.SimpleProducerConfig;
import com.github.wz2cool.localqueue.model.config.SimpleQueueConfig;
import com.github.wz2cool.localqueue.model.enums.ConsumeFromWhere;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;

/**
 * simple queue
 *
 * @author frank
 */
public class SimpleQueue implements IQueue {

    private final Logger logger = LoggerFactory.getLogger(this.getClass());
    private final SimpleQueueConfig config;
    private final SimpleProducer simpleProducer;
    private final Map<String, SimpleConsumer> consumerMap = new ConcurrentHashMap<>();
    private final ConcurrentLinkedQueue<CloseListener> closeListeners = new ConcurrentLinkedQueue<>();
    private volatile boolean isClosed = false;

    public SimpleQueue(SimpleQueueConfig config) {
        this.config = config;
        this.simpleProducer = getProducer();
    }

    @Override
    public boolean offer(String message) {
        return simpleProducer.offer(message);
    }

    @Override
    public boolean offer(String messageKey, String message) {
        return simpleProducer.offer(messageKey, message);
    }

    private SimpleProducer getProducer() {
        return new SimpleProducer(new SimpleProducerConfig.Builder()
                .setDataDir(config.getDataDir())
                .setKeepDays(config.getKeepDays())
                .build());
    }

    @Override
    public synchronized IConsumer getConsumer(final String consumerId) {
        return getConsumer(consumerId, ConsumeFromWhere.LAST);
    }

    @Override
    public synchronized IConsumer getConsumer(final String consumerId, final ConsumeFromWhere consumeFromWhere) {
        SimpleConsumer consumer = consumerMap.get(consumerId);
        if (Objects.nonNull(consumer)) {
            return consumer;
        }

        consumer = new SimpleConsumer(new SimpleConsumerConfig.Builder()
                .setDataDir(config.getDataDir())
                .setConsumerId(consumerId)
                .setConsumeFromWhere(consumeFromWhere)
                .build());
        consumer.addCloseListener(() -> consumerMap.remove(consumerId));
        consumerMap.put(consumerId, consumer);
        return consumer;
    }

    @Override
    public void close() {
        try {
            logDebug("[close] start");
            if (isClosed) {
                logDebug("[close] already closed");
                return;
            }
            if (!simpleProducer.isClosed()) {
                simpleProducer.close();
            }
            for (Map.Entry<String, SimpleConsumer> entry : consumerMap.entrySet()) {
                SimpleConsumer consumer = entry.getValue();
                if (!consumer.isClosed()) {
                    entry.getValue().close();
                }
            }
            for (CloseListener listener : closeListeners) {
                listener.onClose();
            }
            isClosed = true;
        } finally {
            logDebug("[close] end");
        }
    }


    @Override
    public void addCloseListener(CloseListener listener) {
        closeListeners.add(listener);
    }

    // region logger

    private void logDebug(String format) {
        if (logger.isDebugEnabled()) {
            logger.debug(format);
        }
    }

    // endregion
}

1	package com.github.wz2cool.localqueue.impl;
2
3	import com.github.wz2cool.localqueue.IConsumer;
4	import com.github.wz2cool.localqueue.IQueue;
5	import com.github.wz2cool.localqueue.event.CloseListener;
6	import com.github.wz2cool.localqueue.model.config.SimpleConsumerConfig;
7	import com.github.wz2cool.localqueue.model.config.SimpleProducerConfig;
8	import com.github.wz2cool.localqueue.model.config.SimpleQueueConfig;
9	import com.github.wz2cool.localqueue.model.enums.ConsumeFromWhere;
10	import org.slf4j.Logger;
11	import org.slf4j.LoggerFactory;
12
13	import java.util.Map;
14	import java.util.Objects;
15	import java.util.concurrent.ConcurrentHashMap;
16	import java.util.concurrent.ConcurrentLinkedQueue;
17
18	/**
19	* simple queue
20	*
21	* @author frank
22	*/
23	public class SimpleQueue implements IQueue {
24
25	private final Logger logger = LoggerFactory.getLogger(this.getClass());	1✔
26	private final SimpleQueueConfig config;
27	private final SimpleProducer simpleProducer;
28	private final Map<String, SimpleConsumer> consumerMap = new ConcurrentHashMap<>();	1✔
29	private final ConcurrentLinkedQueue<CloseListener> closeListeners = new ConcurrentLinkedQueue<>();	1✔
30	private volatile boolean isClosed = false;	1✔
31
32	public SimpleQueue(SimpleQueueConfig config) {	1✔
33	this.config = config;	1✔
34	this.simpleProducer = getProducer();	1✔
35	}	1✔
36
37	@Override
38	public boolean offer(String message) {
39	return simpleProducer.offer(message);	1✔
40	}
41
42	@Override
43	public boolean offer(String messageKey, String message) {
44	return simpleProducer.offer(messageKey, message);	×
45	}
46
47	private SimpleProducer getProducer() {
48	return new SimpleProducer(new SimpleProducerConfig.Builder()	1✔
49	.setDataDir(config.getDataDir())	1✔
50	.setKeepDays(config.getKeepDays())	1✔
51	.build());	1✔
52	}
53
54	@Override
55	public synchronized IConsumer getConsumer(final String consumerId) {
56	return getConsumer(consumerId, ConsumeFromWhere.LAST);	1✔
57	}
58
59	@Override
60	public synchronized IConsumer getConsumer(final String consumerId, final ConsumeFromWhere consumeFromWhere) {
61	SimpleConsumer consumer = consumerMap.get(consumerId);	1✔
62	if (Objects.nonNull(consumer)) {	1✔
63	return consumer;	×
64	}
65
66	consumer = new SimpleConsumer(new SimpleConsumerConfig.Builder()	1✔
67	.setDataDir(config.getDataDir())	1✔
68	.setConsumerId(consumerId)	1✔
69	.setConsumeFromWhere(consumeFromWhere)	1✔
70	.build());	1✔
71	consumer.addCloseListener(() -> consumerMap.remove(consumerId));	1✔
72	consumerMap.put(consumerId, consumer);	1✔
73	return consumer;	1✔
74	}
75
76	@Override
77	public void close() {
78	try {
79	logDebug("[close] start");	1✔
80	if (isClosed) {	1✔
81	logDebug("[close] already closed");	1✔
82	return;	1✔
83	}
84	if (!simpleProducer.isClosed()) {	1✔
85	simpleProducer.close();	1✔
86	}
87	for (Map.Entry<String, SimpleConsumer> entry : consumerMap.entrySet()) {	1✔
88	SimpleConsumer consumer = entry.getValue();	1✔
89	if (!consumer.isClosed()) {	1✔
90	entry.getValue().close();	1✔
91	}
92	}	1✔
93	for (CloseListener listener : closeListeners) {	1✔
94	listener.onClose();	1✔
95	}	1✔
96	isClosed = true;	1✔
97	} finally {
98	logDebug("[close] end");	1✔
99	}	1✔
100	}	1✔
101
102
103	@Override
104	public void addCloseListener(CloseListener listener) {
105	closeListeners.add(listener);	1✔
106	}	1✔
107
108	// region logger
109
110	private void logDebug(String format) {
111	if (logger.isDebugEnabled()) {	1✔
NEW 112	logger.debug(format);	×
113	}
114	}	1✔
115
116	// endregion
117	}

wz2cool / local-queue / #32

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous