etcd/raft_stats.go

package main

import (
	"container/list"
	"math"
	"time"

	"github.com/coreos/go-raft"
)

const (
	queueCapacity = 200
)

type runtimeStats struct {
}

type packageStats struct {
	sendingTime time.Time
	size        uint64
}

type raftServerStats struct {
	State                 string
	StartTime             time.Time
	Leader                string
	leaderStartTime       time.Time
	LeaderUptime          string
	RecvAppendRequestCnt  uint64
	SendAppendRequestCnt  uint64
	SendAppendReqeustRate uint64
	sendRateQueue         *list.List
	recvRateQueue         *list.List
	SendingRate           float64
}

func (ss *raftServerStats) RecvAppendReq(leaderName string) {
	ss.State = raft.Follower
	if leaderName != ss.Leader {
		ss.Leader = leaderName
		ss.leaderStartTime = time.Now()
	}

	ss.RecvAppendRequestCnt++
}

func (ss *raftServerStats) SendAppendReq() {
	now := time.Now()
	if ss.State != raft.Leader {
		ss.State = raft.Leader
		ss.Leader = r.Name()
		ss.leaderStartTime = now
	}

	if ss.sendRateQueue.Len() < 200 {
		ss.sendRateQueue.PushBack(now)
	} else {
		ss.sendRateQueue.PushBack(now)
		ss.sendRateQueue.Remove(ss.sendRateQueue.Front())
	}

	ss.SendAppendRequestCnt++
}

type raftPeerStats struct {
	Latency          float64 `json:"latency"`
	AvgLatency       float64 `json:"averageLatency"`
	avgLatencySquare float64
	SdvLatency       float64 `json:"sdvLatency"`
	MinLatency       float64 `json:"minLatency"`
	MaxLatency       float64 `json:"maxLatency"`
	FailCnt          uint64  `json:"failsCount"`
	SuccCnt          uint64  `json:"successCount"`
}

func (ps *raftPeerStats) Fail() {
	ps.FailCnt++
}

func (ps *raftPeerStats) Succ(d time.Duration) {

	total := float64(ps.SuccCnt) * ps.AvgLatency
	totalSquare := float64(ps.SuccCnt) * ps.avgLatencySquare

	ps.SuccCnt++

	ps.Latency = float64(d) / (1000000.0)

	if ps.Latency > ps.MaxLatency {
		ps.MaxLatency = ps.Latency
	}

	if ps.Latency < ps.MinLatency {
		ps.MinLatency = ps.Latency
	}

	ps.AvgLatency = (total + ps.Latency) / float64(ps.SuccCnt)
	ps.avgLatencySquare = (totalSquare + ps.Latency*ps.Latency) / float64(ps.SuccCnt)

	// sdv = sqrt(avg(x^2) - avg(x)^2)
	ps.SdvLatency = math.Sqrt(ps.avgLatencySquare - ps.AvgLatency*ps.AvgLatency)
}

type statsQueue struct {
	items [queueCapacity]*packageStats
	size  int
	front int
	back  int
}

func (q *statsQueue) Len() int {
	return q.size
}

func (q *statsQueue) Front() *packageStats {
	if q.size != 0 {
		return q.items[q.front]
	}
	return nil
}

func (q *statsQueue) Back() *packageStats {
	if q.size != 0 {
		return q.items[q.back]
	}
	return nil
}

func (q *statsQueue) Insert(p *packageStats) {
	q.back = (q.back + 1) % queueCapacity
	q.items[q.back] = p
	if q.size == queueCapacity {
		q.front = (q.back + 1) % queueCapacity
	} else {
		q.size++
	}
}