misspell/benchmark_test.go at main · golangci/misspell · 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
package misspell

import (
	"bytes"
	"io"
	"testing"
)

var (
	sampleClean string
	sampleDirty string
	tmpCount    int
	tmp         string
	rep         *Replacer
)

func init() {
	buf := bytes.Buffer{}
	for i := 0; i < len(DictMain); i += 2 {
		buf.WriteString(DictMain[i+1] + " ")

		if i%5 == 0 {
			buf.WriteString("\n")
		}
	}

	sampleClean = buf.String()
	sampleDirty = sampleClean + DictMain[0] + "\n"
	rep = New()
}

// BenchmarkCleanString takes a clean string (one with no errors).
func BenchmarkCleanString(b *testing.B) {
	b.ResetTimer()
	b.ReportAllocs()

	var (
		updated string
		diffs   []Diff
		count   int
	)

	for n := 0; n < b.N; n++ { //nolint:intrange // use Loop -> go1.24
		updated, diffs = rep.Replace(sampleClean)
		count += len(diffs)
	}

	// prevent compiler optimizations
	tmpCount = count
	tmp = updated
}

func discardDiff(_ Diff) {
	tmpCount++
}

// BenchmarkCleanStream takes a clean reader (no misspells) and outputs to a buffer.
func BenchmarkCleanStream(b *testing.B) {
	b.ResetTimer()
	b.ReportAllocs()

	tmpCount = 0
	buf := bytes.NewBufferString(sampleClean)
	out := bytes.NewBuffer(make([]byte, 0, len(sampleClean)+100))

	for n := 0; n < b.N; n++ { //nolint:intrange // use Loop -> go1.24
		buf.Reset()
		buf.WriteString(sampleClean)
		out.Reset()
		rep.ReplaceReader(buf, out, discardDiff)
	}
}

// BenchmarkCleanStreamDiscard takes a clean reader and discards output.
func BenchmarkCleanStreamDiscard(b *testing.B) {
	b.ResetTimer()
	b.ReportAllocs()

	buf := bytes.NewBufferString(sampleClean)
	tmpCount = 0

	for n := 0; n < b.N; n++ { //nolint:intrange // use Loop -> go1.24
		buf.Reset()
		buf.WriteString(sampleClean)
		rep.ReplaceReader(buf, io.Discard, discardDiff)
	}
}

// BenchmarkCleanString takes a clean string (one with no errors).
func BenchmarkDirtyString(b *testing.B) {
	b.ResetTimer()
	b.ReportAllocs()

	var (
		updated string
		diffs   []Diff
		count   int
	)

	for n := 0; n < b.N; n++ { //nolint:intrange // use Loop -> go1.24
		updated, diffs = rep.Replace(sampleDirty)
		count += len(diffs)
	}

	// prevent compiler optimizations
	tmpCount = count
	tmp = updated
}

func BenchmarkCompile(b *testing.B) {
	r := New()

	b.ReportAllocs()
	b.ResetTimer()

	for n := 0; n < b.N; n++ { //nolint:intrange // use Loop -> go1.24
		r.Compile()
	}
}