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
|
package snowflake_proxy
import (
"time"
)
// bytesLogger is an interface which is used to allow logging the throughput
// of the Snowflake. A default bytesLogger(bytesNullLogger) does nothing.
type bytesLogger interface {
AddOutbound(int64)
AddInbound(int64)
GetStat() (in int64, out int64)
}
// bytesNullLogger Default bytesLogger does nothing.
type bytesNullLogger struct{}
// AddOutbound in bytesNullLogger does nothing
func (b bytesNullLogger) AddOutbound(amount int64) {}
// AddInbound in bytesNullLogger does nothing
func (b bytesNullLogger) AddInbound(amount int64) {}
func (b bytesNullLogger) GetStat() (in int64, out int64) { return -1, -1 }
// bytesSyncLogger uses channels to safely log from multiple sources with output
// occuring at reasonable intervals.
type bytesSyncLogger struct {
outboundChan, inboundChan chan int64
statsChan chan bytesLoggerStats
stats bytesLoggerStats
outEvents, inEvents int
start time.Time
}
type bytesLoggerStats struct {
outbound, inbound int64
}
// newBytesSyncLogger returns a new bytesSyncLogger and starts it loggin.
func newBytesSyncLogger() *bytesSyncLogger {
b := &bytesSyncLogger{
outboundChan: make(chan int64, 5),
inboundChan: make(chan int64, 5),
statsChan: make(chan bytesLoggerStats),
}
go b.log()
b.start = time.Now()
return b
}
func (b *bytesSyncLogger) log() {
for {
select {
case amount := <-b.outboundChan:
b.stats.outbound += amount
b.outEvents++
case amount := <-b.inboundChan:
b.stats.inbound += amount
b.inEvents++
case b.statsChan <- b.stats:
b.stats.inbound = 0
b.stats.outbound = 0
b.inEvents = 0
b.outEvents = 0
}
}
}
// AddOutbound add a number of bytes to the outbound total reported by the logger
func (b *bytesSyncLogger) AddOutbound(amount int64) {
b.outboundChan <- amount
}
// AddInbound add a number of bytes to the inbound total reported by the logger
func (b *bytesSyncLogger) AddInbound(amount int64) {
b.inboundChan <- amount
}
// GetStat returns the current inbound and outbound stats from the logger and then zeros the counts
func (b *bytesSyncLogger) GetStat() (in int64, out int64) {
stats := <-b.statsChan
return stats.inbound, stats.outbound
}
func formatTraffic(amount int64) (value int64, unit string) { return amount / 1000, "KB" }
|
