capnproto-java-rpc/runtime/src/main/java/org/capnproto/ReaderArena.java

package org.capnproto;

import java.util.ArrayList;
import java.nio.ByteBuffer;

public final class ReaderArena implements Arena {

    public long limit;
    // current limit

    public final ArrayList<SegmentReader> segments;

    public ReaderArena(ByteBuffer[] segmentSlices, long traversalLimitInWords) {
        this.limit = traversalLimitInWords;
        this.segments = new ArrayList<SegmentReader>();
        for(int ii = 0; ii < segmentSlices.length; ++ii) {
            this.segments.add(new SegmentReader(segmentSlices[ii], this));
        }
    }

    public SegmentReader tryGetSegment(int id) {
        return segments.get(id);
    }

    public final void checkReadLimit(int numBytes) {
        // TODO worry about thread safety?
        if (numBytes > limit) {
            return;
        } else {
            limit -= numBytes;
        }
    }
}