/
util.go
70 lines (63 loc) · 1.46 KB
/
util.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
package main
import "reflect"
import "fmt"
import "sort"
import "strings"
import "compress/flate"
import "unsafe"
import "github.com/bnclabs/gofast"
func bytes2str(bytes []byte) string {
if bytes == nil {
return ""
}
sl := (*reflect.SliceHeader)(unsafe.Pointer(&bytes))
st := &reflect.StringHeader{Data: sl.Data, Len: sl.Len}
return *(*string)(unsafe.Pointer(st))
}
func fixbuffer(buffer []byte, size int64) []byte {
if size == 0 {
return buffer
} else if buffer == nil || int64(cap(buffer)) < size {
return make([]byte, size)
}
return buffer[:size]
}
func printCounts(counts map[string]uint64) {
if counts == nil {
fmt.Println("statistics is nil")
return
}
keys := []string{}
for key := range counts {
keys = append(keys, key)
}
sort.Sort(sort.StringSlice(keys))
s := []string{}
for _, key := range keys {
s = append(s, fmt.Sprintf(`"%v":%v`, key, counts[key]))
}
fmt.Println("stats {", strings.Join(s, ", "), "}")
}
func addCounts(transs ...*gofast.Transport) map[string]uint64 {
if len(transs) > 0 {
counts := transs[0].Stat()
for _, trans := range transs[1:] {
for k, v := range trans.Stat() {
counts[k] += v
}
}
return counts
}
return nil
}
func newconfig(start, end int) map[string]interface{} {
return map[string]interface{}{
"buffersize": options.buffersize,
"chansize": 100000,
"batchsize": 100,
"tags": "",
"opaque.start": start,
"opaque.end": end,
"gzip.level": flate.BestSpeed,
}
}