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
|
package shell_test
import (
"bytes"
"fmt"
"math"
"math/rand"
"strings"
"testing"
"bitbucket.org/creachadair/shell"
)
var input string
// Generate a long random string with balanced quotations for perf testing.
func init() {
var buf bytes.Buffer
src := rand.NewSource(12345)
r := rand.New(src)
const alphabet = "abcdefghijklmnopqrstuvwxyz0123456789 \t\n\n\n"
pick := func(f float64) byte {
pos := math.Ceil(f*float64(len(alphabet))) - 1
return alphabet[int(pos)]
}
const inputLen = 100000
var quote struct {
q byte
n int
}
for i := 0; i < inputLen; i++ {
if quote.n == 0 {
q := r.Float64()
if q < .1 {
quote.q = '"'
quote.n = r.Intn(256)
buf.WriteByte('"')
continue
} else if q < .15 {
quote.q = '\''
quote.n = r.Intn(256)
buf.WriteByte('\'')
continue
}
}
buf.WriteByte(pick(r.Float64()))
if quote.n > 0 {
quote.n--
if quote.n == 0 {
buf.WriteByte(quote.q)
}
}
}
input = buf.String()
}
func BenchmarkSplit(b *testing.B) {
var lens []int
for i := 1; i < len(input); i *= 4 {
lens = append(lens, i)
}
lens = append(lens, len(input))
b.ResetTimer()
for _, n := range lens {
b.Run(fmt.Sprintf("len_%d", n), func(b *testing.B) {
for i := 0; i < b.N; i++ {
shell.NewScanner(strings.NewReader(input[:n])).Each(ignore)
}
})
}
}
func ignore(string) bool { return true }
|
