56 lines
1.9 KiB
Kotlin
56 lines
1.9 KiB
Kotlin
/* Copyright 2017 Andrew Dawson
|
|
*
|
|
* This file is a part of Tusky.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it under the terms of the
|
|
* GNU General Public License as published by the Free Software Foundation; either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* Tusky is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
|
|
* the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
|
|
* Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with Tusky; if not,
|
|
* see <http://www.gnu.org/licenses>. */
|
|
package com.keylesspalace.tusky.network
|
|
|
|
import java.io.IOException
|
|
import java.io.InputStream
|
|
import okhttp3.MediaType
|
|
import okhttp3.RequestBody
|
|
import okio.BufferedSink
|
|
|
|
class ProgressRequestBody(private val content: InputStream, private val contentLength: Long, private val mediaType: MediaType, private val uploadListener: UploadCallback) : RequestBody() {
|
|
fun interface UploadCallback {
|
|
fun onProgressUpdate(percentage: Int)
|
|
}
|
|
|
|
override fun contentType(): MediaType {
|
|
return mediaType
|
|
}
|
|
|
|
override fun contentLength(): Long {
|
|
return contentLength
|
|
}
|
|
|
|
@Throws(IOException::class)
|
|
override fun writeTo(sink: BufferedSink) {
|
|
val buffer = ByteArray(DEFAULT_BUFFER_SIZE)
|
|
var uploaded: Long = 0
|
|
|
|
content.use { content ->
|
|
var read: Int
|
|
while (content.read(buffer).also { read = it } != -1) {
|
|
uploadListener.onProgressUpdate((100 * uploaded / contentLength).toInt())
|
|
uploaded += read.toLong()
|
|
sink.write(buffer, 0, read)
|
|
}
|
|
uploadListener.onProgressUpdate((100 * uploaded / contentLength).toInt())
|
|
}
|
|
}
|
|
|
|
companion object {
|
|
private const val DEFAULT_BUFFER_SIZE = 2048
|
|
}
|
|
}
|