-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: Art Shendrik <[email protected]>
- Loading branch information
Showing
13 changed files
with
1,559 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,151 @@ | ||
| @file:Suppress("KDocUnresolvedReference") | ||
|
|
||
| package fluxo.io | ||
|
|
||
| import fluxo.io.nio.markCompat | ||
| import fluxo.io.nio.resetCompat | ||
| import fluxo.io.util.EMPTY_BYTE_ARRAY | ||
| import java.io.FileInputStream | ||
| import java.io.FileOutputStream | ||
| import java.io.InputStream | ||
| import java.nio.ByteBuffer | ||
| import java.nio.channels.FileChannel | ||
| import kotlin.math.ln | ||
| import kotlin.math.pow | ||
|
|
||
|
|
||
| fun readableFileSize(bytes: Int): String = readableFileSize(bytes.toLong()) | ||
|
|
||
| /** | ||
| * Convert byte size into the human-readable format. | ||
| */ | ||
| fun readableFileSize(bytes: Long): String { | ||
| val unit = 1024.0 | ||
| if (bytes < unit) return "$bytes B" | ||
| val exp = (ln(bytes.toDouble()) / ln(unit)).toInt() | ||
| val pre = "kMGTPE"[exp - 1] | ||
| return "%.1f %sB".format(bytes / unit.pow(exp.toDouble()), pre) | ||
| } | ||
|
|
||
|
|
||
| /** | ||
| * Reads this stream completely into a byte array. | ||
| * | ||
| * **Note**: It is the caller's responsibility to close this stream. | ||
| * | ||
| * @param size Expected size of the data. Will use [readBytes] when [size] == `-1`. | ||
| * | ||
| * @see readBytesExact | ||
| * @see kotlin.io.readBytes | ||
| * @see java.io.DataInputStream.readFully | ||
| * @see java.io.InputStream.readNBytes | ||
| * | ||
| * @throws EOFException On unexpected stream end or when stream is not ended on expected size. | ||
| * @throws IllegalArgumentException When [size] < `-1` | ||
| */ | ||
| @Throws(IOException::class) | ||
| fun InputStream.readBytesFully(size: Int): ByteArray { | ||
| if (size < 0) { | ||
| require(size == -1) { "size < -1: $size" } | ||
|
|
||
| // simple reading of stream with unknown size | ||
| return readBytes() | ||
| } | ||
|
|
||
| if (size == 0) { | ||
| return EMPTY_BYTE_ARRAY | ||
| } | ||
|
|
||
| var offset = 0 | ||
| var remaining = size | ||
| val result = ByteArray(remaining) | ||
| while (remaining > 0) { | ||
| val read = read(result, offset, remaining) | ||
| if (read < 0) { | ||
| break | ||
| } | ||
| remaining -= read | ||
| offset += read | ||
| } | ||
| if (remaining > 0) { | ||
| throw EOFException("$offset < expected size: $size") | ||
| } | ||
| if (read() != -1) { | ||
| throw EOFException("Stream is not ended at expected size: $size") | ||
| } | ||
| return result | ||
| } | ||
|
|
||
| /** | ||
| * Reads up to a specified number of bytes from the input stream. | ||
| * | ||
| * **Note**: It is the caller's responsibility to close this stream. | ||
| * | ||
| * @param size Expected size of the data. | ||
| * @param strict Allows disabling exception on the unexpected end of the stream. | ||
| * | ||
| * @see readBytesFully | ||
| * @see java.io.InputStream.readNBytes | ||
| * | ||
| * @throws EOFException On unexpected stream end or when stream | ||
| */ | ||
| @Throws(IOException::class) | ||
| fun InputStream.readBytesExact(size: Int, strict: Boolean = true): ByteArray { | ||
| if (size == 0) { | ||
| return EMPTY_BYTE_ARRAY | ||
| } | ||
| val result = ByteArray(size) | ||
| var offset = 0 | ||
| var read: Int | ||
| while (offset < size) { | ||
| read = read(result, offset, size - offset) | ||
| if (read < 0) { | ||
| if (strict) { | ||
| throw EOFException("$offset < expected size: $size") | ||
| } | ||
| return result.copyOf(offset) | ||
| } | ||
| offset += read | ||
| } | ||
| return result | ||
| } | ||
|
|
||
|
|
||
| fun ByteBuffer.toArray(): ByteArray { | ||
| val array = ByteArray(remaining()) | ||
| markCompat() | ||
| get(array) | ||
| resetCompat() | ||
| return array | ||
| } | ||
|
|
||
|
|
||
| /** | ||
| * @see kotlin.io.use | ||
| * @see java.nio.channels.FileChannel.lock | ||
| */ | ||
| inline fun <R> FileInputStream.useLocked(block: (FileInputStream) -> R): R = | ||
| use { s -> s.channel.useReadLocked { block(s) } } | ||
|
|
||
| /** | ||
| * @see kotlin.io.use | ||
| * @see java.nio.channels.FileChannel.lock | ||
| */ | ||
| inline fun <R> FileOutputStream.useLocked(block: (FileOutputStream) -> R): R = | ||
| use { s -> s.channel.useLocked { block(s) } } | ||
|
|
||
| /** | ||
| * @see kotlin.io.use | ||
| * @see java.nio.channels.FileChannel.lock | ||
| */ | ||
| inline fun <R> FileChannel.useLocked(block: (FileChannel) -> R): R = | ||
| // lock will be auto-closed on channel close. | ||
| use { ch -> ch.lock(); block(ch) } | ||
|
|
||
| /** | ||
| * @see kotlin.io.use | ||
| * @see java.nio.channels.FileChannel.lock | ||
| */ | ||
| inline fun <R> FileChannel.useReadLocked(block: (FileChannel) -> R): R = | ||
| // lock will be auto-closed on channel close. | ||
| use { ch -> ch.lock(0, java.lang.Long.MAX_VALUE, true); block(ch) } |
Oops, something went wrong.