khenaidoo | ac63710 | 2019-01-14 15:44:34 -0500 | [diff] [blame] | 1 | package metrics |
| 2 | |
| 3 | import ( |
| 4 | "runtime" |
| 5 | "time" |
| 6 | ) |
| 7 | |
| 8 | func (m *Metrics) SetGauge(key []string, val float32) { |
| 9 | if m.HostName != "" && m.EnableHostname { |
| 10 | key = insert(0, m.HostName, key) |
| 11 | } |
| 12 | if m.EnableTypePrefix { |
| 13 | key = insert(0, "gauge", key) |
| 14 | } |
| 15 | if m.ServiceName != "" { |
| 16 | key = insert(0, m.ServiceName, key) |
| 17 | } |
| 18 | m.sink.SetGauge(key, val) |
| 19 | } |
| 20 | |
| 21 | func (m *Metrics) EmitKey(key []string, val float32) { |
| 22 | if m.EnableTypePrefix { |
| 23 | key = insert(0, "kv", key) |
| 24 | } |
| 25 | if m.ServiceName != "" { |
| 26 | key = insert(0, m.ServiceName, key) |
| 27 | } |
| 28 | m.sink.EmitKey(key, val) |
| 29 | } |
| 30 | |
| 31 | func (m *Metrics) IncrCounter(key []string, val float32) { |
| 32 | if m.EnableTypePrefix { |
| 33 | key = insert(0, "counter", key) |
| 34 | } |
| 35 | if m.ServiceName != "" { |
| 36 | key = insert(0, m.ServiceName, key) |
| 37 | } |
| 38 | m.sink.IncrCounter(key, val) |
| 39 | } |
| 40 | |
| 41 | func (m *Metrics) AddSample(key []string, val float32) { |
| 42 | if m.EnableTypePrefix { |
| 43 | key = insert(0, "sample", key) |
| 44 | } |
| 45 | if m.ServiceName != "" { |
| 46 | key = insert(0, m.ServiceName, key) |
| 47 | } |
| 48 | m.sink.AddSample(key, val) |
| 49 | } |
| 50 | |
| 51 | func (m *Metrics) MeasureSince(key []string, start time.Time) { |
| 52 | if m.EnableTypePrefix { |
| 53 | key = insert(0, "timer", key) |
| 54 | } |
| 55 | if m.ServiceName != "" { |
| 56 | key = insert(0, m.ServiceName, key) |
| 57 | } |
| 58 | now := time.Now() |
| 59 | elapsed := now.Sub(start) |
| 60 | msec := float32(elapsed.Nanoseconds()) / float32(m.TimerGranularity) |
| 61 | m.sink.AddSample(key, msec) |
| 62 | } |
| 63 | |
| 64 | // Periodically collects runtime stats to publish |
| 65 | func (m *Metrics) collectStats() { |
| 66 | for { |
| 67 | time.Sleep(m.ProfileInterval) |
| 68 | m.emitRuntimeStats() |
| 69 | } |
| 70 | } |
| 71 | |
| 72 | // Emits various runtime statsitics |
| 73 | func (m *Metrics) emitRuntimeStats() { |
| 74 | // Export number of Goroutines |
| 75 | numRoutines := runtime.NumGoroutine() |
| 76 | m.SetGauge([]string{"runtime", "num_goroutines"}, float32(numRoutines)) |
| 77 | |
| 78 | // Export memory stats |
| 79 | var stats runtime.MemStats |
| 80 | runtime.ReadMemStats(&stats) |
| 81 | m.SetGauge([]string{"runtime", "alloc_bytes"}, float32(stats.Alloc)) |
| 82 | m.SetGauge([]string{"runtime", "sys_bytes"}, float32(stats.Sys)) |
| 83 | m.SetGauge([]string{"runtime", "malloc_count"}, float32(stats.Mallocs)) |
| 84 | m.SetGauge([]string{"runtime", "free_count"}, float32(stats.Frees)) |
| 85 | m.SetGauge([]string{"runtime", "heap_objects"}, float32(stats.HeapObjects)) |
| 86 | m.SetGauge([]string{"runtime", "total_gc_pause_ns"}, float32(stats.PauseTotalNs)) |
| 87 | m.SetGauge([]string{"runtime", "total_gc_runs"}, float32(stats.NumGC)) |
| 88 | |
| 89 | // Export info about the last few GC runs |
| 90 | num := stats.NumGC |
| 91 | |
| 92 | // Handle wrap around |
| 93 | if num < m.lastNumGC { |
| 94 | m.lastNumGC = 0 |
| 95 | } |
| 96 | |
| 97 | // Ensure we don't scan more than 256 |
| 98 | if num-m.lastNumGC >= 256 { |
| 99 | m.lastNumGC = num - 255 |
| 100 | } |
| 101 | |
| 102 | for i := m.lastNumGC; i < num; i++ { |
| 103 | pause := stats.PauseNs[i%256] |
| 104 | m.AddSample([]string{"runtime", "gc_pause_ns"}, float32(pause)) |
| 105 | } |
| 106 | m.lastNumGC = num |
| 107 | } |
| 108 | |
| 109 | // Inserts a string value at an index into the slice |
| 110 | func insert(i int, v string, s []string) []string { |
| 111 | s = append(s, "") |
| 112 | copy(s[i+1:], s[i:]) |
| 113 | s[i] = v |
| 114 | return s |
| 115 | } |