예제 #1
0
파일: main.go 프로젝트: xqbumu/learn
func main() {
	switch opt {

	case "grpc":
		demogrpc.Stress(port, endpoint, keys, vals, connsN, clientsN)

	case "jsonrpc":
		demojsonrpc.Stress(port, endpoint, keys, vals)

	}
}
예제 #2
0
파일: bench_test.go 프로젝트: xqbumu/learn
func BenchmarkStress(b *testing.B) {
	b.StartTimer()
	b.ReportAllocs()

	oncebody := func() {
		switch opt {
		case "grpc":
			port := ":8000"
			endpoint := "localhost" + port
			demogrpc.Stress(port, endpoint, keys, vals, connsN, clientsN)
		case "jsonrpc":
			port := ":8001"
			endpoint := "localhost" + port
			demojsonrpc.Stress(port, endpoint, keys, vals)
		}
	}
	once.Do(oncebody)
}