iip/loadbalanceclient_test.go at master · truexf/iip · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
// Copyright 2021 fangyousong(方友松). All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package iip

import (
	"flag"
	"fmt"
	"github.com/truexf/goutil"
	"strings"
	"testing"
	"time"
)

var (
	pflbcIIPServerAddr = flag.String("lbcaddr", ":9090#1,:9090#1,:9090#1", "")
)

func BenchmarkPFIIPBalanceClient(b *testing.B) {
	LogClosing = false
	flag.Parse()
	lbc, err := NewLoadBalanceClient(100, 1000, *pflbcIIPServerAddr)
	if err != nil {
		b.Fatalf("new lbc fail,%s", err.Error())
	}

	echoData := `1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest
					1testtesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttesttest` + fmt.Sprintf("%d", time.Now().UnixNano())
	echoData = strings.Repeat(echoData, 10)
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			bts, err := lbc.DoRequest2("localhost", "/echo_benchmark", NewDefaultRequest([]byte(echoData)), time.Second*3)
			if err != nil {
				b.Fatalf(err.Error())
			} else {
				if goutil.UnsafeBytesToString(bts) != echoData {
					b.Fatalf("not equal")
				}
			}
		}
	})

}