用 Golang 采集 Nginx 接口流量大小

本文介绍: 在开发和运维中，我们经常需要监控和分析服务器的接口流量大小，特别是对于部署了 Ngin x 的服务器。本文将介绍如何使用 Golang 采集 Ngin x 接口流量大小，并展示如何将这些数据进行实时监控和分析。

在开发和运维中，我们经常需要监控和分析服务器的接口流量大小，特别是对于部署了 Ngin x 的服务器。本文将介绍如何使用 Golang 采集 Ngin x 接口流量大小，并展示如何将这些数据进行实时监控和分析。

在开始之前，我们需要进行一些准备工作。

我们首先需要从 Nginx 的统计接口获取数据。可以使用 Golang 的 net/http 包发送 HTTP 请求并获取响应数据。

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
)

func main() {
	resp, err := http.Get("http://localhost/nginx_status")
	if err != nil {
		fmt.Println("获取 Nginx 统计数据失败:", err)
		return
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("读取 Nginx 统计数据失败:", err)
		return
	}

	fmt.Println(string(body))
}

上述代码中，我们使用 http.Get 方法发送 HTTP GET 请求，并使用 ioutil.ReadAll 方法将响应数据读取到变量 body 中。您需要将 http://localhost/nginx_status 替换为您实际的 Nginx 统计接口地址。

接下来，我们需要解析 Nginx 统计数据中的流量大小。根据 Nginx 统计模块的文档，我们可以从响应数据中提取我们需要的信息。

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
)

func main() {
	resp, err := http.Get("http://localhost/nginx_status")
	if err != nil {
		fmt.Println("获取 Nginx 统计数据失败:", err)
		return
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("读取 Nginx 统计数据失败:", err)
		return
	}

	stats := string(body)

	lines := strings.Split(stats, "n")
	for _, line := range lines {
		if strings.Contains(line, "Active connections") {
			fmt.Println(line)
		} else if strings.HasPrefix(line, "server accepts handled") {
			parts := strings.Fields(line)
			fmt.Println("接受的连接数:", parts[2])
			fmt.Println("处理的连接数:", parts[3])
			fmt.Println("已经完成的请求次数:", parts[4])
		} else if strings.HasPrefix(line, "Reading: ") {
			parts := strings.Fields(line)
			fmt.Println("正在读取的连接数:", parts[1])
		} else if strings.HasPrefix(line, "Writing: ") {
			parts := strings.Fields(line)
			fmt.Println("正在写入的连接数:", parts[1])
		} else if strings.HasPrefix(line, "Waiting: ") {
			parts := strings.Fields(line)
			fmt.Println("等待的连接数:", parts[1])
		}
	}
}

为了实现实时监控和分析，我们可以使用 Golang 的 time 包定时采集数据，并将数据保存到文件或发送到服务器进行进一步分析。

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
	"time"
)

func main() {
	ticker := time.NewTicker(10 * time.Second)
	defer ticker.Stop()

	for range ticker.C {
		resp, err := http.Get("http://localhost/nginx_status")
		if err != nil {
			fmt.Println("获取 Nginx 统计数据失败:", err)
			continue
		}

		body, err := ioutil.ReadAll(resp.Body)
		resp.Body.Close()
		if err != nil {
			fmt.Println("读取 Nginx 统计数据失败:", err)
			continue
		}

		stats := string(body)

		lines := strings.Split(stats, "n")
		for _, line := range lines {
			// 解析流量数据...
		}
	}
}

上述代码使用 time.NewTicker 创建一个定时器，每隔 10 秒钟采集一次数据。在每次定时器触发时，发送 HTTP 请求获取数据，并进行解析和处理。您可以根据需要调整定时器的间隔。

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"
	"time"

	"github.com/go-echarts/go-echarts/charts"
	"github.com/go-echarts/go-echarts/opts"
	"github.com/go-echarts/go-echarts/templates"
)

var (
	activeConnections []opts.LineData
	accepts           []opts.LineData
	handled           []opts.LineData
	requests          []opts.LineData
	reading           []opts.LineData
	writing           []opts.LineData
	waiting           []opts.LineData
)

func main() {
	http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
		page := charts.NewPage()

		line := charts.NewLine()
		line.SetGlobalOptions(
			charts.InitOpts{PageTitle: "接口流量监控"},
			charts.ToolboxOpts{Show: true},
			charts.YAxisOpts{Name: "数量"},
			charts.XAxisOpts{Name: "时间"},
		)

		line.AddXAxis(timeValues())
		line.AddYAxis("Active Connections", activeConnections)
		line.AddYAxis("Accepts", accepts)
		line.AddYAxis("Handled", handled)
		line.AddYAxis("Requests", requests)
		line.AddYAxis("Reading", reading)
		line.AddYAxis("Writing", writing)
		line.AddYAxis("Waiting", waiting)

		page.Add(line)

		w.Header().Set("Content-Type", "text/html; charset=utf-8")
		page.Render(w)
	})

	go func() {
		ticker := time.NewTicker(10 * time.Second)
		defer ticker.Stop()

		for range ticker.C {
			resp, err := http.Get("http://localhost/nginx_status")
			if err != nil {
				fmt.Println("获取 Nginx 统计数据失败:", err)
				continue
			}

			body, err := ioutil.ReadAll(resp.Body)
			resp.Body.Close()
			if err != nil {
				fmt.Println("读取 Nginx 统计数据失败:", err)
				continue
			}

			stats := string(body)

			lines := strings.Split(stats, "n")
			for _, line := range lines {
				// 解析流量数据...
			}

			line := charts.NewLine()
			line.SetGlobalOptions(
				charts.InitOpts{PageTitle: "接口流量监控"},
				charts.ToolboxOpts{Show: true},
				charts.YAxisOpts{Name: "数量"},
				charts.XAxisOpts{Name: "时间"},
			)

			line.AddXAxis(timeValues())
			line.AddYAxis("Active Connections", activeConnections)
			line.AddYAxis("Accepts", accepts)
			line.AddYAxis("Handled", handled)
			line.AddYAxis("Requests", requests)
			line.AddYAxis("Reading", reading)
			line.AddYAxis("Writing", writing)
			line.AddYAxis("Waiting", waiting)

			page := charts.NewPage()
			page.Add(line)

			s := templates.MustString(echartsTemplate, page)
			ioutil.WriteFile("index.html", []byte(s), 0644)
		}
	}()

	http.ListenAndServe(":8080", nil)
}

func timeValues() []string {
	t := time.Now().Format("2006-01-02 15:04:05")
	return []string{t}
}

const echartsTemplate = `
<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <title>{{"{{.PageTitle}}"}}</title>
    {{ template "asset" . }}
</head>
<body>
    {{ template "js" . }}
</body>
</html>