InfluxDB伺服器啟動流程

操作系統 ： CentOS7.3.1611_x64

go語言版本：1.8.3 linux/amd64

InfluxDB版本：1.1.0

源碼路徑： github.com/influxdata/influxdb/cmd/influxd

程式入口（main.go）：

func main() {
    rand.Seed(time.Now().UnixNano())

    m := NewMain()
    if err := m.Run(os.Args[1:]...); err != nil {
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

在main函數中，調用Main.Run函數。

命令行參數為空或"run"

如命令行參數為空或"run"，則執行如下流程：

 1 case "", "run":
 2     cmd := run.NewCommand()
 3 
 4     // Tell the server the build details.
 5     cmd.Version = version
 6     cmd.Commit = commit
 7     cmd.Branch = branch
 8 
 9     if err := cmd.Run(args...); err != nil {
10         return fmt.Errorf("run: %s", err)
11     }
12 
13     signalCh := make(chan os.Signal, 1)
14     signal.Notify(signalCh, os.Interrupt, syscall.SIGTERM)
15     m.Logger.Println("Listening for signals")
16 
17     // Block until one of the signals above is received
18     select {
19     case <-signalCh:
20         m.Logger.Println("Signal received, initializing clean shutdown...")
21         go func() {
22             cmd.Close()
23         }()
24     }
25 
26     // Block again until another signal is received, a shutdown timeout elapses,
27     // or the Command is gracefully closed
28     m.Logger.Println("Waiting for clean shutdown...")
29     select {
30     case <-signalCh:
31         m.Logger.Println("second signal received, initializing hard shutdown")
32     case <-time.After(time.Second * 30):
33         m.Logger.Println("time limit reached, initializing hard shutdown")
34     case <-cmd.Closed:
35         m.Logger.Println("server shutdown completed")
36     }
37 
38     // goodbye.

執行 Command.Run 函數，後面跟的代碼是處理系統信號的相關內容。

Command.Run函數內容如下：

 1 func (cmd *Command) Run(args ...string) error {
 2     // Parse the command line flags.
 3     options, err := cmd.ParseFlags(args...)
 4     if err != nil {
 5         return err
 6     }
 7 
 8     // Print sweet InfluxDB logo.
 9     fmt.Print(logo)
10 
11     // Configure default logging.
12     log.SetPrefix("[run] ")
13     log.SetFlags(log.LstdFlags)
14 
15     // Set parallelism.
16     runtime.GOMAXPROCS(runtime.NumCPU())
17 
18     // Mark start-up in log.
19     log.Printf("InfluxDB starting, version %s, branch %s, commit %s",
20         cmd.Version, cmd.Branch, cmd.Commit)
21     log.Printf("Go version %s, GOMAXPROCS set to %d", runtime.Version(), runtime.GOMAXPROCS(0))
22 
23     // Write the PID file.
24     if err := cmd.writePIDFile(options.PIDFile); err != nil {
25         return fmt.Errorf("write pid file: %s", err)
26     }
27 
28     // Parse config
29     config, err := cmd.ParseConfig(options.GetConfigPath())
30     if err != nil {
31         return fmt.Errorf("parse config: %s", err)
32     }
33 
34     // Apply any environment variables on top of the parsed config
35     if err := config.ApplyEnvOverrides(); err != nil {
36         return fmt.Errorf("apply env config: %v", err)
37     }
38 
39     // Validate the configuration.
40     if err := config.Validate(); err != nil {
41         return fmt.Errorf("%s. To generate a valid configuration file run `influxd config > influxdb.generated.conf`", err)
42     }
43 
44     if config.HTTPD.PprofEnabled {
45         // Turn on block profiling to debug stuck databases
46         runtime.SetBlockProfileRate(int(1 * time.Second))
47     }
48 
49     // Create server from config and start it.
50     buildInfo := &BuildInfo{
51         Version: cmd.Version,
52         Commit:  cmd.Commit,
53         Branch:  cmd.Branch,
54         Time:    cmd.BuildTime,
55     }
56     s, err := NewServer(config, buildInfo)
57     if err != nil {
58         return fmt.Errorf("create server: %s", err)
59     }
60     s.CPUProfile = options.CPUProfile
61     s.MemProfile = options.MemProfile
62     if err := s.Open(); err != nil {
63         return fmt.Errorf("open server: %s", err)
64     }
65     cmd.Server = s
66 
67     // Begin monitoring the server's error channel.
68     go cmd.monitorServerErrors()
69 
70     return nil
71 }

解釋如下：

1、解析命令行參數並放入options變數中

2、列印InfluxDB字元串logo

3、設置日誌首碼

4、設置程式最大使用cpu的數量

預設使用伺服器上的所有cpu，最壞情況下會導致cpu占用100%的場景出現。

5、添加啟動日誌

6、記錄pid文件

該功能需要在命令行參數中指定pid文件路徑才可以。

7、載入配置文件

解析並校驗配置文件，如果沒有問題則配置文件生效。

8、設置profile信息並啟動伺服器

9、啟動各項服務

執行Server.Open函數（run/server.go）啟動各項服務，具體內容可以在Server.Open函數中查看。

10、執行monitorServerErrors用於監控伺服器出錯情況

命令行參數為"backup"

如果命令行參數為"backup"，則執行如下流程：

1 case "backup":
2     name := backup.NewCommand()
3     if err := name.Run(args...); err != nil {
4         return fmt.Errorf("backup: %s", err)
5     }

數據備份流程。

命令行參數為"restore"

如果命令行參數為"restore"，則執行如下流程：

case "restore":
    name := restore.NewCommand()
    if err := name.Run(args...); err != nil {
        return fmt.Errorf("restore: %s", err)
    }

數據恢復流程。

命令行參數為"config"

如果命令行參數為"config"，則執行如下流程：

case "config":
    if err := run.NewPrintConfigCommand().Run(args...); err != nil {
        return fmt.Errorf("config: %s", err)
    }

輸出預設的配置信息。

命令行參數為"help"

如果命令行參數為"help"，則執行如下流程：

case "help":
    if err := help.NewCommand().Run(args...); err != nil {
        return fmt.Errorf("help: %s", err)
    }

輸出幫助信息。

好，就這些了，希望對你有幫助。

本文github地址：

https://github.com/mike-zhang/mikeBlogEssays/blob/master/2018/20180105_InfluxDB伺服器啟動流程.rst

歡迎補充