func SpawnWorkers(cc ClientConstructor, p TestParameters) (*result.Result, *Sync) {
var addr, rate, duration, clients = p.Addr, p.Rate, p.Duration, p.Clients
log.Printf("Hitting %s at %f reqs/s by %d clients during %v.", addr, rate, clients, duration)
globalSync = NewSync(clients)
batches := clients / 1024
globalResult := result.NewResult(duration, rate)
thread_rate := float64(rate) / float64(clients)
go printStarted()
start_threads := func(n_threads int, rate float64) {
clientTP := TestParameters{Duration: duration, Rate: rate, Addr: addr, Clients: 1, SaveSamples: p.SaveSamples, SampleFile: p.SampleFile}
for i := 0; i < n_threads; i++ {
go client(cc, clientTP, &globalResult)
}
return
}
for i := 0; i < batches; i++ {
start_threads(1024, thread_rate)
time.Sleep(time.Duration(100 * 1000000)) //100 millisecs
}
start_threads(clients%1024, thread_rate)
return &globalResult, &globalSync
}
func client(cc ClientConstructor, tp TestParameters, globalResult *result.Result) {
// var addr string, rate float64, duration time.Duration = tp.Addr, tp.Rate, tp.Duration
addr, rate, duration := tp.Addr, tp.Rate, tp.Duration
var err error
localResult := result.NewResult(duration, rate)
client := cc()
buffer := make([]byte, 65536, 65536)
// conn, err := net.DialTimeout("tcp", addr[7:len(addr)-1], time.Duration(300* 1000000000))
// checkerr(err)
succeeded := client.Call(addr)
if !succeeded {
fmt.Printf("Couldn't dial %s\n", addr)
globalSync.signalDone()
return
}
// checkerr(err)
// log.Printf("Client target rate %v\n", rate)
loop_time := time.Duration(50000)
random_generator := rand.New(rand.NewSource(time.Now().UnixNano()))
next_deadline := time_to_deadline(random_generator, rate) - loop_time
globalSync.signalReady() //ready.Done()
globalSync.awaitGo() // start.Wait()
start_time := time.Now()
t_read := start_time
t_last_read := t_read
had_error := false
var body []byte
var saveBody = tp.SaveSamples
var emptyString = ""
for { // time.Since(start_time) < duration
if start_time.Add(duration).Before(t_read.Add(next_deadline)) {
// fmt.Printf("Loop time: %v\n", loop_time)
mt.Lock()
globalResult.CombineWith(localResult)
mt.Unlock()
// log.Printf("Got %d responses in %v (%v), Rate: %v\n", localResult.N_latencySamples, time.Since(start_time), localResult.TotLatency, float64(localResult.N_latencySamples)/duration.Seconds())
globalSync.signalDone() // wg.Done()
time.Sleep(start_time.Add(duration).Sub(t_read))
client.Close()
break
}
time.Sleep(next_deadline)
if false || had_error { //!*keepAlive
log.Println("Opening new connection")
// conn.Close()
client.Close()
client.Call(addr)
// conn, err = net.DialTimeout("tcp", addr[7:len(addr)-1], time.Duration(30* 1000000000))
// checkerr(err)
had_error = false
// reader = bufio.NewReader(conn)
}
t_start := time.Now()
var t_stamp time.Time
// var body []byte
if saveBody {
body, t_stamp, err = client.Request(buffer)
} else {
_, t_stamp, err = client.Request(buffer)
}
// t_returned := time.Now()
// _, err = clients.Req_tcp( conn, buffer, true)//*keepAlive
if err != nil {
log.Println("Read ERROR:", err)
// log.Println("Read ERROR:", err, "Response: ", body)
localResult.N_errors++
had_error = true
// wg.Done()
// return
continue
}
t_stamp = t_start // make timing to be before writing of request. correct?
// fmt.Printf("request body")
t_last_read = t_read
t_read = time.Now()
time_spent := t_read.Sub(t_stamp)
// req_time += time_spent
// localResult.AddSample(time_spent)
if saveBody {
localResult.AddSample(result.Sample{Latency: time_spent, IssueTime: t_stamp, Response: string(body)})
} else {
localResult.AddSample(result.Sample{Latency: time_spent, IssueTime: t_start, Response: emptyString})
}
//localResult.AddSample(result.Sample{Latency: time_spent, IssueTime: t_start, Response: string(body)})
this_loop := t_read.Sub(t_last_read) - time_spent - next_deadline
loop_time = time.Duration(int64(0.95*float64(loop_time.Nanoseconds()) + 0.05*float64(this_loop.Nanoseconds())))
next_deadline = time_to_deadline(random_generator, rate) - time_spent - loop_time
//log.Printf("Got response in %v, %v \n", t_read.Sub(t_start), t_int.Sub(t_start))
// log.Printf("Got response in %v, %v \n%s", t_read.Sub(t_start), t_int.Sub(t_start), body)
}
// clients.Req_tcp( conn, buffer, false)
// conn.Close()
}
