grpc
/
grpc-go
mirror of https://github.com/grpc/grpc-go.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change cpu core set and var declaration

This commit is contained in:
Menghan Li 2016-04-28 16:43:23 -07:00
parent 3a13913bba
commit bb1be7190b
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
benchmark/worker/benchmark_client.go
View File

@ -91,15 +91,16 @@ func startBenchmarkClient(config *testpb.ClientConfig) (*benchmarkClient, error)
grpclog.Printf(" - core limit: %v", config.CoreLimit) grpclog.Printf(" - core limit: %v", config.CoreLimit)
// Use one cpu core by default // Use one cpu core by default
numOfCores := 1 // TODO: change default number of cores used if 1 is not fastest.
if config.CoreLimit > 0 { if config.CoreLimit > 1 {
numOfCores = int(config.CoreLimit) runtime.GOMAXPROCS(int(config.CoreLimit))
} }
runtime.GOMAXPROCS(numOfCores)
grpclog.Printf(" - payload config: %v", config.PayloadConfig) grpclog.Printf(" - payload config: %v", config.PayloadConfig)
var payloadReqSize, payloadRespSize int var (
var payloadType string payloadReqSize, payloadRespSize int
payloadType string
)
if config.PayloadConfig != nil { if config.PayloadConfig != nil {
switch c := config.PayloadConfig.Payload.(type) { switch c := config.PayloadConfig.Payload.(type) {
case *testpb.PayloadConfig_BytebufParams: case *testpb.PayloadConfig_BytebufParams:

8
benchmark/worker/benchmark_server.go
View File

@ -93,7 +93,7 @@ func startBenchmarkServer(config *testpb.ServerConfig, serverPort int) (*benchma
grpclog.Printf(" - core limit: %v", config.CoreLimit) grpclog.Printf(" - core limit: %v", config.CoreLimit)
// Use one cpu core by default. // Use one cpu core by default.
numOfCores := 1 numOfCores := 1
if config.CoreLimit > 0 { if config.CoreLimit > 1 {
numOfCores = int(config.CoreLimit) numOfCores = int(config.CoreLimit)
} }
runtime.GOMAXPROCS(numOfCores) runtime.GOMAXPROCS(numOfCores)
@ -108,8 +108,10 @@ func startBenchmarkServer(config *testpb.ServerConfig, serverPort int) (*benchma
} }
grpclog.Printf(" - payload config: %v", config.PayloadConfig) grpclog.Printf(" - payload config: %v", config.PayloadConfig)
var addr string var (
var close func() addr string
close func()
)
if config.PayloadConfig != nil { if config.PayloadConfig != nil {
switch payload := config.PayloadConfig.Payload.(type) { switch payload := config.PayloadConfig.Payload.(type) {
case *testpb.PayloadConfig_BytebufParams: case *testpb.PayloadConfig_BytebufParams: