opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add distribution variable to be used by open loop

Browse Source
This commit is contained in:
Menghan Li
2016-04-19 15:50:32 -07:00
parent b5e39adfb2
commit fc76af457f
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
benchmark/worker/benchmark_client.go
View File

@ -114,17 +114,24 @@ func startBenchmarkClientWithSetup(setup *testpb.ClientConfig) (*benchmarkClient
}) })
grpclog.Printf(" - load params: %v", setup.LoadParams) grpclog.Printf(" - load params: %v", setup.LoadParams)
// TODO distribution
var dist *int
switch lp := setup.LoadParams.Load.(type) { switch lp := setup.LoadParams.Load.(type) {
case *testpb.LoadParams_ClosedLoop: case *testpb.LoadParams_ClosedLoop:
grpclog.Printf(" - %v", lp.ClosedLoop) grpclog.Printf(" - %v", lp.ClosedLoop)
case *testpb.LoadParams_Poisson: case *testpb.LoadParams_Poisson:
grpclog.Printf(" - %v", lp.Poisson) grpclog.Printf(" - %v", lp.Poisson)
return nil, grpc.Errorf(codes.InvalidArgument, "unsupported load params: %v", setup.LoadParams)
// TODO poisson
case *testpb.LoadParams_Uniform: case *testpb.LoadParams_Uniform:
grpclog.Printf(" - %v", lp.Uniform) grpclog.Printf(" - %v", lp.Uniform)
return nil, grpc.Errorf(codes.InvalidArgument, "unsupported load params: %v", setup.LoadParams)
case *testpb.LoadParams_Determ: case *testpb.LoadParams_Determ:
grpclog.Printf(" - %v", lp.Determ) grpclog.Printf(" - %v", lp.Determ)
return nil, grpc.Errorf(codes.InvalidArgument, "unsupported load params: %v", setup.LoadParams)
case *testpb.LoadParams_Pareto: case *testpb.LoadParams_Pareto:
grpclog.Printf(" - %v", lp.Pareto) grpclog.Printf(" - %v", lp.Pareto)
return nil, grpc.Errorf(codes.InvalidArgument, "unsupported load params: %v", setup.LoadParams)
default: default:
return nil, grpc.Errorf(codes.InvalidArgument, "unknown load params: %v", setup.LoadParams) return nil, grpc.Errorf(codes.InvalidArgument, "unknown load params: %v", setup.LoadParams)
} }
@ -133,9 +140,15 @@ func startBenchmarkClientWithSetup(setup *testpb.ClientConfig) (*benchmarkClient
bc.stop = make(chan bool) bc.stop = make(chan bool)
switch setup.RpcType { switch setup.RpcType {
case testpb.RpcType_UNARY: case testpb.RpcType_UNARY:
doCloseLoopUnaryBenchmark(bc.histogram, bc.conns, rpcCount, payloadReqSize, payloadRespSize, bc.stop) if dist == nil {
doCloseLoopUnaryBenchmark(bc.histogram, bc.conns, rpcCount, payloadReqSize, payloadRespSize, bc.stop)
}
// TODO else do open loop
case testpb.RpcType_STREAMING: case testpb.RpcType_STREAMING:
doCloseLoopStreamingBenchmark(bc.histogram, bc.conns, rpcCount, payloadReqSize, payloadRespSize, payloadType, bc.stop) if dist == nil {
doCloseLoopStreamingBenchmark(bc.histogram, bc.conns, rpcCount, payloadReqSize, payloadRespSize, payloadType, bc.stop)
}
// TODO else do open loop
default: default:
return nil, grpc.Errorf(codes.InvalidArgument, "unknown rpc type: %v", setup.RpcType) return nil, grpc.Errorf(codes.InvalidArgument, "unknown rpc type: %v", setup.RpcType)
} }