Module: TShield::Grpc::VCR
- Included in:
- RequestHandler
- Defined in:
- lib/tshield/grpc/vcr.rb
Overview
Grpc vcr module
Defined Under Namespace
Classes: FilePath
Instance Method Summary collapse
- #complete_path(module_name, method_name) ⇒ Object
- #create_destiny(module_name, method_name) ⇒ Object
- #encode_colon(value) ⇒ Object
- #handler_in_vcr_mode(method_name, request, parameters, options) ⇒ Object
- #hexdigest(request) ⇒ Object
- #initialize ⇒ Object
- #request_count ⇒ Object
- #save_error(error) ⇒ Object
- #save_request(request) ⇒ Object
- #save_response(response) ⇒ Object
- #saved_error ⇒ Object
- #saved_response ⇒ Object
- #send_request(request, module_name, options, method_name) ⇒ Object
Instance Method Details
#complete_path(module_name, method_name) ⇒ Object
116 117 118 119 120 |
# File 'lib/tshield/grpc/vcr.rb', line 116 def complete_path(module_name, method_name) @session_name = (@session || {})[:name] module_name = @configuration.windows_compatibility? ? encode_colon(module_name) : module_name ['requests', @session_name, module_name, method_name.to_s, @digest].compact end |
#create_destiny(module_name, method_name) ⇒ Object
122 123 124 125 126 127 128 129 130 131 132 133 |
# File 'lib/tshield/grpc/vcr.rb', line 122 def create_destiny(module_name, method_name) current_path = [] path = complete_path(module_name, method_name) TShield.logger.info("using path #{path}") path.each do |inner_path| current_path << inner_path destiny = File.join current_path Dir.mkdir destiny unless File.exist? destiny end path end |
#encode_colon(value) ⇒ Object
70 71 72 |
# File 'lib/tshield/grpc/vcr.rb', line 70 def encode_colon(value) value.gsub(':', '%3a') end |
#handler_in_vcr_mode(method_name, request, parameters, options) ⇒ Object
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
# File 'lib/tshield/grpc/vcr.rb', line 24 def handler_in_vcr_mode(method_name, request, parameters, ) parameters.peer =~ /ipv6:\[(.+?)\]|ipv4:(.+?):/ peer = Regexp.last_match(1) || Regexp.last_match(2) TShield.logger.info("request from #{parameters.peer}") @session = TShield::Sessions.current(peer) @digest = hexdigest(request) counter = @session ? request_count.current(@digest) : 0 TShield.logger.info("grpc using session #{@session || 'default'}") module_name = ['module'] path = create_destiny(module_name, method_name) @file_path = FilePath.new(path, counter) save_request(request) response = {} saved_error begin response = saved_response if response TShield.logger.info("returning saved response for request #{request.to_json} saved into #{@digest}") request_count.add(@digest) if @session return response end response = send_request(request, module_name, , method_name) save_response(response) rescue GRPC::BadStatus => e save_error({ code: e.code, details: e.details }) raise e end request_count.add(@digest) if @session response end |
#hexdigest(request) ⇒ Object
135 136 137 |
# File 'lib/tshield/grpc/vcr.rb', line 135 def hexdigest(request) Digest::SHA1.hexdigest request.to_json end |
#initialize ⇒ Object
20 21 22 |
# File 'lib/tshield/grpc/vcr.rb', line 20 def initialize @configuration = TShield::Configuration.singleton end |
#request_count ⇒ Object
66 67 68 |
# File 'lib/tshield/grpc/vcr.rb', line 66 def request_count @session[:grpc_counter] end |
#save_error(error) ⇒ Object
99 100 101 102 103 104 |
# File 'lib/tshield/grpc/vcr.rb', line 99 def save_error(error) file = File.open(File.join(@file_path.path, "#{@file_path.count}.error"), 'w') file.puts error.to_json file.close request_count.add(@digest) if @session end |
#save_request(request) ⇒ Object
93 94 95 96 97 |
# File 'lib/tshield/grpc/vcr.rb', line 93 def save_request(request) file = File.open(File.join(@file_path.path, "#{@file_path.count}.original_request"), 'w') file.puts request.to_json file.close end |
#save_response(response) ⇒ Object
106 107 108 109 110 111 112 113 114 |
# File 'lib/tshield/grpc/vcr.rb', line 106 def save_response(response) file = File.open(File.join(@file_path.path, "#{@file_path.count}.response"), 'w') file.puts response.to_json file.close response_class = File.open(File.join(@file_path.path, "#{@file_path.count}.response_class"), 'w') response_class.puts response.class.to_s response_class.close end |
#saved_error ⇒ Object
83 84 85 86 87 88 89 90 91 |
# File 'lib/tshield/grpc/vcr.rb', line 83 def saved_error error_file = File.join(@file_path.path, "#{@file_path.count}.error") return false unless File.exist? error_file request_count.add(@digest) if @session content = JSON.parse File.open(error_file).read grpc_error = GRPC::BadStatus.new(content['code'], content['details']) raise grpc_error end |
#saved_response ⇒ Object
74 75 76 77 78 79 80 81 |
# File 'lib/tshield/grpc/vcr.rb', line 74 def saved_response response_file = File.join(@file_path.path, "#{@file_path.count}.response") return false unless File.exist? response_file content = JSON.parse File.open(response_file).read response_class = File.open(File.join(@file_path.path, "#{@file_path.count}.response_class")).read.strip Kernel.const_get(response_class).new(content) end |
#send_request(request, module_name, options, method_name) ⇒ Object
59 60 61 62 63 64 |
# File 'lib/tshield/grpc/vcr.rb', line 59 def send_request(request, module_name, , method_name) TShield.logger.info("calling server to get response for #{request.to_json}") client_class = Object.const_get("#{module_name}::Stub") client_instance = client_class.new(['hostname'], :this_channel_is_insecure) client_instance.send(method_name, request) end |