|
|
@@ -0,0 +1,91 @@
|
|
|
+// Copyright 2016 CoreOS, Inc.
|
|
|
+//
|
|
|
+// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
+// you may not use this file except in compliance with the License.
|
|
|
+// You may obtain a copy of the License at
|
|
|
+//
|
|
|
+// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+//
|
|
|
+// Unless required by applicable law or agreed to in writing, software
|
|
|
+// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
+// See the License for the specific language governing permissions and
|
|
|
+// limitations under the License.
|
|
|
+
|
|
|
+package cmd
|
|
|
+
|
|
|
+import (
|
|
|
+ "bytes"
|
|
|
+ "fmt"
|
|
|
+ "sort"
|
|
|
+ "sync"
|
|
|
+ "time"
|
|
|
+)
|
|
|
+
|
|
|
+type timeSeries struct {
|
|
|
+ timestamp int64
|
|
|
+ avgLatency time.Duration
|
|
|
+ throughPut int64
|
|
|
+}
|
|
|
+
|
|
|
+type TimeSeries []timeSeries
|
|
|
+
|
|
|
+func (t TimeSeries) Swap(i, j int) { t[i], t[j] = t[j], t[i] }
|
|
|
+func (t TimeSeries) Len() int { return len(t) }
|
|
|
+func (t TimeSeries) Less(i, j int) bool { return t[i].timestamp < t[j].timestamp }
|
|
|
+
|
|
|
+type secondPoint struct {
|
|
|
+ totalLatency time.Duration
|
|
|
+ count int64
|
|
|
+}
|
|
|
+
|
|
|
+type secondPoints struct {
|
|
|
+ mu sync.Mutex
|
|
|
+ tm map[int64]secondPoint
|
|
|
+}
|
|
|
+
|
|
|
+func newSecondPoints() *secondPoints {
|
|
|
+ return &secondPoints{tm: make(map[int64]secondPoint)}
|
|
|
+}
|
|
|
+
|
|
|
+func (sp *secondPoints) Add(ts time.Time, lat time.Duration) {
|
|
|
+ sp.mu.Lock()
|
|
|
+ defer sp.mu.Unlock()
|
|
|
+
|
|
|
+ tk := ts.Unix()
|
|
|
+ if v, ok := sp.tm[tk]; !ok {
|
|
|
+ sp.tm[tk] = secondPoint{totalLatency: lat, count: 1}
|
|
|
+ } else {
|
|
|
+ v.totalLatency += lat
|
|
|
+ v.count += 1
|
|
|
+ sp.tm[tk] = v
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (sp *secondPoints) getTimeSeries() TimeSeries {
|
|
|
+ sp.mu.Lock()
|
|
|
+ defer sp.mu.Unlock()
|
|
|
+
|
|
|
+ tslice := make(TimeSeries, len(sp.tm))
|
|
|
+ i := 0
|
|
|
+ for k, v := range sp.tm {
|
|
|
+ tslice[i] = timeSeries{
|
|
|
+ timestamp: k,
|
|
|
+ avgLatency: time.Duration(v.totalLatency) / time.Duration(v.count),
|
|
|
+ throughPut: v.count,
|
|
|
+ }
|
|
|
+ i++
|
|
|
+ }
|
|
|
+ sort.Sort(tslice)
|
|
|
+ return tslice
|
|
|
+}
|
|
|
+
|
|
|
+func (ts TimeSeries) String() string {
|
|
|
+ buf := new(bytes.Buffer)
|
|
|
+ buf.WriteString("Sample in one second (unix latency throughput):\n")
|
|
|
+ for i := range ts {
|
|
|
+ buf.WriteString(fmt.Sprintf("%7d %10s %5d\n", ts[i].timestamp, ts[i].avgLatency, ts[i].throughPut))
|
|
|
+ }
|
|
|
+ buf.WriteString("\n")
|
|
|
+ return buf.String()
|
|
|
+}
|
Gyu-Ho Lee