pifpaf.go

package main

import (
	"fmt"
	"sync"
	"sync/atomic"
	"time"

	"github.com/valyala/fasthttp"
)

type benchResult [][]BenchResult

func pifpaf(i int, benchResultsAll *benchResult, client *fasthttp.Client, enough, queries *int64, wg *sync.WaitGroup) {
	defer wg.Done()

	var myQueries int64

	testRun := argv.testRun

	uri := []byte(argv.serverAddr)
	uriBase := len(uri)

	for atomic.LoadInt64(enough) == 0 {
		for bulletIdx, bullet := range bullets {
			uri = append(uri[:uriBase], bullet.Request.URI...)

			req := fasthttp.AcquireRequest()
			req.SetRequestURIBytes(uri)
			for _, header := range bullet.Request.Headers {
				req.Header.SetBytesKV(header.Key, header.Value)
			}
			if len(bullet.Request.Body) > 0 {
				req.SetBody(bullet.Request.Body)
			}
			if !bullet.Request.IsGet {
				req.Header.SetMethod(`POST`)
			}

			resp := fasthttp.AcquireResponse()

			oneBenchResult := BenchResult{bulletIdx: bulletIdx}

			myQueries++

			tnow := time.Now()
			err := client.DoTimeout(req, resp, 2*time.Second)
			oneBenchResult.dur = time.Since(tnow)
			if err != nil {
				oneBenchResult.status = -1
				//fmt.Println(`client.DoTimeout fail:`, err)
			} else {
				oneBenchResult.status = resp.StatusCode()
				oneBenchResult.body = append(oneBenchResult.body, resp.Body()...)
			}
			(*benchResultsAll)[i] = append((*benchResultsAll)[i], oneBenchResult)

			fasthttp.ReleaseRequest(req)
			fasthttp.ReleaseResponse(resp)
		}

		if testRun {
			break
		}
	}

	atomic.AddInt64(queries, myQueries)
}

func pifpafTank(ii int, benchResultsAll *benchResult, client *fasthttp.Client, queries, currBullet *int64, wg *sync.WaitGroup) {
	defer wg.Done()

	var myQueries int64

	uri := []byte(argv.serverAddr)
	uriBase := len(uri)

	bulletIdx := atomic.AddInt64(currBullet, 1) % int64(len(bullets))

	muMaxReqNo.Lock()
	if maxReqNo < ii {
		fmt.Printf("\rBullet %d", ii)
		maxReqNo = ii
	}
	muMaxReqNo.Unlock()

	bullet := bullets[bulletIdx]
	uri = append(uri[:uriBase], bullet.Request.URI...)

	req := fasthttp.AcquireRequest()
	req.SetRequestURIBytes(uri)
	for _, header := range bullet.Request.Headers {
		req.Header.SetBytesKV(header.Key, header.Value)
	}
	if len(bullet.Request.Body) > 0 {
		req.SetBody(bullet.Request.Body)
	}
	if !bullet.Request.IsGet {
		req.Header.SetMethod(`POST`)
	}

	resp := fasthttp.AcquireResponse()

	oneBenchResult := BenchResult{bulletIdx: int(bulletIdx)}

	myQueries++

	tnow := time.Now()
	err := client.DoTimeout(req, resp, 2*time.Second)
	oneBenchResult.dur = time.Since(tnow)
	if err != nil {
		oneBenchResult.status = -1
		//fmt.Println(`client.DoTimeout fail:`, err)
	} else {
		oneBenchResult.status = resp.StatusCode()
		oneBenchResult.body = append(oneBenchResult.body, resp.Body()...)
	}
	(*benchResultsAll)[ii] = append((*benchResultsAll)[ii], oneBenchResult)

	fasthttp.ReleaseRequest(req)
	fasthttp.ReleaseResponse(resp)

	atomic.AddInt64(queries, myQueries)
}