forked from KNEMC/KLALB
70 lines
1.4 KiB
Java
70 lines
1.4 KiB
Java
package org.kne.cloud.network;
|
|
|
|
import java.lang.foreign.Arena;
|
|
import java.lang.foreign.MemorySegment;
|
|
import java.lang.foreign.MemorySegment.Scope;
|
|
import java.lang.ref.PhantomReference;
|
|
import java.lang.ref.Reference;
|
|
import java.lang.ref.ReferenceQueue;
|
|
import java.nio.ByteBuffer;
|
|
import java.util.concurrent.atomic.AtomicLong;
|
|
import java.util.concurrent.atomic.AtomicReference;
|
|
import java.util.concurrent.locks.ReentrantLock;
|
|
|
|
public class HeapBufferedArena implements Arena{
|
|
private static long blockSize=32768L;
|
|
private MemorySegment segment;
|
|
private long pos=0;
|
|
|
|
|
|
|
|
|
|
public static void setBlockSize(long blockSizex) {
|
|
blockSize = blockSizex;
|
|
}
|
|
|
|
|
|
|
|
|
|
private void allocateNew(long blockSize2) {
|
|
segment=MemorySegment.ofArray(ByteBufferAllocator.allocateUninitializedArray((int) blockSize2));
|
|
pos=0;
|
|
}
|
|
|
|
|
|
public HeapBufferedArena() {
|
|
super();
|
|
allocateNew(blockSize);
|
|
}
|
|
|
|
@Override
|
|
public MemorySegment allocate(long byteSize, long byteAlignment) {
|
|
if(byteSize>blockSize)
|
|
return MemorySegment.ofArray(ByteBufferAllocator.allocateUninitializedArray((int) byteSize));
|
|
do{
|
|
long newpos=pos+byteSize;
|
|
if(newpos>blockSize) {
|
|
allocateNew(blockSize);
|
|
}else {
|
|
MemorySegment ms=segment.asSlice(pos, byteSize);
|
|
pos=newpos;
|
|
return ms;
|
|
}
|
|
}while(true);
|
|
|
|
}
|
|
|
|
@Override
|
|
public Scope scope() {
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public void close() {
|
|
}
|
|
public static HeapBufferedArena ofBuffered() {
|
|
return new HeapBufferedArena();
|
|
}
|
|
|
|
}
|