From 0234795825640022ccd43476e3c679d4e054a46f Mon Sep 17 00:00:00 2001
From: Andrew Newdigate <andrew@gitlab.com>
Date: Mon, 26 Aug 2019 18:49:58 +0200
Subject: Remove all references to Gitlab::CorrelationId

This class has been replaced with Labkit::Correlation::CorrelationId
---
 lib/gitlab/correlation_id.rb    | 40 ----------------------------------------
 lib/gitlab/tracing.rb           |  2 +-
 spec/lib/gitlab/tracing_spec.rb |  2 +-
 3 files changed, 2 insertions(+), 42 deletions(-)
 delete mode 100644 lib/gitlab/correlation_id.rb

diff --git a/lib/gitlab/correlation_id.rb b/lib/gitlab/correlation_id.rb
deleted file mode 100644
index 0f9bde4390e..00000000000
--- a/lib/gitlab/correlation_id.rb
+++ /dev/null
@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-module Gitlab
-  module CorrelationId
-    LOG_KEY = 'correlation_id'.freeze
-
-    class << self
-      def use_id(correlation_id, &blk)
-        # always generate a id if null is passed
-        correlation_id ||= new_id
-
-        ids.push(correlation_id || new_id)
-
-        begin
-          yield(current_id)
-        ensure
-          ids.pop
-        end
-      end
-
-      def current_id
-        ids.last
-      end
-
-      def current_or_new_id
-        current_id || new_id
-      end
-
-      private
-
-      def ids
-        Thread.current[:correlation_id] ||= []
-      end
-
-      def new_id
-        SecureRandom.uuid
-      end
-    end
-  end
-end
diff --git a/lib/gitlab/tracing.rb b/lib/gitlab/tracing.rb
index 29517591c51..7732d7c9d9c 100644
--- a/lib/gitlab/tracing.rb
+++ b/lib/gitlab/tracing.rb
@@ -30,7 +30,7 @@ module Gitlab
       # Avoid using `format` since it can throw TypeErrors
       # which we want to avoid on unsanitised env var input
       tracing_url_template.to_s
-        .gsub(/\{\{\s*correlation_id\s*\}\}/, Gitlab::CorrelationId.current_id.to_s)
+        .gsub(/\{\{\s*correlation_id\s*\}\}/, Labkit::Correlation::CorrelationId.current_id.to_s)
         .gsub(/\{\{\s*service\s*\}\}/, Gitlab.process_name)
     end
   end
diff --git a/spec/lib/gitlab/tracing_spec.rb b/spec/lib/gitlab/tracing_spec.rb
index db75ce2a998..e913bb600ec 100644
--- a/spec/lib/gitlab/tracing_spec.rb
+++ b/spec/lib/gitlab/tracing_spec.rb
@@ -59,7 +59,7 @@ describe Gitlab::Tracing do
       it 'returns the correct state for .tracing_url' do
         expect(described_class).to receive(:tracing_url_enabled?).and_return(tracing_url_enabled?)
         allow(described_class).to receive(:tracing_url_template).and_return(tracing_url_template)
-        allow(Gitlab::CorrelationId).to receive(:current_id).and_return(correlation_id)
+        allow(Labkit::Correlation::CorrelationId).to receive(:current_id).and_return(correlation_id)
         allow(Gitlab).to receive(:process_name).and_return(process_name)
 
         expect(described_class.tracing_url).to eq(tracing_url)
-- 
cgit v1.2.1