atatus 1.3.0 → 1.4.0

data/lib/atatus/transport/headers.rb

@@ -0,0 +1,83 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Atatus
+  module Transport
+    # @api private
+    class Headers
+      HEADERS = {
+        'Content-Type' => 'application/x-ndjson',
+        'Transfer-Encoding' => 'chunked'
+      }.freeze
+      GZIP_HEADERS = HEADERS.merge(
+        'Content-Encoding' => 'gzip'
+      ).freeze
+
+      def initialize(config, initial: {})
+        @config = config
+        @hash = build!(initial)
+      end
+
+      attr_accessor :hash
+
+      def [](key)
+        @hash[key]
+      end
+
+      def []=(key, value)
+        @hash[key] = value
+      end
+
+      def merge(other)
+        self.class.new(@config, initial: @hash.merge(other))
+      end
+
+      def merge!(other)
+        @hash.merge!(other)
+        self
+      end
+
+      def to_h
+        @hash
+      end
+
+      def chunked
+        merge(
+          @config.http_compression? ? GZIP_HEADERS : HEADERS
+        )
+      end
+
+      private
+
+      def build!(headers)
+        headers[:'User-Agent'] = UserAgent.new(@config).to_s
+
+        if (token = @config.secret_token)
+          headers[:Authorization] = "Bearer #{token}"
+        end
+
+        if (api_key = @config.api_key)
+          headers[:Authorization] = "ApiKey #{api_key}"
+        end
+
+        headers
+      end
+    end
+  end
+end

data/lib/atatus/transport/serializers.rb

@@ -1,6 +1,21 @@
-# frozen_string_literal: true
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
 
-require 'json'
+# frozen_string_literal: true
 
 module Atatus
   module Transport
@@ -57,8 +72,6 @@ module Atatus
         end
 
         attr_reader :transaction, :span, :error, :metadata, :metricset
-
-        # rubocop:disable Metrics/MethodLength
         def serialize(resource)
           case resource
           when Transaction
@@ -75,7 +88,6 @@ module Atatus
             raise UnrecognizedResource, resource.inspect
           end
         end
-        # rubocop:enable Metrics/MethodLength
       end
 
       def self.new(config)

data/lib/atatus/transport/serializers/context_serializer.rb

@@ -1,3 +1,20 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 # frozen_string_literal: true
 
 module Atatus
@@ -13,13 +30,13 @@ module Atatus
             tags: mixed_object(context.labels),
             request: build_request(context.request),
             response: build_response(context.response),
-            user: build_user(context.user)
+            user: build_user(context.user),
+            service: build_service(context.service)
           }
         end
 
         private
 
-        # rubocop:disable Metrics/MethodLength
         def build_request(request)
           return unless request
 
@@ -34,7 +51,6 @@ module Atatus
             url: build_url(request.url)
           }
         end
-        # rubocop:enable Metrics/MethodLength
 
         def build_response(response)
           return unless response
@@ -79,6 +95,17 @@ module Atatus
             hash: keyword_field(url.hash)
           }
         end
+
+        def build_service(service)
+          return unless service
+
+          {
+            framework: {
+              name: keyword_field(service.framework.name),
+              version: keyword_field(service.framework.version)
+            }
+          }
+        end
       end
     end
   end

data/lib/atatus/transport/serializers/error_serializer.rb

@@ -1,3 +1,20 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 # frozen_string_literal: true
 
 module Atatus
@@ -9,7 +26,6 @@ module Atatus
           @context_serializer ||= ContextSerializer.new(config)
         end
 
-        # rubocop:disable Metrics/MethodLength, Metrics/AbcSize
         def build(error)
           base = {
             id: error.id,
@@ -36,7 +52,6 @@ module Atatus
 
           { error: base }
         end
-        # rubocop:enable Metrics/MethodLength, Metrics/AbcSize
 
         private
 

data/lib/atatus/transport/serializers/metadata_serializer.rb

@@ -1,3 +1,20 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 # frozen_string_literal: true
 
 module Atatus
@@ -18,31 +35,36 @@ module Atatus
 
         private
 
-        # rubocop:disable Metrics/MethodLength, Metrics/AbcSize
         def build_service(service)
-          {
-            name: keyword_field(service.name),
-            environment: keyword_field(service.environment),
-            version: keyword_field(service.version),
-            agent: {
-              name: keyword_field(service.agent.name),
-              version: keyword_field(service.agent.version)
-            },
-            framework: {
-              name: keyword_field(service.framework.name),
-              version: keyword_field(service.framework.version)
-            },
-            language: {
-              name: keyword_field(service.language.name),
-              version: keyword_field(service.language.version)
-            },
-            runtime: {
-              name: keyword_field(service.runtime.name),
-              version: keyword_field(service.runtime.version)
+          base =
+            {
+              name: keyword_field(service.name),
+              environment: keyword_field(service.environment),
+              version: keyword_field(service.version),
+              agent: {
+                name: keyword_field(service.agent.name),
+                version: keyword_field(service.agent.version)
+              },
+              framework: {
+                name: keyword_field(service.framework.name),
+                version: keyword_field(service.framework.version)
+              },
+              language: {
+                name: keyword_field(service.language.name),
+                version: keyword_field(service.language.version)
+              },
+              runtime: {
+                name: keyword_field(service.runtime.name),
+                version: keyword_field(service.runtime.version)
+              }
             }
-          }
+
+          if node_name = service.node_name
+            base[:node] = { name: keyword_field(node_name) }
+          end
+
+          base
         end
-        # rubocop:enable Metrics/MethodLength, Metrics/AbcSize
 
         def build_process(process)
           {

data/lib/atatus/transport/serializers/metricset_serializer.rb

@@ -1,3 +1,20 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 # frozen_string_literal: true
 
 module Atatus
@@ -6,13 +23,24 @@ module Atatus
       # @api private
       class MetricsetSerializer < Serializer
         def build(metricset)
-          {
-            metricset: {
-              timestamp: metricset.timestamp.to_i,
-              tags: keyword_object(metricset.labels),
-              samples: build_samples(metricset.samples)
-            }
+          payload = {
+            timestamp: metricset.timestamp.to_i,
+            samples: build_samples(metricset.samples)
           }
+
+          if metricset.tags?
+            payload[:tags] = mixed_object(metricset.tags)
+          end
+
+          if metricset.transaction
+            payload[:transaction] = metricset.transaction
+          end
+
+          if metricset.span
+            payload[:span] = metricset.span
+          end
+
+          { metricset: payload }
         end
 
         private

data/lib/atatus/transport/serializers/span_serializer.rb

@@ -1,3 +1,20 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
 # frozen_string_literal: true
 
 module Atatus
@@ -13,12 +30,11 @@ module Atatus
 
         attr_reader :context_serializer
 
-        # rubocop:disable Metrics/MethodLength
         def build(span)
           {
             span: {
               id: span.id,
-              transaction_id: span.transaction_id,
+              transaction_id: span.transaction.id,
               parent_id: span.parent_id,
               name: keyword_field(span.name),
               type: join_type(span),
@@ -26,45 +42,64 @@ module Atatus
               context: context_serializer.build(span.context),
               stacktrace: span.stacktrace.to_a,
               timestamp: span.timestamp,
-              trace_id: span.trace_id
+              trace_id: span.trace_id,
+              sample_rate: span.sample_rate
             }
           }
         end
-        # rubocop:enable Metrics/MethodLength
 
         # @api private
         class ContextSerializer < Serializer
+          # rubocop:disable Metrics/CyclomaticComplexity
           def build(context)
             return unless context
 
-            { sync: context.sync }.tap do |base|
-              base[:db] = build_db(context.db) if context.db
-              base[:http] = build_http(context.http) if context.http
+            base = {}
+
+            base[:tags] = mixed_object(context.labels) if context.labels.any?
+            base[:sync] = context.sync unless context.sync.nil?
+            base[:db] = build_db(context.db) if context.db
+            base[:http] = build_http(context.http) if context.http
+
+            if context.destination
+              base[:destination] = build_destination(context.destination)
             end
+
+            base
           end
+          # rubocop:enable Metrics/CyclomaticComplexity
 
           private
 
           def build_db(db)
-            return unless db
-
             {
               instance: db.instance,
               statement: Util.truncate(db.statement, max_length: 10_000),
               type: db.type,
-              user: db.user
+              user: db.user,
+              rows_affected: db.rows_affected
             }
           end
 
           def build_http(http)
-            return unless http
-
             {
               url: http.url,
               status_code: http.status_code.to_i,
               method: keyword_field(http.method)
             }
           end
+
+          def build_destination(destination)
+            {
+              service: {
+                name: keyword_field(destination.name),
+                resource: keyword_field(destination.resource),
+                type: keyword_field(destination.type)
+              },
+              address: keyword_field(destination.address),
+              port: destination.port
+            }
+          end
         end
 
         private