1#!/usr/bin/env ruby 2 3# Copyright 2016 gRPC authors. 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17# Worker and worker service implementation 18 19this_dir = File.expand_path(File.dirname(__FILE__)) 20lib_dir = File.join(File.dirname(this_dir), 'lib') 21$LOAD_PATH.unshift(lib_dir) unless $LOAD_PATH.include?(lib_dir) 22$LOAD_PATH.unshift(this_dir) unless $LOAD_PATH.include?(this_dir) 23 24require 'grpc' 25 26# produces a string of null chars (\0 aka pack 'x') of length l. 27def nulls(l) 28 fail 'requires #{l} to be +ve' if l < 0 29 [].pack('x' * l).force_encoding('ascii-8bit') 30end 31 32# load the test-only certificates 33def load_test_certs 34 this_dir = File.expand_path(File.dirname(__FILE__)) 35 data_dir = File.join(File.dirname(this_dir), 'spec/testdata') 36 files = ['ca.pem', 'server1.key', 'server1.pem'] 37 files.map { |f| File.open(File.join(data_dir, f)).read } 38end 39 40 41# A EnumeratorQueue wraps a Queue yielding the items added to it via each_item. 42class EnumeratorQueue 43 extend Forwardable 44 def_delegators :@q, :push 45 46 def initialize(sentinel) 47 @q = Queue.new 48 @sentinel = sentinel 49 end 50 51 def each_item 52 return enum_for(:each_item) unless block_given? 53 loop do 54 r = @q.pop 55 break if r.equal?(@sentinel) 56 fail r if r.is_a? Exception 57 yield r 58 end 59 end 60end 61 62# A PingPongEnumerator reads requests and responds one-by-one when enumerated 63# via #each_item 64class PingPongEnumerator 65 def initialize(reqs) 66 @reqs = reqs 67 end 68 69 def each_item 70 return enum_for(:each_item) unless block_given? 71 sr = Grpc::Testing::SimpleResponse 72 pl = Grpc::Testing::Payload 73 @reqs.each do |req| 74 yield sr.new(payload: pl.new(body: nulls(req.response_size))) 75 end 76 end 77end 78