-
Notifications
You must be signed in to change notification settings - Fork 0
/
speed_test.go
103 lines (77 loc) · 1.67 KB
/
speed_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
91
92
93
94
95
96
97
98
99
100
101
102
103
package unite
import (
"crypto/rand"
"io"
"testing"
"time"
)
func writeSpeedStatistic(writer io.Writer, size int64) (int64, error) {
b := make([]byte, size)
_, err := rand.Read(b)
if err != nil {
return 0, err
}
start := time.Now()
_, err = writer.Write(b)
if err != nil {
return 0, err
}
end := time.Now()
cost := end.UnixNano() - start.UnixNano()
return cost, nil
}
func TestCrossSpeedStatistic(t *testing.T) {
unite, err := CreateUniteFile("./test.unite")
if err != nil {
panic(err)
}
totalCost := int64(0)
size := int64(1 << 10)
totalSize := int64(1 << 30)
count := totalSize / size
var files [2]*file
file1, err := unite.Create("test1")
if err != nil {
panic(err)
}
file2, err := unite.Create("test2")
if err != nil {
panic(err)
}
files[0] = file1
files[1] = file2
for i := int64(0); i < count*2; i++ {
cost, err := writeSpeedStatistic(files[i%2], size)
if err != nil {
panic(err)
}
totalCost += cost
}
t.Logf("write speed: %f MB/s write count %d ", float64(totalSize*2)/(float64(totalCost)/1e9)/1e6, count)
}
func TestSpeed(t *testing.T) {
unite, err := CreateUniteFile("./test.unite")
if err != nil {
panic(err)
}
file, err := unite.Create("speed.test")
if err != nil {
panic(err)
}
//file, err := os.Create("speed.test")
//if err != nil {
// panic(err)
//}
totalCost := int64(0)
size := int64(1 << 10)
totalSize := int64(1 << 30)
count := totalSize / size
for i := int64(0); i < count; i++ {
cost, err := writeSpeedStatistic(file, size)
if err != nil {
panic(err)
}
totalCost += cost
}
t.Logf("write speed: %f MB/s write count %d ", float64(totalSize)/(float64(totalCost)/1e9)/1e6, count)
}