com.firenio.buffer.PooledHeapByteBuf Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firenio-all Show documentation

The all in one project of firenio

There is a newer version: 1.3.6

/*
 * Copyright 2015 The FireNio Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.firenio.buffer;

import com.firenio.collection.ObjectPool;

final class PooledHeapByteBuf extends HeapByteBuf {

    private final PooledByteBufAllocator allocator;
    private final ObjectPool    pool;
    private       int                    capacity;
    private       int                    unitOffset;

    PooledHeapByteBuf(PooledByteBufAllocator allocator, ObjectPool pool, byte[] memory) {
        super(memory);
        this.pool = pool;
        this.allocator = allocator;
    }

    @Override
    public long address() {
        return allocator.getAddress();
    }

    @Override
    public int capacity() {
        return capacity;
    }

    @Override
    public void capacity(int cap) {
        this.capacity = cap;
    }

    @Override
    public final void expansion(int cap) {
        allocator.expansion(this, cap);
    }

    @Override
    public boolean isPooled() {
        return true;
    }

    @Override
    protected PooledHeapByteBuf produce(int unitOffset, int unitSize) {
        int unit = allocator.getUnit();
        this.offset = unitOffset * unit;
        this.capacity = unitSize * unit;
        this.abs_read_index = offset;
        this.abs_write_index = offset;
        this.unitOffset = unitOffset;
        this.referenceCount = 1;
        return this;
    }

    @Override
    protected final void release0() {
        allocator.release(this);
    }

    @Override
    protected int unitOffset() {
        return unitOffset;
    }

    @Override
    protected void unitOffset(int unitOffset) {
        this.unitOffset = unitOffset;
    }

    @Override
    protected void recycleObject() {
        pool.push(this);
    }

}