func handle_client(conn net.Conn, cc core.ClientConstructor) { reader := bufio.NewReader(conn) dec := json.NewDecoder(reader) enc := json.NewEncoder(conn) var p core.TestParameters if err := dec.Decode(&p); err != nil { log.Println(err) return } log.Println("Received test parameters from commander.") // log.Printf("Hitting %s at %f reqs/s by %d clients during %v.", p.Addr, p.Rate, p.Clients, p.Duration) globalResult, sync := core.SpawnWorkers(cc, p) sync.WaitReady() // ready.Wait(); conn.Write([]byte("DONE\n")) log.Println("Sent DONE notification to commander.") line, err := reader.ReadString('\n') if err != nil || line != "GO\n" { log.Print("Did not receive GO command:", line, err, "\n") return } log.Println("Received go command from commander, starting loading") sync.Go() // start.Done() sync.WaitDone() //wg.Wait() summary := result.ResultSummary{p.Clients, globalResult.AverageThroughput(), globalResult.N_errors} enc.Encode(summary) result.PrintResult(*globalResult, p.Clients) if p.SaveSamples { result.SaveToFile(globalResult, generateHostDateFileName(p.SampleFile)) } log.Printf("Send result: %v\n", summary) conn.Close() log.Println("Closed connection to master.\n") }
func main() { flag.Parse() if *cpuprofile != "" { f, err := os.Create(*cpuprofile) if err != nil { log.Fatal(err) } pprof.StartCPUProfile(f) defer pprof.StopCPUProfile() } rand.Seed(time.Now().Unix()) // numprocs, numprocsdef := 0, 0 hw_threads := runtime.NumCPU() // if *threads < hw_threads { // hw_threads = *threads // } // numprocsdef = runtime.GOMAXPROCS(hw_threads) // numprocsdef = runtime.GOMAXPROCS(4) numprocs := runtime.GOMAXPROCS(0) if *server { distr.StartServer(clientType) return } log.Printf("====================================================\n") hostname, _ := os.Hostname() log.Printf("Starting go load generator with %d HW threads at %s \n", numprocs, hostname) duration, err := time.ParseDuration(*p_duration) if err != nil { panic(err) } // log.Printf("Hitting %s at %d reqs/s by %d clients during %v.", *p_addr, *p_rate, *threads, duration) parameters := core.TestParameters{Addr: *p_addr, Rate: float64(*p_rate), Duration: duration, Clients: *threads, SaveSamples: (*dataOutput != ""), SampleFile: *dataOutput} var globalResult result.Result if *commander { serverfilecontents, err := ioutil.ReadFile(*serverfile) if err != nil { panic(err) } lines := strings.Split(string(serverfilecontents), "\n") var servers []string for _, line := range lines { if strings.TrimSpace(line) != "" { servers = append(servers, line) } } log.Printf("Using serverfile %s with contents (%d) %v\n", *serverfile, len(servers), servers) globalResult = distr.StartCommander(parameters, servers, clientType) } else { globalResult = core.StartTest(parameters, clientType) result.PrintResult(globalResult, *threads) if parameters.SaveSamples { result.SaveToFile(&globalResult, parameters.SampleFile) } } // if totalResponses > 0 // log.Printf("TOTAL %d responses in %v, Rate: %v, avgLatency: %v\n", totalResponses, duration, float64(totalResponses)/duration.Seconds(), time.Duration(avgLatency)) }