123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149 |
- package toolbox
- import (
- "fmt"
- "sync"
- "time"
- )
- type Statistics struct {
- RequestURL string
- RequestController string
- RequestNum int64
- MinTime time.Duration
- MaxTime time.Duration
- TotalTime time.Duration
- }
- type URLMap struct {
- lock sync.RWMutex
- LengthLimit int
- urlmap map[string]map[string]*Statistics
- }
- func (m *URLMap) AddStatistics(requestMethod, requestURL, requestController string, requesttime time.Duration) {
- m.lock.Lock()
- defer m.lock.Unlock()
- if method, ok := m.urlmap[requestURL]; ok {
- if s, ok := method[requestMethod]; ok {
- s.RequestNum++
- if s.MaxTime < requesttime {
- s.MaxTime = requesttime
- }
- if s.MinTime > requesttime {
- s.MinTime = requesttime
- }
- s.TotalTime += requesttime
- } else {
- nb := &Statistics{
- RequestURL: requestURL,
- RequestController: requestController,
- RequestNum: 1,
- MinTime: requesttime,
- MaxTime: requesttime,
- TotalTime: requesttime,
- }
- m.urlmap[requestURL][requestMethod] = nb
- }
- } else {
- if m.LengthLimit > 0 && m.LengthLimit <= len(m.urlmap) {
- return
- }
- methodmap := make(map[string]*Statistics)
- nb := &Statistics{
- RequestURL: requestURL,
- RequestController: requestController,
- RequestNum: 1,
- MinTime: requesttime,
- MaxTime: requesttime,
- TotalTime: requesttime,
- }
- methodmap[requestMethod] = nb
- m.urlmap[requestURL] = methodmap
- }
- }
- func (m *URLMap) GetMap() map[string]interface{} {
- m.lock.RLock()
- defer m.lock.RUnlock()
- var fields = []string{"requestUrl", "method", "times", "used", "max used", "min used", "avg used"}
- var resultLists [][]string
- content := make(map[string]interface{})
- content["Fields"] = fields
- for k, v := range m.urlmap {
- for kk, vv := range v {
- result := []string{
- fmt.Sprintf("% -50s", k),
- fmt.Sprintf("% -10s", kk),
- fmt.Sprintf("% -16d", vv.RequestNum),
- fmt.Sprintf("%d", vv.TotalTime),
- fmt.Sprintf("% -16s", toS(vv.TotalTime)),
- fmt.Sprintf("%d", vv.MaxTime),
- fmt.Sprintf("% -16s", toS(vv.MaxTime)),
- fmt.Sprintf("%d", vv.MinTime),
- fmt.Sprintf("% -16s", toS(vv.MinTime)),
- fmt.Sprintf("%d", time.Duration(int64(vv.TotalTime)/vv.RequestNum)),
- fmt.Sprintf("% -16s", toS(time.Duration(int64(vv.TotalTime)/vv.RequestNum))),
- }
- resultLists = append(resultLists, result)
- }
- }
- content["Data"] = resultLists
- return content
- }
- func (m *URLMap) GetMapData() []map[string]interface{} {
- m.lock.Lock()
- defer m.lock.Unlock()
- var resultLists []map[string]interface{}
- for k, v := range m.urlmap {
- for kk, vv := range v {
- result := map[string]interface{}{
- "request_url": k,
- "method": kk,
- "times": vv.RequestNum,
- "total_time": toS(vv.TotalTime),
- "max_time": toS(vv.MaxTime),
- "min_time": toS(vv.MinTime),
- "avg_time": toS(time.Duration(int64(vv.TotalTime) / vv.RequestNum)),
- }
- resultLists = append(resultLists, result)
- }
- }
- return resultLists
- }
- var StatisticsMap *URLMap
- func init() {
- StatisticsMap = &URLMap{
- urlmap: make(map[string]map[string]*Statistics),
- }
- }
|