vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 1 | #!/usr/bin/env ruby |
| 2 | |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | # Copyright 2016 gRPC authors. |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 4 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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 |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 8 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 10 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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. |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 16 | |
| 17 | # Worker and worker service implementation |
| 18 | |
| 19 | this_dir = File.expand_path(File.dirname(__FILE__)) |
| 20 | lib_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 | |
| 24 | require 'grpc' |
| 25 | require 'optparse' |
| 26 | require 'histogram' |
| 27 | require 'etc' |
| 28 | require 'facter' |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 29 | require 'client' |
| 30 | require 'qps-common' |
| 31 | require 'server' |
Ken Payson | 5a2c918 | 2016-07-26 17:15:08 -0700 | [diff] [blame] | 32 | require 'src/proto/grpc/testing/services_services_pb' |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 33 | |
| 34 | class WorkerServiceImpl < Grpc::Testing::WorkerService::Service |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 35 | def cpu_cores |
| 36 | Facter.value('processors')['count'] |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 37 | end |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 38 | def run_server(reqs) |
| 39 | q = EnumeratorQueue.new(self) |
| 40 | Thread.new { |
vjpai | e8ff388 | 2016-03-29 18:31:34 -0700 | [diff] [blame] | 41 | bms = '' |
vjpai | ed25a33 | 2016-03-30 12:16:25 -0700 | [diff] [blame] | 42 | gtss = Grpc::Testing::ServerStatus |
Jan Tattermusch | 3849335 | 2016-04-18 18:07:49 -0700 | [diff] [blame] | 43 | reqs.each do |req| |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 44 | case req.argtype.to_s |
| 45 | when 'setup' |
vjpai | e8ff388 | 2016-03-29 18:31:34 -0700 | [diff] [blame] | 46 | bms = BenchmarkServer.new(req.setup, @server_port) |
vjpai | ed25a33 | 2016-03-30 12:16:25 -0700 | [diff] [blame] | 47 | q.push(gtss.new(stats: bms.mark(false), port: bms.get_port)) |
Jan Tattermusch | 3849335 | 2016-04-18 18:07:49 -0700 | [diff] [blame] | 48 | when 'mark' |
vjpai | ed25a33 | 2016-03-30 12:16:25 -0700 | [diff] [blame] | 49 | q.push(gtss.new(stats: bms.mark(req.mark.reset), cores: cpu_cores)) |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 50 | end |
| 51 | end |
vjpai | e8ff388 | 2016-03-29 18:31:34 -0700 | [diff] [blame] | 52 | bms.stop |
Alexander Polcyn | 847f9ec | 2016-08-15 18:44:14 -0700 | [diff] [blame] | 53 | q.push(self) |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 54 | } |
| 55 | q.each_item |
| 56 | end |
| 57 | def run_client(reqs) |
| 58 | q = EnumeratorQueue.new(self) |
| 59 | Thread.new { |
vjpai | d7f43b3 | 2016-03-29 18:38:10 -0700 | [diff] [blame] | 60 | client = '' |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 61 | reqs.each do |req| |
vjpai | ad1c1cc | 2016-03-30 09:58:46 -0700 | [diff] [blame] | 62 | case req.argtype.to_s |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 63 | when 'setup' |
| 64 | client = BenchmarkClient.new(req.setup) |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 65 | q.push(Grpc::Testing::ClientStatus.new(stats: client.mark(false))) |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 66 | when 'mark' |
vjpai | ed25a33 | 2016-03-30 12:16:25 -0700 | [diff] [blame] | 67 | q.push(Grpc::Testing::ClientStatus.new(stats: |
| 68 | client.mark(req.mark.reset))) |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 69 | end |
| 70 | end |
vjpai | ad1c1cc | 2016-03-30 09:58:46 -0700 | [diff] [blame] | 71 | client.shutdown |
Alexander Polcyn | 847f9ec | 2016-08-15 18:44:14 -0700 | [diff] [blame] | 72 | q.push(self) |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 73 | } |
| 74 | q.each_item |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 75 | end |
| 76 | def core_count(_args, _call) |
vjpai | 75291c9 | 2016-03-21 07:59:26 -0700 | [diff] [blame] | 77 | Grpc::Testing::CoreResponse.new(cores: cpu_cores) |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 78 | end |
| 79 | def quit_worker(_args, _call) |
| 80 | Thread.new { |
| 81 | sleep 3 |
| 82 | @server.stop |
| 83 | } |
| 84 | Grpc::Testing::Void.new |
| 85 | end |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 86 | def initialize(s, sp) |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 87 | @server = s |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 88 | @server_port = sp |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 89 | end |
| 90 | end |
| 91 | |
| 92 | def main |
| 93 | options = { |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 94 | 'driver_port' => 0, |
| 95 | 'server_port' => 0 |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 96 | } |
| 97 | OptionParser.new do |opts| |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 98 | opts.banner = 'Usage: [--driver_port <port>] [--server_port <port>]' |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 99 | opts.on('--driver_port PORT', '<port>') do |v| |
| 100 | options['driver_port'] = v |
| 101 | end |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 102 | opts.on('--server_port PORT', '<port>') do |v| |
| 103 | options['server_port'] = v |
| 104 | end |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 105 | end.parse! |
Alexander Polcyn | 847f9ec | 2016-08-15 18:44:14 -0700 | [diff] [blame] | 106 | |
| 107 | # Configure any errors with client or server child threads to surface |
| 108 | Thread.abort_on_exception = true |
| 109 | |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 110 | s = GRPC::RpcServer.new |
| 111 | s.add_http2_port("0.0.0.0:" + options['driver_port'].to_s, |
| 112 | :this_port_is_insecure) |
vjpai | 45be26e | 2016-03-29 17:21:28 -0700 | [diff] [blame] | 113 | s.handle(WorkerServiceImpl.new(s, options['server_port'].to_i)) |
vjpai | fdeacd9 | 2016-03-17 14:52:20 -0700 | [diff] [blame] | 114 | s.run |
| 115 | end |
| 116 | |
| 117 | main |