io.taig.communicator.body.Response.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of communicator_2.11 Show documentation

An OkHttp wrapper for Scala built with Android in mind

There is a newer version: 3.5.1

package io.taig.communicator.body

import java.io.InterruptedIOException

import io.taig.communicator.Progress
import okhttp3.ResponseBody
import okio._

/**
 * A ResponseBody wrapper that takes care of notifying the event listener
 *
 * @param wrapped Wrapped ResponseBody
 * @param event Event listener to update on progress, may be null
 */
class Response( wrapped: ResponseBody, var event: Progress.Receive ⇒ Unit, zipped: Boolean ) extends ResponseBody {
    /**
     * Prevent recreating the length object on every listener call
     */
    private lazy val length = contentLength() match {
        case -1     ⇒ None
        case length ⇒ Some( length )
    }

    @throws[InterruptedIOException]( "If the request was canceled" )
    private def update( current: Long ) = if ( event != null ) event( Progress.Receive( current, length ) )

    override def source() = {
        if ( zipped ) {
            Okio.buffer( new GzipSource( new Source( wrapped.source() ) ) )
        } else {
            Okio.buffer( new Source( wrapped.source() ) )
        }
    }

    override def contentLength() = wrapped.contentLength()

    override def contentType() = wrapped.contentType()

    private class Source( wrapped: okio.BufferedSource )
            extends ForwardingSource( wrapped ) {
        private var current = 0L

        override def read( sink: Buffer, count: Long ) = {
            if ( current == 0 ) {
                update( 0 )
            }

            super.read( sink, count ) match {
                case count if count != -1 ⇒
                    current += count
                    update( current )
                    count
                case _ ⇒ -1
            }
        }
    }
}