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()
}
//加载配置文件
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, )
//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 = # default
output = [
{ name="influxdb-1", location="127.0.0.1:8089", mtu= },
{ name="influxdb-", location="127.0.0.1:7089", mtu= },
]

总结:

1、加载配置文件 relay.toml

2、根据配置文件创建relay,启动监听端口,接收请求

3、把接收到的请求写入到influxdb1-2节点

后面再研究下核心模块relay怎么把接收到的数据同时写入,influxdb多个节点。

上一篇:Emoji表情符号录入MySQL数据库报错的解决方案(MySQL utf8与utf8mb4区别)


下一篇:python redis自带门神 lock 方法