基于zookeeper的分布式队列之FIFO队列

2、FIFO队列

fifo就是数据结构中的,first in first out的意思,就是先进先出,这是队列的特点,那么结合队列的特点如何通过zookeeper来实现这个特点了,看图:

基于zookeeper的分布式队列之FIFO队列_第1张图片

1、入队列,我们可以在/queue节点下,不同的客户端创建临时有序节点,通过zookeeper的znode的类型我们知道,有序节点创建时会带编号的,且这个编号的有序递增的,就是为队列的先进创建的条件

2、出队列,如何先出队列呢,要获取到/queue节点下的所有子节点,然后获取子节点上的数字,对数字进行排序,然后取得到数字最小的节点,我们把这个节点pop出队列就可以了。

 

OK,代码搞起~~~~~

核心代码:

public class FIFOQueue {
    protected final ZkClient zkClient;

    protected final String root;

    //  /queue/n_00000000010
    protected static final String Node_name = "n_";

    public FIFOQueue(ZkClient zkClient, String root) {
        this.zkClient = zkClient;
        this.root = root;
    }

    public int size() {
        return zkClient.getChildren(root).size();
    }

    public boolean isEmpty() {
        return zkClient.getChildren(root).size() == 0;
    }

    public boolean push(T element) throws Exception {

        String path = root.concat("/").concat(Node_name);
        try {
            zkClient.createEphemeralSequential(path, element);
        }catch (ZkNoNodeException e) {
            zkClient.createPersistent(root);
            push(element);
        }catch (Exception e) {
            throw ExceptionUtil.convertToRuntimeException(e);
        }

        return true;
    }

    //拿到编号最小的,把最小数据删除
    public T poll() throws Exception  {

        try {
            List childrens =
                    zkClient.getChildren(root);
            if(childrens.size() == 0) {
                return null;
            }

            Collections.sort(childrens, new Comparator() {
                @Override
                public int compare(String o1, String o2) {
                    return getNodeNumber(o1,Node_name).compareTo
                            (getNodeNumber(o2,Node_name));
                }
            });

            //n_0000000001
            String litterNode = childrens.get(0);
            String fullPath = root.concat("/").concat(litterNode);
            T data = (T)zkClient.readData(fullPath);
            zkClient.delete(fullPath);
            return data;
        } catch (Exception e) {
            throw ExceptionUtil.convertToRuntimeException(e);
        }
    }

    private String getNodeNumber(String str,String nodeName) {
        int index = str.lastIndexOf(nodeName);
        if(index  >= 0) {
            index += Node_name.length();
            return index <= str.length() ? str.substring(index) : "";
        }
        return str;
    }
 }

如是,zookeeper的FIFO队列就完成了~~~

你可能感兴趣的:(zookeeper,FIFO,zookeeper队列,先进先出)