-
Notifications
You must be signed in to change notification settings - Fork 0
/
bench_test.go
90 lines (79 loc) · 2.19 KB
/
bench_test.go
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
package pgx_collect_test
import (
"database/sql"
"testing"
"time"
pgxc "github.com/zolstein/pgx-collect"
pgxc_internal "github.com/zolstein/pgx-collect/internal"
. "github.com/zolstein/pgx-collect/internal/testutils"
)
func BenchmarkInitialize(b *testing.B) {
type Person struct {
Name string
Age sql.NullInt64
}
type Record struct {
ID int64
Person
CreatedAt time.Time
}
rows := MakeMockRows("id,name,age,created_at", nil)
b.Run("cold", func(b *testing.B) {
b.Run("struct-by-pos", func(b *testing.B) {
var rowTo pgxc.RowSpec[Record] = pgxc.RowToStructByPos[Record]
scanner := rowTo.Scanner()
b.ResetTimer()
for i := 0; i < b.N; i++ {
pgxc_internal.ClearStructFieldCaches()
scanner.Initialize(rows)
}
})
b.Run("struct-by-name", func(b *testing.B) {
var rowTo pgxc.RowSpec[Record] = pgxc.RowToStructByName[Record]
scanner := rowTo.Scanner()
b.ResetTimer()
for i := 0; i < b.N; i++ {
pgxc_internal.ClearStructFieldCaches()
scanner.Initialize(rows)
}
})
b.Run("struct-by-name-lax", func(b *testing.B) {
var rowTo pgxc.RowSpec[Record] = pgxc.RowToStructByNameLax[Record]
scanner := rowTo.Scanner()
b.ResetTimer()
for i := 0; i < b.N; i++ {
pgxc_internal.ClearStructFieldCaches()
scanner.Initialize(rows)
}
})
})
b.Run("hot", func(b *testing.B) {
b.Run("struct-by-pos", func(b *testing.B) {
var rowTo pgxc.RowSpec[Record] = pgxc.RowToStructByPos[Record]
scanner := rowTo.Scanner()
scanner.Initialize(rows) // First time to populate cache
b.ResetTimer()
for i := 0; i < b.N; i++ {
scanner.Initialize(rows)
}
})
b.Run("struct-by-name", func(b *testing.B) {
var rowTo pgxc.RowSpec[Record] = pgxc.RowToStructByName[Record]
scanner := rowTo.Scanner()
scanner.Initialize(rows) // First time to populate cache
b.ResetTimer()
for i := 0; i < b.N; i++ {
scanner.Initialize(rows)
}
})
b.Run("struct-by-name-lax", func(b *testing.B) {
var rowTo pgxc.RowSpec[Record] = pgxc.RowToStructByNameLax[Record]
scanner := rowTo.Scanner()
scanner.Initialize(rows) // First time to populate cache
b.ResetTimer()
for i := 0; i < b.N; i++ {
scanner.Initialize(rows)
}
})
})
}