go開源項目influxdb-relay源碼分析（一）

influxdb-relay項目地址: https://github.com/influxdata/influxdb-relay，主要作為負載均衡節點，寫入多個influxdb節點，起到高可用效果。

不多說，開搞~

1、main.go

package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    "os/signal"

    "github.com/influxdata/influxdb-relay/relay"
)

var (
    //influxdb-realy啟動時接收的配置文件參數,因為是falg.String所以接收的是一個字符串類型的參數
    //flag.String 函數第一個參數代表啟動時參數的名稱，第二個參數代表默認值，第三個參數代表用途描述，此方法返回一個指針
    //那么啟動realy時應該是這個樣子的： ./influxdb-relay  -config  relay.toml  ———— "-config" 就是匹配上面flag.String函數第一個參數, "relay.toml" 是-config的值,這里代表配置文件的路徑
    configFile = flag.String("config", "", "Configuration file to use")
)

func main() {
    //flag.Parse是用來解析啟動時傳入的參數，必須調用
    flag.Parse()

    //如果啟動參數，配置文件路徑為空
    if *configFile == "" {
        //記錄日志
        fmt.Fprintln(os.Stderr, "Missing configuration file")
        //打印需要的參數提示,也就是會輸出上面flag.String函數第三個參數
        flag.PrintDefaults()
        //退出
        os.Exit(1)
    }
    //加載配置文件
    cfg, err := relay.LoadConfigFile(*configFile)
    //LoadConfigFile方法如下： 主要做的事情是把配置文件反序列化成一個Config對象然后return，也就是cfg
    // func LoadConfigFile(filename string) (cfg Config, err error) {
    //    f, err := os.Open(filename)
    //    if err != nil {
    //        return cfg, err
    //    }
    //    defer f.Close()
    //    return cfg, toml.NewDecoder(f).Decode(&cfg)
    // }
    if err != nil {
        fmt.Fprintln(os.Stderr, "Problem loading config file:", err)
    }
    //根據Config對象 創建relay組件
    r, err := relay.New(cfg)
    if err != nil {
        log.Fatal(err)
    }
    //定義接收操作系統信號的channel
    sigChan := make(chan os.Signal, 1)
    //notify用於監聽信號,如果操作系統向當前進程發送中斷信號(os.Interrupt 表示中斷信號)
    signal.Notify(sigChan, os.Interrupt)
    //如果接收到中斷信號，開一個協程調用relay的Stop方法 停止http監聽
    go func() {
        <-sigChan
        r.Stop()
    }()

    log.Println("starting relays...")
    //relay 運行
    r.Run()
}

2、配置文件 relay.toml

[[http]]
name = "example-http"
bind-addr = "127.0.0.1:9096"
output = [
    { name="influxdb-1", location = "http://127.0.0.1:8086/write" },
    { name="influxdb-2", location = "http://127.0.0.1:7086/write" },
]

[[udp]]
name = "example-udp"
bind-addr = "127.0.0.1:19096"
read-buffer = 0 # default
output = [
    { name="influxdb-1", location="127.0.0.1:8089", mtu=512 },
    { name="influxdb-2", location="127.0.0.1:7089", mtu=1024 },
]

 

總結：

1、加載配置文件 relay.toml

2、根據配置文件創建relay，啟動監聽端口，接收請求

3、把接收到的請求寫入到influxdb1-2節點

 

后面再研究下核心模塊relay怎么把接收到的數據同時寫入，influxdb多個節點。