瀏覽代碼

Support secure client

vjpai 9 年之前
父節點
當前提交
d9389a849d
共有 1 個文件被更改,包括 5 次插入5 次删除
  1. 5 5
      src/ruby/qps/client.rb

+ 5 - 5
src/ruby/qps/client.rb

@@ -57,17 +57,17 @@ end
 
 
 class BenchmarkClient
 class BenchmarkClient
   def initialize(config)
   def initialize(config)
+    opts = {}
     if config.security_params
     if config.security_params
       if config.security_params.use_test_ca
       if config.security_params.use_test_ca
         certs = load_test_certs
         certs = load_test_certs
         cred = GRPC::Core::ChannelCredentials.new(certs[0])
         cred = GRPC::Core::ChannelCredentials.new(certs[0])
       else
       else
-        p 'Unsupported to use non-test CA (TBD)'
-        exit
+        cred = GRPC::Core::ChannelCredentials.new()
       end
       end
       if config.security_params.server_host_override
       if config.security_params.server_host_override
-        p 'Unsupported to use server host override (TBD)'
-        exit
+        opts[GRPC::Core::Channel::SSL_TARGET] =
+          config.security_params.server_host_override
       end
       end
     else
     else
       cred = :this_channel_is_insecure
       cred = :this_channel_is_insecure
@@ -96,7 +96,7 @@ class BenchmarkClient
         end
         end
         gtbss = Grpc::Testing::BenchmarkService::Stub
         gtbss = Grpc::Testing::BenchmarkService::Stub
         st = config.server_targets
         st = config.server_targets
-        stub = gtbss.new(st[i % st.length], cred)
+        stub = gtbss.new(st[i % st.length], cred, **opts)
         case config.rpc_type
         case config.rpc_type
         when :UNARY
         when :UNARY
           unary_ping_ponger(req,stub,config,waiter)
           unary_ping_ponger(req,stub,config,waiter)