griffin-interceptors 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fe0fd2e86343914ca2f2f4a6716c415a2ac24936
-  data.tar.gz: 4932bb8474d59d681889c86a607eb1f1c76c1758
+  metadata.gz: eaf4ab1433928865a32bd22d19c1283f826ea526
+  data.tar.gz: 43d9b6c7f9a636fa79ae6e4adf8800c0c5326952
 SHA512:
-  metadata.gz: 1f2da5faf64741eaf1455fc209443fb5c2a99310fb39a23d27f949024d74b005692e027262d5b3a69d0b56be57b5f66350113b4810a899881a6697388bb0fb44
-  data.tar.gz: 3a726be452970dffe6a7b0bb9117069352ae4d4a5e3d5a1442669a492518855fd28f04f729b63d4da46a4afa7de4c0796f108be59a1b6f96dca1048f23b49389
+  metadata.gz: 57917963336f2b3ba83d84da9c3c59d2c2ac3b3352317375307facadd00b55edc8d415c72c3015bfc4d70295fc1bb06211d4f0c17b382666dc3bd323514f0e6f
+  data.tar.gz: 6f198f8b51e9a9afd723cc6d285307685e226941af9ed51f67a465a9e193c4853a1c21fbd3bf834db08f7bd8a1682cc96cae0e21cc74643728c8e6cf4fd5ae6e

data/griffin-interceptors.gemspec

@@ -20,7 +20,8 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
 
-  spec.add_dependency 'griffin', '>= 0.1.5'
+  spec.add_dependency 'get_process_mem', '~> 0.2.3'
+  spec.add_dependency 'griffin', '>= 0.2.3'
 
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake'

data/lib/griffin/interceptors/server/worker_killer_interceptor.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require 'get_process_mem'
+
+module Griffin
+  module Interceptors
+    module Server
+      class WorkerKillerInterceptor < GRPC::ServerInterceptor
+        def initialize(memory_limit_min: 1024**3, memory_limit_max: 2 * (1024**3), check_cycle: 16)
+          @worker_memory_limit_max = memory_limit_max
+          @worker_memory_limit_min = memory_limit_min
+          @worker_check_cycle = check_cycle
+          @worker_memory_limit = @worker_memory_limit_min + rand(@worker_memory_limit_max - @worker_memory_limit_min + 1)
+          @worker_check_count = 0
+          @sent_signals = false
+        end
+
+        def request_response(*)
+          yield.tap do
+            break if @sent_signals
+
+            @worker_process_start ||= Time.now
+            @worker_check_count += 1
+
+            if (@worker_check_count % @worker_check_cycle) == 0
+              rss = GetProcessMem.new.bytes
+              if rss > @worker_memory_limit
+                Griffin.logger.warn("Worker (pid: #{Process.pid}) exceeds memory limit (#{rss.to_i} bytes > #{@worker_memory_limit} bytes)")
+                send_restart_signal(@worker_process_start)
+              end
+
+              @worker_check_count = 0
+            end
+          end
+        end
+
+        alias_method :server_streamer, :request_response
+        alias_method :client_streamer, :request_response
+        alias_method :bidi_streamer, :request_response
+
+        private
+
+        def send_restart_signal(start_time)
+          # No need to get a lock, sending signals duplicately is acceptable
+          @sent_signals = true
+          pid = Process.pid
+          alive_sec = (Time.now - start_time).round
+
+          Griffin.logger.info("Send signal(USR1) pid: #{pid}, alive_sec: #{alive_sec}")
+          Process.kill :USR1, pid
+        end
+      end
+    end
+  end
+end

data/lib/griffin/interceptors/version.rb

@@ -1,5 +1,5 @@
 module Griffin
   module Interceptors
-    VERSION = '0.1.4'.freeze
+    VERSION = '0.1.5'.freeze
   end
 end

metadata

@@ -1,29 +1,43 @@
 --- !ruby/object:Gem::Specification
 name: griffin-interceptors
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Yuta Iwama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-01-10 00:00:00.000000000 Z
+date: 2019-02-12 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: get_process_mem
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.3
 - !ruby/object:Gem::Dependency
   name: griffin
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.1.5
+        version: 0.2.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.1.5
+        version: 0.2.3
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -112,6 +126,7 @@ files:
 - lib/griffin/interceptors/server/rails_exception_interceptor.rb
 - lib/griffin/interceptors/server/raven_interceptor.rb
 - lib/griffin/interceptors/server/timeout_interceptor.rb
+- lib/griffin/interceptors/server/worker_killer_interceptor.rb
 - lib/griffin/interceptors/server/x_request_id_interceptor.rb
 - lib/griffin/interceptors/version.rb
 homepage: https://github.com/cookpad/griffin-interceptors