99 lines
2.9 KiB
Java
99 lines
2.9 KiB
Java
/*
|
|
* Copyright 2021 The Netty Project
|
|
*
|
|
* The Netty Project licenses this file to you 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:
|
|
*
|
|
* https://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 io.netty.buffer.api.internal;
|
|
|
|
import io.netty.buffer.ByteBuf;
|
|
import io.netty.buffer.ByteBufAllocator;
|
|
import io.netty.buffer.api.Buffer;
|
|
import io.netty.buffer.api.Drop;
|
|
import io.netty.buffer.api.adaptor.BufferIntegratable;
|
|
import io.netty.buffer.api.adaptor.ByteBufAdaptor;
|
|
import io.netty.buffer.api.adaptor.ByteBufAllocatorAdaptor;
|
|
import io.netty.util.IllegalReferenceCountException;
|
|
import io.netty.util.ReferenceCounted;
|
|
|
|
public abstract class AdaptableBuffer<T extends ResourceSupport<Buffer, T>>
|
|
extends ResourceSupport<Buffer, T> implements BufferIntegratable, Buffer {
|
|
protected AdaptableBuffer(Drop<T> drop) {
|
|
super(drop);
|
|
}
|
|
|
|
private volatile ByteBufAdaptor adaptor;
|
|
|
|
public ByteBuf initialise(ByteBufAllocatorAdaptor alloc, int maxCapacity) {
|
|
return new ByteBufAdaptor(alloc, this, maxCapacity);
|
|
}
|
|
|
|
@Override
|
|
public ByteBuf asByteBuf() {
|
|
ByteBufAdaptor bba = adaptor;
|
|
if (bba == null) {
|
|
ByteBufAllocatorAdaptor alloc = (ByteBufAllocatorAdaptor) ByteBufAllocator.DEFAULT;
|
|
return adaptor = new ByteBufAdaptor(alloc, this, Integer.MAX_VALUE);
|
|
}
|
|
return bba;
|
|
}
|
|
|
|
@Override
|
|
public int refCnt() {
|
|
return isAccessible()? 1 + countBorrows() : 0;
|
|
}
|
|
|
|
@Override
|
|
public ReferenceCounted retain() {
|
|
return retain(1);
|
|
}
|
|
|
|
@Override
|
|
public ReferenceCounted retain(int increment) {
|
|
for (int i = 0; i < increment; i++) {
|
|
acquire();
|
|
}
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public ReferenceCounted touch() {
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public ReferenceCounted touch(Object hint) {
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public boolean release() {
|
|
return release(1);
|
|
}
|
|
|
|
@Override
|
|
public boolean release(int decrement) {
|
|
int refCount = 1 + countBorrows();
|
|
if (!isAccessible() || decrement > refCount) {
|
|
throw new IllegalReferenceCountException(refCount, -decrement);
|
|
}
|
|
for (int i = 0; i < decrement; i++) {
|
|
try {
|
|
close();
|
|
} catch (RuntimeException e) {
|
|
throw new IllegalReferenceCountException(e);
|
|
}
|
|
}
|
|
return !isAccessible();
|
|
}
|
|
}
|