Added: Datadog::Utils::Compression

Author: delner

lib/ddtrace/utils/compression.rb

@@ -0,0 +1,27 @@
+require 'zlib'
+
+module Datadog
+  module Utils
+    # Common database-related utility functions.
+    module Compression
+      module_function
+
+      def gzip(string, level: nil, strategy: nil)
+        sio = StringIO.new
+        sio.binmode
+        gz = Zlib::GzipWriter.new(sio, level, strategy)
+        gz.write(string)
+        gz.close
+        sio.string
+      end
+
+      def gunzip(string, encoding = ::Encoding::ASCII_8BIT)
+        sio = StringIO.new(string)
+        gz = Zlib::GzipReader.new(sio, encoding: encoding)
+        gz.read
+      ensure
+        gz && gz.close
+      end
+    end
+  end
+end

spec/ddtrace/utils/compression_spec.rb

@@ -0,0 +1,26 @@
+require 'securerandom'
+require 'ddtrace/utils/compression'
+
+RSpec.describe Datadog::Utils::Compression do
+  describe '::gzip' do
+    subject(:gzip) { described_class.gzip(unzipped) }
+    let(:unzipped) { SecureRandom.uuid }
+
+    it { is_expected.to be_a_kind_of(String) }
+
+    context 'when result is unzipped' do
+      subject(:gunzip) { described_class.gunzip(gzip) }
+      it { is_expected.to eq(unzipped) }
+    end
+  end
+
+  describe '::gunzip' do
+    subject(:gunzip) { described_class.gunzip(zipped) }
+
+    context 'given a zipped string' do
+      let(:zipped) { described_class.gzip(unzipped) }
+      let(:unzipped) { SecureRandom.uuid }
+      it { is_expected.to eq(unzipped) }
+    end
+  end
+end