基于prometheus 实现exporter新增监控指标

基于node_exporter增加

cmd/collector/rds_files_linux.go

package collector

import (
    "github.com/go-kit/log"
    "github.com/prometheus/client_golang/prometheus"
)

func init() {
    registerCollector("system", defaultEnabled, NewSystemCollector)
}

type SystemCollector struct {
    log log.Logger
}

func NewSystemCollector(logger log.Logger) (Collector, error) {
    return &SystemCollector{log: logger}, nil
}

var sysInfo = map[string]*prometheus.Desc{
    "dir_size": prometheus.NewDesc(
        "rds_dir_size",
        "自定义指定文件夹的大小",
        []string{}, prometheus.Labels{},
    ),
    "file_size": prometheus.NewDesc(
        "rds_file_size",
        "自定义指定文件的大小",
        []string{}, prometheus.Labels{},
    ),
}

func (SystemCollector) Update(ch chan<- prometheus.Metric) error {
    var dirSize = DirSize("/root/go/src/node_mysqld_exporter/cmd/collector")
    ch <- prometheus.MustNewConstMetric(
        sysInfo["dir_size"],
        prometheus.GaugeValue, float64(dirSize),
    )
    var fileSize = DirSize("/root/go/src/node_mysqld_exporter/cmd/collector/cpu_linux.go")
    ch <- prometheus.MustNewConstMetric(
        sysInfo["file_size"],
        prometheus.GaugeValue, float64(fileSize),
    )
    return nil
}

cmd/collector/utils.go

package collector

import (
    "io/ioutil"
    "log"
    "os"
    "path/filepath"
    "sync"
    "time"
)

var syncM sync.Map

var wait sync.WaitGroup

func DirSize(path string) int64 {
    var size int64
    filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
        if err != nil {
            log.Println(err.Error())
            return err
        }
        if !info.IsDir() {
            size += info.Size()
        }
        return err
    })
    return size
}

func getFileSize(path string) int64 {
    if !exists(path) {
        return 0
    }
    fileInfo, err := os.Stat(path)
    log.Println(fileInfo)
    if err != nil {
        return 0
    }
    return fileInfo.Size()
}

func exists(path string) bool {
    _, err := os.Stat(path)
    return err == nil || os.IsExist(err)
}

func SizeOfDirPath(dirPath string) float64 {
    log.Println("扫描文件夹：" + dirPath)
    t := time.Now()
    wait.Add(1)
    go scanDir(dirPath, &syncM)
    wait.Wait()

    var fileCount int
    var dirSize float64

    syncM.Range(func(key, value interface{}) bool {
        fileCount++
        v := value.(float64)
        dirSize += v
        return true
    })
    log.Println("花费的时间：" + time.Since(t).String())
    log.Println("文件的数量：", fileCount)
    log.Println("文件夹大小：", dirSize)
    return dirSize
}

func scanDir(path string, syncM *sync.Map) {
    defer wait.Done()
    dirAry, err := ioutil.ReadDir(path)
    if err != nil {
        panic(err)
    }
    for _, e := range dirAry {
        if e.IsDir() {
            wait.Add(1)
            go scanDir(filepath.Join(path, e.Name()), syncM)
        } else {
            syncM.Store(filepath.Join(path, e.Name()), (float64(e.Size()) / 1024))
        }
    }

}

验证

# HELP rds_dir_size 自定义指定文件夹的大小
# TYPE rds_dir_size gauge
rds_dir_size 1.591099e+06
# HELP rds_file_size 自定义指定文件的大小
# TYPE rds_file_size gauge
rds_file_size 15226