84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
package go_benchmark
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"gitea.urkob.com/urko/crono"
|
|
)
|
|
|
|
func doRequest(wg *sync.WaitGroup, host, employeeID string, errChan chan error) error {
|
|
defer wg.Done()
|
|
url := host + "/employee_wi/" + employeeID
|
|
client := &http.Client{}
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
|
|
if err != nil {
|
|
err = fmt.Errorf("http.NewRequest: %s", err)
|
|
errChan <- err
|
|
return err
|
|
}
|
|
req.Header.Add("Cache-Control", "no-cache")
|
|
req.Header.Add("Content-Type", "application/json")
|
|
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
err = fmt.Errorf("client.Do: %s", err)
|
|
errChan <- err
|
|
return err
|
|
}
|
|
defer res.Body.Close()
|
|
|
|
readedBody, err := io.ReadAll(res.Body)
|
|
if err != nil {
|
|
err = fmt.Errorf("ioutil.ReadAll: %s", err)
|
|
errChan <- err
|
|
return err
|
|
}
|
|
|
|
log.Println("readed", string(readedBody))
|
|
|
|
return nil
|
|
}
|
|
|
|
func BenchmarkNoLog(host, employeeID string, totalRequests int) error {
|
|
return benchmark(host, employeeID, totalRequests, false)
|
|
}
|
|
func Benchmark(host, employeeID string, totalRequests int) error {
|
|
return benchmark(host, employeeID, totalRequests, true)
|
|
}
|
|
|
|
func benchmark(host, employeeID string, totalRequests int, dolog bool) error {
|
|
cr := crono.New()
|
|
if dolog {
|
|
defer cr.Table()
|
|
}
|
|
|
|
wg := &sync.WaitGroup{}
|
|
errChan := make(chan error, 1)
|
|
wg.Add(totalRequests + 1)
|
|
go func(errChan chan error) {
|
|
defer wg.Done()
|
|
defer func() {
|
|
errChan <- nil
|
|
}()
|
|
for i := 0; i < totalRequests; i++ {
|
|
go doRequest(wg, host, employeeID, errChan)
|
|
}
|
|
}(errChan)
|
|
|
|
wg.Wait()
|
|
|
|
if err := <-errChan; err != nil {
|
|
return fmt.Errorf("Benchmark: %s", err)
|
|
}
|
|
if dolog {
|
|
cr.MarkAndRestart(fmt.Sprintf("go handle %d response", totalRequests))
|
|
}
|
|
|
|
return nil
|
|
}
|