org.noear.water.protocol.solution.MsgQueueRedis 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.protocol.MsgQueue;
import org.noear.water.utils.TextUtils;

import java.io.IOException;
import java.util.function.Consumer;

/**
 * 基于 Redis 适配队列
 * */
public class MsgQueueRedis implements MsgQueue {
    RedisClient _redisX = null;
    String _queue_name;

    public MsgQueueRedis(String name, RedisClient redisX) {
        _queue_name = name;
        _redisX = redisX;

        if(TextUtils.isEmpty(_queue_name)){
            _queue_name = "water_msg_queue";
        }
    }

    @Override
    public boolean push(String msg_id) {
        _redisX.open((rs) -> rs.key(_queue_name).listAdd(msg_id));
        return true;
    }


    @Override
    public String poll() {
        return _redisX.openAndGet((rs) -> rs.key(_queue_name).listPop());
    }

    @Override
    public void pollGet(Consumer callback) {
        _redisX.open((rs) -> {
            while (true) {
                String msg = rs.key(_queue_name).listPop();

                if (msg == null) {
                    break;
                }

                callback.accept(msg);
            }
        });
    }


    @Override
    public long count() {
        return _redisX.openAndGet(rs -> rs.key(_queue_name).listLen());
    }

    @Override
    public void close() throws IOException {

    }
}