import androidx.annotation.NonNull;
import androidx.annotation.Nullable;

import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.ReentrantLock;

import hooyee.IDataElement;

/**
 * 一個(gè)緩存隊(duì)列，用于管理
 * [a[], b[], c[]]形式的數(shù)據(jù)酸员，從其中取出指定長(zhǎng)度的數(shù)據(jù)乏屯，若指定的長(zhǎng)度小于隊(duì)首的可用長(zhǎng)度姻僧，則返回該隊(duì)首元素（該元素不出隊(duì)）帽蝶，并修改該元素的region.
 * 若大于隊(duì)首称近，則直接返回隊(duì)首造虎，并將隊(duì)首元素出隊(duì)傅蹂，可用范圍是 T 的[startPosition, endPosition)
 * @param <T>
 */
public class DataCache<T extends IDataElement> {
    private long MAX_SIZE;
    private final AtomicLong totalSize = new AtomicLong();
    /**
     * dataQueue的poll和offer操作也需要保證線程安全，因此變更數(shù)據(jù)結(jié)構(gòu)
     */
    Queue<T> dataQueue = new ConcurrentLinkedQueue<>();

    /**
     * offer和poll需要通過(guò)overflow來(lái)確認(rèn)算凿，而overflow又依賴(lài)于updateSize份蝴，因此需要保證poll和offer的整個(gè)流程的原子性（其實(shí)只要保證offer，因?yàn)閜oll是會(huì)減小他的值氓轰，這樣只會(huì)更加滿足條件）
     **/
    private final ReentrantLock pollLock = new ReentrantLock();
    private final ReentrantLock offerLock = new ReentrantLock();

    public DataCache() {
        MAX_SIZE = 1024 * 1024 * 4;
    }

    public DataCache(long maxSize) {
        MAX_SIZE = maxSize;
    }

    /**
     * 線程安全的, 只用于T為 IElement的時(shí)候婚夫，若不是與不帶參的效果一樣
     * @param length 從隊(duì)列的第一個(gè)元素中獲取指定長(zhǎng)度的數(shù)據(jù)，如果第一個(gè)元素小于改長(zhǎng)度署鸡，調(diào)用poll案糙，若大于則調(diào)用peek,并移動(dòng)offset等信息
     * @return
     */
    public @Nullable T poll(int length) {
        pollLock.lock();
        try {
            T data = dataQueue.peek();
            if (data != null) {
                removeIfOver(data, length);
            }
            return data;
        } finally {
            pollLock.unlock();
        }
    }

    /**
     * 若是queue中的元素可用長(zhǎng)度[endPosition, length)超過(guò)入?yún)ength限嫌，則不出隊(duì)列，以備下次使用
     * @param data
     * @param length
     */
    private boolean removeIfOver(@NonNull T data, int length) {
        // 未被讀取的數(shù)據(jù)
        if (data.getLength() - data.getEndPosition() > length) {
            // 大于length只截取[start, end)部分
            // 移動(dòng)被讀取的region[start, end)
            data.setStartPosition(data.getEndPosition());
            data.setEndPosition(data.getEndPosition() + length);
            return false;
        } else {
            data.setStartPosition(data.getEndPosition());
            data.setEndPosition(data.getLength());
            dataQueue.poll();
            updateSize(-data.getLength());
            return true;
        }
    }

    /**
     * 線程安全的
     *
     * @param data
     */
    public void offer(@NonNull T data) {
        if (overflow()) {
            return;
        }
        offerLock.lock();
        try {
            if (!overflow()) {
                dataQueue.offer(data);
                updateSize(data.getLength());
            } else {

            }
        } finally {
            offerLock.unlock();
        }
    }

    /**
     * 清空數(shù)據(jù)
     */
    public void clear() {
        dataQueue.clear();
        totalSize.set(0);
    }

    private void updateSize(long size) {
        totalSize.addAndGet(size);
    }

    /**
     * true : 滿了
     *
     * @return
     */
    private boolean overflow() {
        return totalSize.get() >= MAX_SIZE;
    }
}