require 'zlib'
require 'stringio'

module Dalli
  class Compressor
    def self.compress(data)
      Zlib::Deflate.deflate(data)
    end

    def self.decompress(data)
      Zlib::Inflate.inflate(data)
    end
  end

  class GzipCompressor
    def self.compress(data)
      io = StringIO.new("w")
      gz = Zlib::GzipWriter.new(io)
      gz.write(data)
      gz.close
      io.string
    end

    def self.decompress(data)
      io = StringIO.new(data, "rb")
      Zlib::GzipReader.new(io).read
    end
  end
end
