lame.gzip.DeflateCompressor.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lame_2.13 Show documentation

lame

The newest version!

/*
 * Copyright (C) 2009-2019 Lightbend Inc. 
 */

package lame.gzip

import java.util.zip.Deflater

import akka.util.{ByteString, ByteStringBuilder}

import scala.annotation.tailrec

object DeflateCompressor {
  val MinBufferSize = 1024

  @tailrec
  def drainDeflater(
      deflater: Deflater,
      buffer: Array[Byte],
      result: ByteStringBuilder = new ByteStringBuilder(),
      flush: Boolean = false
  ): ByteString = {
    val flushI = if (flush) Deflater.SYNC_FLUSH else Deflater.NO_FLUSH
    val len = deflater.deflate(buffer, 0, buffer.length, flushI)
    if (len > 0) {
      result ++= ByteString.fromArray(buffer, 0, len)
      drainDeflater(deflater, buffer, result)
    } else {
      require(deflater.needsInput())
      result.result()
    }
  }
}