org.noear.luffy.utils.EventPipeline Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of luffy.base Show documentation
There is a newer version: 1.8.1
package org.noear.luffy.utils;

import java.util.ArrayList;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.function.Consumer;

/**
 * 写入时，先写到队列
 * 提交时，每次提交100条；消费完后暂停1秒
 *
 * */
public class EventPipeline implements TaskUtils.ITask {
    private long interval = 500;
    private long interval_min = 100;

    private int packetSize = 100; //必须大于100
    private int packetSize_min = 100; //必须大于100

    private Consumer> handler; //不要用Act1Ex

    public EventPipeline(Consumer> handler) {
        this.handler = handler;

        TaskUtils.run(this);
    }

    public EventPipeline(long interval,int packetSize, Consumer> handler) {
        this.handler = handler;

        setInterval(interval);
        setPacketSize(packetSize);
        TaskUtils.run(this);
    }

    //
    //
    //

    Queue queueLocal = new LinkedBlockingQueue<>();

    public void add(Event event) {
        try {
            queueLocal.add(event);
        } catch (Exception ex) {
            //不打印
        }
    }




    /**
     * 获取任务间隔时间
     * */
    @Override
    public long getInterval() {
        return interval;
    }

    public void setInterval(long interval) {
        if (interval >= interval_min) {
            this.interval = interval;
        }
    }

    /**
     * 设置管道包大小
     * */
    public void setPacketSize(int packetSize) {
        if (packetSize >= packetSize_min) {
            this.packetSize = packetSize;
        }
    }

    @Override
    public void exec() throws Exception {

        while (true) {
            List list = new ArrayList<>(packetSize);

            collectDo(list);

            if (list.size() > 0) {
                handler.accept(list);
            } else {
                break;
            }
        }
    }

    private void collectDo(List list) {
        //收集最多100条日志
        //
        int count = 0;
        while (true) {
            Event event = queueLocal.poll();

            if (event == null) {
                break;
            } else {
                list.add(event);
                count++;

                if (count == packetSize) {
                    break;
                }
            }
        }
    }
}