uap-go/uaparser/benchmark_test.go at 76a10f6cf6a19f664212ec46301bd541b062615a · jkroepke/uap-go · 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package uaparser

import (
	"fmt"
	"log"
	"math/rand"
	"os"
	"strings"
	"testing"

	"gopkg.in/yaml.v3"
)

var benchedParser *Parser
var benchedParserWithOptions *Parser
var largeUasSample []string

func init() {
	var err error

	regexes, err := os.ReadFile("../uap-core/regexes.yaml")
	if err != nil {
		log.Fatal(err)
	}

	var def RegexDefinitions

	if err := yaml.Unmarshal(regexes, &def); err != nil {
		log.Fatal(err)
	}

	benchedParser, err = New(WithRegexDefinitions(def))
	if err != nil {
		log.Fatal(err)
	}
	benchedParserWithOptions, err = New(WithRegexDefinitions(def),
		WithMode(EOsLookUpMode|EUserAgentLookUpMode),
		WithMatchIdxNotOk(20),
		WithSort(true, WithMissesThreshold(100)),
		WithDebug(true),
		WithCacheSize(cDefaultCacheSize),
	)

	if err != nil {
		log.Fatal(err)
	}

	uasContent, err := os.ReadFile("../uas")
	if err != nil {
		log.Fatal(err)
	}
	largeUasSample = strings.Split(string(uasContent), "\n")
}

func BenchmarkParser(b *testing.B) {
	for i := 0; i < b.N; i++ {
		for _, ua := range uas {
			benchedParser.Parse(ua)
		}
	}
}

func BenchmarkParserWithOptions(b *testing.B) {
	for i := 0; i < b.N; i++ {
		for _, ua := range uas {
			benchedParserWithOptions.Parse(ua)
		}
	}
}

func BenchmarkParserWithDifferentCacheSize(b *testing.B) {
	sizes := []int{cDefaultCacheSize, cDefaultCacheSize * 2, cDefaultCacheSize * 3, cDefaultCacheSize * 4}

	regexes, err := os.ReadFile("../uap-core/regexes.yaml")
	if err != nil {
		log.Fatal(err)
	}

	var def RegexDefinitions

	if err := yaml.Unmarshal(regexes, &def); err != nil {
		log.Fatal(err)
	}

	for _, size := range sizes {
		parser, err := New(WithRegexDefinitions(def),
			WithMode(EOsLookUpMode|EUserAgentLookUpMode|EDeviceLookUpMode),
			WithMatchIdxNotOk(cDefaultMatchIdxNotOk),
			WithSort(false, WithMissesThreshold(cDefaultMissesTreshold)),
			WithDebug(false),
			WithCacheSize(size),
		)
		if err != nil {
			log.Fatal(err)
		}

		b.ResetTimer()
		b.Run(fmt.Sprintf("CacheSize=%d", size), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				index := rand.Intn(len(largeUasSample))
				parser.Parse(largeUasSample[index])
			}
		})
	}

}

var uas = []string{
	"Mozilla/5.0 (Windows NT 6.0; rv:2.0b6pre) Gecko/20100907 Firefox/4.0b6pre",
	"Mozilla/5.0 (Windows NT 5.2; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
	"Mozilla/5.0 (Windows NT 6.0; rv:2.0b6pre) Gecko/20100907 Firefox/4.0b6pre",
	"Mozilla/5.0 (Windows NT 5.2; rv:2.1.1) Gecko/ Firefox/5.0.1",
	"Mozilla/5.0 (Windows; U; Windows NT 6.1; fr; rv:1.9.2.8) Gecko/20100722 Firefox 3.6.8 GTB7.1",
	"Mozilla/5.0 (Windows NT 5.2; rv:2.0.1) Gecko Firefox/5.0.1",
	"Mozilla/5.0 (Android; Linux armv7l; rv:2.0b6pre) Gecko/20100907 Firefox/4.0b6pre Fennec/2.0b1pre",
	"Mozilla/5.0 (X11; U; Linux x86_64; en-US; rv:1.9.1.8) Gecko/20100408 Thunderbird/3.0.3",
	"Mozilla/5.0 (Macintosh; Intel Mac OS X 10.5; rv:2.0.1) Gecko/20100101 Firefox/4.0.1 SeaMonkey/2.1.1",
	"ACS-NF	3.0	ACS-NF/3.0 NEC-c616/001.00",
	"BlackBerry7730	3.7.1	BlackBerry7730/3.7.1 UP.Link/5.1.2.5",
	"Galeon	1.3.18	Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.7.3) Gecko/20041020 Galeon/1.3.18",
	"Opera/9.80 (J2ME/MIDP; Opera Mini/7.1.32052/35.5706; U; id) Presto/2.8.119 Version/11.10",
}