tursom/TursomServer
0
0
Fork 0
mirror of https://github.com/tursom/TursomServer.git synced 2025-03-13 11:20:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

add NioBuffers

Browse Source
This commit is contained in:
tursom 2021-07-18 01:27:27 +08:00
parent d403a6192c
commit 159fe0ff52
2 changed files with 77 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
ts-core/ts-buffer/src/main/kotlin/cn/tursom/core/buffer/ByteBufferExtension.kt
View File

@ -11,13 +11,9 @@ import cn.tursom.core.buffer.NioBuffers.Lists.Companion.readLists
import cn.tursom.core.buffer.NioBuffers.Lists.Companion.writeLists
import cn.tursom.core.buffer.NioBuffers.Sequences.Companion.readSequences
import cn.tursom.core.buffer.NioBuffers.Sequences.Companion.writeSequences
import cn.tursom.core.buffer.NioBuffers.finishRead
import cn.tursom.core.buffer.NioBuffers.finishWrite
import cn.tursom.core.buffer.NioBuffers.getReadNioBufferList
import cn.tursom.core.buffer.NioBuffers.getWriteNioBufferList
import cn.tursom.core.buffer.NioBuffers.readNioBuffers
import cn.tursom.core.buffer.NioBuffers.writeNioBuffers
import cn.tursom.core.buffer.impl.ArrayByteBuffer
import cn.tursom.core.buffer.impl.HeapByteBuffer
import cn.tursom.core.buffer.impl.ListByteBuffer
import cn.tursom.core.toBytes
import cn.tursom.core.toInt
import java.nio.ByteOrder
@ -51,24 +47,6 @@ inline fun <T> ByteBuffer.write(block: (java.nio.ByteBuffer) -> T): T {
}
}
//inline fun <T> MultipleByteBuffer.reads(block: (Sequence<java.nio.ByteBuffer>) -> T): T {
// val bufferList = readBufferSequence()
// try {
// return block(bufferList)
// } finally {
// finishRead(bufferList)
// }
//}
//inline fun <T> MultipleByteBuffer.writes(block: (Sequence<java.nio.ByteBuffer>) -> T): T {
// val bufferList = writeBufferSequence()
// try {
// return block(bufferList)
// } finally {
// finishWrite(bufferList)
// }
//}
fun ReadableByteChannel.read(buffer: ByteBuffer): Int {
if (this is ScatteringByteChannel) {
val arrays = buffer.getExtension(NioBuffers.Arrays)
@ -122,37 +100,19 @@ fun WritableByteChannel.write(buffer: ByteBuffer): Int {
return buffer.read { write(it) }
}
//fun ScatteringByteChannel.read(buffer: MultipleByteBuffer): Long {
// return buffer.writeBuffers { read(it.toList().toTypedArray()) }
//}
//fun GatheringByteChannel.write(buffer: MultipleByteBuffer): Long {
// return buffer.readBuffers { write(it.toList().toTypedArray()) }
//}
fun ScatteringByteChannel.read(buffers: Array<out ByteBuffer>): Long {
val bufferList = buffers.iterator().getWriteNioBufferList()
val bufferArray = bufferList.toTypedArray()
return try {
read(bufferArray)
} finally {
val iterator = bufferList.iterator()
buffers.iterator().finishWrite(iterator)
}
fun ScatteringByteChannel.read(
buffers: Array<out ByteBuffer>
): Long = buffers.writeNioBuffers { bufferList ->
read(bufferList.toTypedArray())
}
fun GatheringByteChannel.write(buffers: Array<out ByteBuffer>): Long {
val bufferList = buffers.iterator().getReadNioBufferList()
val bufferArray = bufferList.toTypedArray()
return try {
write(bufferArray)
} finally {
val iterator = bufferList.iterator()
buffers.iterator().finishRead(iterator)
}
fun GatheringByteChannel.write(
buffers: Array<out ByteBuffer>
): Long = buffers.readNioBuffers { bufferList ->
write(bufferList.toTypedArray())
}
fun Array<out ByteBuffer>.asMultipleByteBuffer() = ArrayByteBuffer(*this)
fun Array<out ByteBuffer>.asMultipleByteBuffer() = ArrayByteBuffer(buffers = this)
val Collection<ByteBuffer>.readable: Int
get() {
@ -231,8 +191,3 @@ fun ByteBuffer.putLongWithSize(l: Long, size: Int, byteOrder: ByteOrder = ByteOr
}
}
}
fun main() {
println(HeapByteBuffer(1).getExtension(NioBuffers.Sequences))
println(ListByteBuffer().getExtension(NioBuffers.Sequences))
}

66
ts-core/ts-buffer/src/main/kotlin/cn/tursom/core/buffer/NioBuffers.kt
View File

@ -2,6 +2,72 @@ package cn.tursom.core.buffer
object NioBuffers {
inline fun <T> Array<out ByteBuffer>.readNioBuffers(
action: (bufferList: List<java.nio.ByteBuffer>) -> T
): T {
val bufferList = getReadNioBufferList()
try {
return action(bufferList)
} finally {
finishRead(bufferList.iterator())
}
}
inline fun <T> Sequence<ByteBuffer>.readNioBuffers(
action: (bufferList: List<java.nio.ByteBuffer>) -> T
): T {
val bufferList = getReadNioBufferList()
try {
return action(bufferList)
} finally {
finishRead(bufferList.iterator())
}
}
inline fun <T> Iterable<ByteBuffer>.readNioBuffers(
action: (bufferList: List<java.nio.ByteBuffer>) -> T
): T {
val bufferList = getReadNioBufferList()
try {
return action(bufferList)
} finally {
finishRead(bufferList.iterator())
}
}
inline fun <T> Array<out ByteBuffer>.writeNioBuffers(
action: (bufferList: List<java.nio.ByteBuffer>) -> T
): T {
val bufferList = getWriteNioBufferList()
try {
return action(bufferList)
} finally {
finishWrite(bufferList.iterator())
}
}
inline fun <T> Sequence<ByteBuffer>.writeNioBuffers(
action: (bufferList: List<java.nio.ByteBuffer>) -> T
): T {
val bufferList = getWriteNioBufferList()
try {
return action(bufferList)
} finally {
finishWrite(bufferList.iterator())
}
}
inline fun <T> Iterable<ByteBuffer>.writeNioBuffers(
action: (bufferList: List<java.nio.ByteBuffer>) -> T
): T {
val bufferList = getWriteNioBufferList()
try {
return action(bufferList)
} finally {
finishWrite(bufferList.iterator())
}
}
fun Array<out ByteBuffer>.getReadNioBufferList(): List<java.nio.ByteBuffer> = iterator().getReadNioBufferList()
fun Sequence<ByteBuffer>.getReadNioBufferList(): List<java.nio.ByteBuffer> = iterator().getReadNioBufferList()
fun Iterable<ByteBuffer>.getReadNioBufferList(): List<java.nio.ByteBuffer> = iterator().getReadNioBufferList()