tursom/TursomServer
0
0
Fork 0
mirror of https://github.com/tursom/TursomServer.git synced 2025-03-24 08:30:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

add method ByteBuffer.read(InputStream)

Browse Source
This commit is contained in:
tursom 2020-01-21 12:42:34 +08:00
parent 3a11dac3b4
commit bfebd1e5c6
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main/kotlin/cn/tursom/core/buffer/ByteBuffer.kt
View File

@ -2,6 +2,7 @@ package cn.tursom.core.buffer
import cn.tursom.core.forEachIndex
import java.io.Closeable
import java.io.InputStream
import java.io.OutputStream
import kotlin.math.min
@ -198,6 +199,17 @@ interface ByteBuffer : Closeable {
array.forEachIndex(index, index + size - 1, this::put)
}
fun put(inputStream: InputStream) {
if (hasArray) {
val read = inputStream.read(array, writeOffset, writeable)
writePosition += read
} else {
val buffer = ByteArray(10 * 1024)
val read = inputStream.read(buffer)
put(buffer, 0, read)
}
}
fun putByte(byte: Byte): Unit = put(byte)
fun putChar(char: Char): Unit = put(char)
fun putShort(short: Short): Unit = put(short)