nuclei/pkg/protocols/common/generators/maps_bench_test.go at dev · stringsbuilder/nuclei · 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
package generators

import (
	"fmt"
	"testing"
)

func BenchmarkMergeMaps(b *testing.B) {
	map1 := map[string]interface{}{
		"key1": "value1",
		"key2": "value2",
		"key3": "value3",
		"key4": "value4",
		"key5": "value5",
	}
	map2 := map[string]interface{}{
		"key6":  "value6",
		"key7":  "value7",
		"key8":  "value8",
		"key9":  "value9",
		"key10": "value10",
	}
	map3 := map[string]interface{}{
		"key11": "value11",
		"key12": "value12",
		"key13": "value13",
	}

	for i := 1; i <= 3; i++ {
		b.Run(fmt.Sprintf("%d-maps", i), func(b *testing.B) {
			b.ReportAllocs()
			for b.Loop() {
				switch i {
				case 1:
					_ = MergeMaps(map1)
				case 2:
					_ = MergeMaps(map1, map2)
				case 3:
					_ = MergeMaps(map1, map2, map3)
				}
			}
		})
	}
}

func BenchmarkCopyMap(b *testing.B) {
	map1 := map[string]interface{}{
		"key1": "value1",
		"key2": "value2",
		"key3": "value3",
		"key4": "value4",
		"key5": "value5",
	}

	for i := 1; i <= 1; i++ {
		b.Run(fmt.Sprintf("%d-maps", i), func(b *testing.B) {
			b.ReportAllocs()
			for b.Loop() {
				switch i {
				case 1:
					_ = CopyMap(map1)
				}
			}
		})
	}
}

func BenchmarkMergeMapsInto(b *testing.B) {
	map1 := map[string]interface{}{
		"key1": "value1",
		"key2": "value2",
		"key3": "value3",
		"key4": "value4",
		"key5": "value5",
	}
	map2 := map[string]interface{}{
		"key6":  "value6",
		"key7":  "value7",
		"key8":  "value8",
		"key9":  "value9",
		"key10": "value10",
	}
	map3 := map[string]interface{}{
		"key11": "value11",
		"key12": "value12",
		"key13": "value13",
	}
	map4 := map[string]interface{}{
		"key14": "value14",
		"key15": "value15",
		"key16": "value16",
	}

	for i := 1; i <= 3; i++ {
		b.Run(fmt.Sprintf("%d-maps", i), func(b *testing.B) {
			b.ReportAllocs()
			for b.Loop() {
				switch i {
				case 1:
					MergeMapsInto(map1, map2)
				case 2:
					MergeMapsInto(map1, map2, map3)
				case 3:
					MergeMapsInto(map1, map2, map3, map4)
				}
			}
		})
	}
}