操作系统 ： CentOS7.3.1611_x64
go语言版本：1.8.3 linux/amd64
InfluxDB版本：1.1.0
这里以添加 syncd 服务为例记录下InfluxDB添加新服务的流程。
添加主服务代码
在 influxdata/influxdb/services 目录建立 syncd 文件夹，用于存放 syncd 服务相关代码。
1、添加服务配置相关内容
添加 config.go 文件，示例内容如下：
package syncd // E-Mail : Mike_Zhang@live.com type Config struct { Enabled bool `toml:"enabled"` LogEnabled bool `toml:"log-enabled"` RemoteTSDBHost string `toml:"remote-host"` DefaultDB string `toml:"defaultDB"` } func NewConfig() Config { return Config{ Enabled: true, LogEnabled: true, RemoteTSDBHost: "http://127.0.0.1:8086", DefaultDB: "Monitor", } }
解释如下：
- 定义 Config 用于存放具体配置；
- 添加 NewConfig 函数，用于创建 Config 对象；
2、添加 syncd 服务的具体内容
添加 syncd.go 文件，示例内容如下：
package syncd // E-Mail : Mike_Zhang@live.com import ( "log" "os" "time" ) type Service struct { Logger *log.Logger remote_host string DefaultDB string username string password string } func NewService(c Config) *Service { return &Service{ remote_host: c.RemoteTSDBHost, DefaultDB: c.DefaultDB, username: "root", password: "root", Logger: log.New(os.Stderr, "[syncd] ", log.LstdFlags), } } func (s *Service) Run() { for { cur_time := time.Now().Unix() s.Logger.Printf("current timestamp : %d ", cur_time) time.Sleep(1 * time.Second) } }
解释如下： * 定义Service结构；
- 添加 NewService 函数，用于创建具体服务；
- 添加 Run 函数，实现具体服务
该函数作为入口实现具体的服务，这里以在日志中定时打印时间戳作为具体的服务内容。
在服务器主程序中启动服务
1、添加配置相关代码
进入 influxdata/influxdb/cmd/influxd 目录，修改 run/config.go 文件。
- 引入 syncd 服务
在 import 中加入如下代码：
"github.com/influxdata/influxdb/services/syncd"
- 定义配置
在 Config 结构中加入如下变量：
SyncdConfig syncd.Config `toml:"syncd"`
- 初始化配置
在 NewConfig 函数中加入如下代码：
c.SyncdConfig = syncd.NewConfig()
使用 git diff 命令查看如下 ：
[root@localhost run]# git diff config.go diff --git a/cmd/influxd/run/config.go b/cmd/influxd/run/config.go index 36e4f14..01df0cc 100644 --- a/cmd/influxd/run/config.go +++ b/cmd/influxd/run/config.go @@ -27,6 +27,7 @@ import ( "github.com/influxdata/influxdb/services/precreator" "github.com/influxdata/influxdb/services/retention" "github.com/influxdata/influxdb/services/subscriber" + "github.com/influxdata/influxdb/services/syncd" "github.com/influxdata/influxdb/services/udp" "github.com/influxdata/influxdb/tsdb" ) @@ -48,6 +49,7 @@ type Config struct { Monitor monitor.Config `toml:"monitor"` Subscriber subscriber.Config `toml:"subscriber"` HTTPD httpd.Config `toml:"http"` + SyncdConfig syncd.Config `toml:"syncd"` GraphiteInputs []graphite.Config `toml:"graphite"` CollectdInputs []collectd.Config `toml:"collectd"` OpenTSDBInputs []opentsdb.Config `toml:"opentsdb"` @@ -84,6 +86,7 @@ func NewConfig() *Config { c.Retention = retention.NewConfig() c.BindAddress = DefaultBindAddress + c.SyncdConfig = syncd.NewConfig() return c } [root@localhost run]#
2、添加启动服务代码
进入 influxdata/influxdb/cmd/influxd 目录，修改 run/command.go 文件
- 引入 syncd 服务
在 import 中加入如下代码：
"github.com/influxdata/influxdb/services/syncd"
- 添加启动代码
在 Command->Run 函数中加入如下代码（go cmd.monitorServerErrors() 之前）：
// start syncd syncdInstance := syncd.NewService(config.SyncdConfig) go syncdInstance.Run()
在 Config 结构中加入如下变量：
使用 git diff 命令查看如下 ：
[root@localhost run]# git diff command.go diff --git a/cmd/influxd/run/command.go b/cmd/influxd/run/command.go index 51036f1..8743f04 100644 --- a/cmd/influxd/run/command.go +++ b/cmd/influxd/run/command.go @@ -1,6 +1,7 @@ package run import ( + "github.com/influxdata/influxdb/services/syncd" "flag" "fmt" "io" @@ -120,6 +121,11 @@ func (cmd *Command) Run(args ...string) error { } cmd.Server = s + // start syncd + syncdInstance := syncd.NewService(config.SyncdConfig) + go syncdInstance.Run() + + // Begin monitoring the server's error channel. go cmd.monitorServerErrors() [root@localhost run]#
测试服务
进入 influxdata/influxdb/cmd/influxd 目录，执行 go build 命令，并将编译好的二进制文件copy到bin目录，具体如下：
[root@localhost influxd]# go build [root@localhost influxd]# cp influxd /usr/bin/ cp: overwrite ‘/usr/bin/influxd’? y [root@localhost influxd]#
启动InfluxDB服务器，在控制台可以看到如下内容：
[root@localhost influxdb]# influxd 8888888 .d888 888 8888888b. 888888b. 888 d88P" 888 888 "Y88b 888 "88b 888 888 888 888 888 888 .88P 888 88888b. 888888 888 888 888 888 888 888 888 8888888K. 888 888 "88b 888 888 888 888 Y8bd8P' 888 888 888 "Y88b 888 888 888 888 888 888 888 X88K 888 888 888 888 888 888 888 888 888 Y88b 888 .d8""8b. 888 .d88P 888 d88P 8888888 888 888 888 888 "Y88888 888 888 8888888P" 8888888P" [run] 2018/02/07 04:24:28 InfluxDB starting, version unknown, branch unknown, commit unknown [run] 2018/02/07 04:24:28 Go version go1.8.3, GOMAXPROCS set to 2 [run] 2018/02/07 04:24:28 Using configuration at: /etc/influxdb/influxdb.conf [store] 2018/02/07 04:24:29 Using data dir: /var/lib/influxdb/data ... [syncd] 2018/02/07 21:56:11 current timestamp : 1518058571 [syncd] 2018/02/07 21:56:12 current timestamp : 1518058572 [syncd] 2018/02/07 21:56:13 current timestamp : 1518058573
生成新的配置文件：
influxd config > new.conf
可以看到 syncd 服务默认配置如下：
[syncd] enabled = true log-enabled = true remote-host = "http://127.0.0.1:8086" defaultDB = "Monitor"
好，就这些了，希望对你有帮助。
本文github地址：
https://github.com/mike-zhang/mikeBlogEssays/blob/master/2018/20180210_InfluxDB添加新服务.rst
欢迎补充