InfluxDB添加新服务

操作系统： 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\n", cur_time)

            time.Sleep( * 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

--- a/cmd/influxd/run/config.go

+++ b/cmd/influxd/run/config.go

@@ -, +, @@ 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"

 )

@@ -, +, @@ 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"`

@@ -, +, @@ 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

--- a/cmd/influxd/run/command.go

+++ b/cmd/influxd/run/command.go

@@ -, +, @@

 package run

 import (

+    "github.com/influxdata/influxdb/services/syncd"

        "flag"

        "fmt"

        "io"

@@ -, +, @@ 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

            .d888                    8888888b.  888888b.

               d88P"  888                   888  "Y88b   "88b

                                             .88P

      88888b.               8888888K.

       "88b 888    888 888  888  Y8bd8P' 888    888 888  "Y88b

                   X88K

              Y88b  .d8""8b.   .d88P    d88P

           "Y88888 888  888 8888888P"  8888888P"

[run] // :: InfluxDB starting, version unknown, branch unknown, commit unknown

[run] // :: Go version go1.8.3, GOMAXPROCS set to

[run] // :: Using configuration at: /etc/influxdb/influxdb.conf

[store] // :: Using data dir: /var/lib/influxdb/data

...

[syncd] // :: current timestamp :

[syncd] // :: current timestamp :

[syncd] // :: current timestamp :

生成新的配置文件：

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

欢迎补充