InfluxDB 1.8如何通过Golang API判断数据库是否存在及不存在时的创建方法

刚好之前处理过类似的需求，用InfluxDB官方的Golang客户端库就能轻松搞定。我们需要用到github.com/influxdata/influxdb/client/v2——这是操作InfluxDB 1.8最稳定的SDK，下面分步骤说明实现方式：

1. 判断指定数据库是否存在 #

核心思路是执行SHOW DATABASES查询，然后遍历结果集检查目标数据库是否在列表中。

实现代码片段： #

import (
    "context"
    "fmt"
    "log"

    "github.com/influxdata/influxdb/client/v2"
)

// databaseExists 检查指定数据库是否存在
func databaseExists(c client.Client, dbName string) (bool, error) {
    // 构造查询语句，SHOW DATABASES是全局查询，无需指定数据库
    q := client.Query{
        Command:  "SHOW DATABASES",
        Database: "",
    }

    // 执行查询
    response, err := c.Query(context.Background(), q)
    if err != nil {
        return false, fmt.Errorf("执行SHOW DATABASES失败: %w", err)
    }
    if response.Error() != nil {
        return false, fmt.Errorf("查询错误: %w", response.Error())
    }

    // 遍历结果集，匹配目标数据库名
    for _, result := range response.Results {
        for _, series := range result.Series {
            for _, value := range series.Values {
                if len(value) > 0 && value[0] == dbName {
                    return true, nil
                }
            }
        }
    }

    return false, nil
}

2. 不存在则创建数据库 #

如果检查返回false，执行CREATE DATABASE <dbname>语句即可完成创建。

实现代码片段： #

// createDatabase 创建指定数据库
func createDatabase(c client.Client, dbName string) error {
    // 构造创建语句，特殊字符的数据库名用双引号包裹
    q := client.Query{
        Command:  fmt.Sprintf("CREATE DATABASE \"%s\"", dbName),
        Database: "",
    }

    // 执行查询
    response, err := c.Query(context.Background(), q)
    if err != nil {
        return fmt.Errorf("执行CREATE DATABASE失败: %w", err)
    }
    if response.Error() != nil {
        return fmt.Errorf("创建数据库错误: %w", response.Error())
    }

    return nil
}

完整流程示例 #

把上面的函数组合起来，形成完整的使用流程：

func main() {
    // 配置InfluxDB连接信息
    cfg := client.HTTPConfig{
        Addr:     "http://localhost:8086", // 你的InfluxDB地址
        Username: "admin",                  // 开启认证时填写用户名
        Password: "password",               // 开启认证时填写密码
    }

    // 创建客户端连接
    c, err := client.NewHTTPClient(cfg)
    if err != nil {
        log.Fatalf("创建InfluxDB客户端失败: %v", err)
    }
    defer c.Close()

    targetDB := "my_monitor_db"

    // 检查数据库是否存在
    exists, err := databaseExists(c, targetDB)
    if err != nil {
        log.Fatalf("检查数据库存在性失败: %v", err)
    }

    if !exists {
        // 创建数据库
        err := createDatabase(c, targetDB)
        if err != nil {
            log.Fatalf("创建数据库失败: %v", err)
        }
        log.Printf("数据库 %s 创建成功", targetDB)
    } else {
        log.Printf("数据库 %s 已存在", targetDB)
    }

    // 接下来就可以正常写入数据到该数据库了
}

注意事项： #

• 先安装依赖：执行go get github.com/influxdata/influxdb/client/v2
• 如果InfluxDB未开启认证，可省略Username和Password配置
• 数据库名包含特殊字符时，必须用双引号包裹（示例中已处理）

内容的提问来源于stack exchange，提问作者張三瘋