流量回放工具之 GoReplay output-http-stats（HTTP请求统计）源码分析

最新推荐文章于 2022-11-23 09:22:09 发布

zuozewei

最新推荐文章于 2022-11-23 09:22:09 发布

阅读量1.7k

点赞数 1

分类专栏：性能文章标签： golang http 开发语言

本文链接：https://blog.csdn.net/zuozewei/article/details/120950233

版权

性能专栏收录该内容

213 篇文章 110 订阅

订阅专栏

文章目录

前言

GoReplay 可以报告 stats 请求队列的统计信息。通过结合使用--output-http-stats和选项，默认统计信息每 5 秒输出到给控制台。

参数：–stats --output-http-stats

 -output-http-stats  //每5秒钟输出一次输出队列的状态
        Report http output queue stats to console every N milliseconds. See output-http-stats-ms
  -output-http-stats-ms int
        Report http output queue stats to console every N milliseconds. default: 5000 (default 5000)

控制台输出是这样的：

在这里插入图片描述

其中倒数第二列等同于当前的TPS；

output_http.go

HTTPOutputConfig 统计信息收集
是否收集统计信息，统计输出间隔是多少

	if o.config.Stats {
		o.queueStats = NewGorStat("output_http", o.config.StatsMs)
	}

在这里插入图片描述

统计信息收集：

// PluginWrite writes message to this plugin
// 统计信息收集
func (o *HTTPOutput) PluginWrite(msg *Message) (n int, err error) {
	if !isRequestPayload(msg.Meta) {
		return len(msg.Data), nil
	}

	select {
	case <-o.stop:
		return 0, ErrorStopped
	case o.queue <- msg:
	}

	if o.config.Stats {
		o.queueStats.Write(len(o.queue))
	}
	if len(o.queue) > 0 {
		// try to start a new worker to serve
		if atomic.LoadInt32(&o.activeWorkers) < int32(o.config.WorkersMax) {
			go o.startWorker()
			atomic.AddInt32(&o.activeWorkers, 1)
		}
	}
	return len(msg.Data) + len(msg.Meta), nil
}

在这里插入图片描述

gor_stat.go

NewGorStat 统计类：

// NewGorStat统计类
func NewGorStat(statName string, rateMs int) (s *GorStat) {
	s = new(GorStat)
	s.statName = statName
	s.rateMs = rateMs
	s.latest = 0
	s.mean = 0
	s.max = 0
	s.count = 0

	if Settings.Stats {
		go s.reportStats()
	}
	return
}

在这里插入图片描述

写入时做统计：

//写入时做统计
func (s *GorStat) Write(latest int) {
	if Settings.Stats {
		if latest > s.max {
			s.max = latest
		}
		if latest != 0 {
			s.mean = ((s.mean * s.count) + latest) / (s.count + 1)
		}
		s.latest = latest
		s.count = s.count + 1
	}
}

在这里插入图片描述

打印输出，简单的sleep：

//打印输出，简单的sleep
func (s *GorStat) reportStats() {
	Debug(0, "\n", s.statName+":latest,mean,max,count,count/second,gcount")
	for {
		Debug(0, "\n", s)
		s.Reset()
		time.Sleep(time.Duration(s.rateMs) * time.Millisecond)
	}

在这里插入图片描述

字符串类型强转：

//字符串类型转换
func (s *GorStat) String() string {
	return s.statName + ":" + strconv.Itoa(s.latest) + "," + strconv.Itoa(s.mean) + "," + strconv.Itoa(s.max) + "," + strconv.Itoa(s.count) + "," + strconv.Itoa(s.count/(s.rateMs/1000.0)) + "," + strconv.Itoa(runtime.NumGoroutine())
}