network_resiliency 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b75b4aad26c41942f4d38ad70abb82d8f098eb377e9e5f44d1f6dde41ab6225
-  data.tar.gz: 47c899a650d66cea1117743ce4b53ee7bc50cc4fef8dfca3337aa071d0f81964
+  metadata.gz: b26da5cafa1321e4b284d8b3ade1184d4423973f388b1e4288b3be3d79153c3b
+  data.tar.gz: 6a4588995b8fdab21e424235bb31d2d83bef8d6b43d2b4ab1f9f3e9df08e19cf
 SHA512:
-  metadata.gz: ef7d805366f84daf7a27f5fb6dc88ef846741456203fa23f30512d780ae46065b5519caa474e7750b260e2a9bca5b9e296cb4def7e60402cfabdf0ae4aa1bf7c
-  data.tar.gz: 11d564818a9700a685c8d3dc9873de64181079fffffcbc0dd192d2ddbb47f0b75e43e92ba08d4523376a0a617ba60c9dd599e07fce1c055e48398b38f2e40ee3
+  metadata.gz: d6b110ad5631918d1077f4ba77a63da113fa9b19f156f439f6b46a2f8817d770740e157d07d7752b443568b3d1f86f416ff707d3ff0598a71b0355dabf1aa0c8
+  data.tar.gz: ab91f3ac1febb0504b8b0a64f0f12014636fd86bfc1ea3f1c1904a38660b8be791cc07835fed7154775d150bbc131243f4e892e255ecd7a96f68dbd7d1cd637a

data/CHANGELOG.md

@@ -1,3 +1,9 @@
+###  v0.2.0  (2023-10-01)
+- high level patch
+- redis adapter
+- mock redis
+- freeze time in specs
+
 ###  v0.1.1  (2023-09-27)
 - prevent double patching
 - filter ip addresses

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    network_resiliency (0.1.1)
+    network_resiliency (0.2.0)
 
 GEM
   remote: https://rubygems.org/

data/lib/network_resiliency/adapter/http.rb

@@ -6,6 +6,8 @@ module NetworkResiliency
       extend self
 
       def patch(instance = nil)
+        return if patched?(instance)
+
         (instance&.singleton_class || Net::HTTP).prepend(Instrumentation)
       end
 

data/lib/network_resiliency/adapter/redis.rb

@@ -3,61 +3,73 @@ require "redis"
 module NetworkResiliency
   module Adapter
     module Redis
-      def connect(redis_config)
-        puts "RedisClient.connect(#{client.connect_timeout})"
+      extend self
 
-        old_timeout = client.connect_timeout
+      def patch(instance = nil)
+        return if patched?(instance)
 
-        already_connected = client.connected?
+        if instance
+          unless instance.is_a?(::Redis)
+            raise ArgumentError, "expected Redis instance, found: #{instance}"
+          end
 
-        # client.connect_timeout = 0.1
-        ts = -NetworkResiliency.timestamp
-        super
-      ensure
-        client.connect_timeout = old_timeout
+          client = instance.instance_variable_get(:@client)
 
-        ts += NetworkResiliency.timestamp
+          unless client.is_a?(::Redis::Client)
+            raise ArgumentError, "unsupported Redis client: #{client}"
+          end
 
-        note = already_connected ? " (already_connected)" : ""
+          client.singleton_class.prepend(Instrumentation)
+        else
+          ::Redis::Client.prepend(Instrumentation)
+        end
+      end
+
+      def patched?(instance = nil)
+        if instance
+          client = instance.instance_variable_get(:@client)
 
-        puts "connect time: #{ts}#{note}"
+          client && client.singleton_class.ancestors.include?(Instrumentation)
+        else
+          ::Redis::Client.ancestors.include?(Instrumentation)
+        end
       end
 
-      def call(command, redis_config)
-        puts "RedisClient.call"
+      module Instrumentation
+        # def initialize(...)
+        #   super
 
-        key = "#{id}:call"
-        ts = -NetworkResiliency.timestamp
-        with_timeout(1) { super }
-      ensure
-        ts += NetworkResiliency.timestamp
-        puts "call time: #{ts}"
+        #   @network_resiliency_attempts = options[:reconnect_attempts]
+        #   options[:reconnect_attempts] = 0
+        # end
 
-        NetworkResiliency.record(self, id, ts)
-      end
+        def establish_connection
+          return super unless NetworkResiliency.enabled?(:redis)
 
-      def id
-        "redis:#{client.host}"
-      end
+          begin
+            ts = -NetworkResiliency.timestamp
 
-      def with_timeout(timeout)
-        old_timeouts = [
-          client.connect_timeout,
-          client.read_timeout,
-          client.write_timeout,
-        ]
+            super
+          rescue ::Redis::CannotConnectError => e
+            # capture error
+            raise
+          ensure
+            ts += NetworkResiliency.timestamp
 
-        client.connect_timeout = timeout
-        client.read_timeout = timeout
-        client.write_timeout = timeout
+            # grab underlying exception within Redis wrapper
+            error = e ? e.cause.class : nil
 
-        yield
-      ensure
-        client.connect_timeout, client.read_timeout, client.write_timeout = *old_timeouts
+            NetworkResiliency.record(
+              adapter: "redis",
+              action: "connect",
+              destination: host,
+              error: error,
+              duration: ts,
+            )
+          end
+        end
       end
     end
   end
 end
 
-# RedisClient.register(NetworkResiliency::Adapter::Redis)
-# Redis.new(middlewares: [NetworkResiliency::Adapter::Redis])

data/lib/network_resiliency/version.rb

@@ -1,3 +1,3 @@
 module NetworkResiliency
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/lib/network_resiliency.rb

@@ -15,10 +15,25 @@ module NetworkResiliency
     yield self
   end
 
+  def patch(*adapters)
+    adapters.each do |adapter|
+      case adapter
+      when :http
+        Adapter::HTTP.patch
+      when :redis
+        Adapter::Redis.patch
+      else
+        raise NotImplementedError
+      end
+    end
+  end
+
   def enabled?(adapter)
     return true if @enabled.nil?
 
-    @enabled.is_a?(Proc) ? @enabled.call(adapter) : @enabled
+    @enabled.is_a?(Proc) ? !!@enabled.call(adapter) : @enabled
+  rescue
+    false
   end
 
   def enabled=(enabled)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: network_resiliency
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Daniel Pepper
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-09-28 00:00:00.000000000 Z
+date: 2023-10-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: byebug