完善influx

.gitignore

@@ -18,3 +18,4 @@ go.sum
 tmp/
 .vscode/
 logs/
+snippet

influx/influx.go

@@ -3,6 +3,7 @@ package influx
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"time"
 
@@ -101,3 +102,31 @@ func ReadSample() error {
 	}
 	return nil
 }
+
+//创建数据库，并配置默认过期策略（单位:天）
+func CreateDB(dbname string, expired uint) error {
+	if dbname == "" || expired == 0 {
+		return errors.New("无效参数")
+	}
+	db, err := New()
+	if err != nil {
+		return err
+	}
+	defer db.Close()
+
+	createDbSQL := client.NewQuery(fmt.Sprintf("CREATE DATABASE %s", dbname), "", "")
+	result, err := db.Query(createDbSQL)
+	if err != nil {
+		return err
+	}
+	if result.Error() != nil {
+		return result.Error()
+	}
+	// 过期策略
+	createRPSQL := client.NewQuery(fmt.Sprintf("CREATE RETENTION POLICY default ON %s DURATION %dd REPLICATION 1 DEFAULT", dbname, expired), dbname, "")
+	result, err = db.Query(createRPSQL)
+	if err != nil {
+		return err
+	}
+	return result.Error()
+}