iot init

pkg/tdengine/tdengine_table.go

@@ -0,0 +1,162 @@
+package tdengine
+
+import (
+	"errors"
+	"fmt"
+	"github.com/kakuilan/kgo"
+	"strconv"
+	"strings"
+)
+
+// RunSql 运行
+func (s *TdEngine) RunSql(sql string) (err error) {
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// InsertDevice 数据入库
+func (s *TdEngine) InsertDevice(deviceKey string, data map[string]any) (err error) {
+
+	if len(data) == 0 {
+		return
+	}
+	var (
+		field = []string{}
+		value = []string{}
+	)
+	for k, v := range data {
+		field = append(field, k)
+		value = append(value, "'"+kgo.KConv.ToStr(v)+"'")
+	}
+
+	sql := "INSERT INTO ? (?) VALUES (?)"
+	_, err = s.db.Exec(sql, deviceKey, strings.Join(field, ","), strings.Join(value, ","))
+
+	return
+}
+
+// CreateStable 创建超级表 rowdata 源数据，在需要数据解析时使用
+func (s *TdEngine) CreateStable(table string) (err error) {
+	columns := []string{"ts TIMESTAMP,rowdata NCHAR(255)"}
+	sql := fmt.Sprintf("CREATE STABLE IF NOT EXISTS %s.%s (%s) TAGS (device nchar(64))", s.dbName, table, strings.Join(columns, ","))
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// CreateTable 添加子表
+func (s *TdEngine) CreateTable(stable, table string) (err error) {
+	sql := fmt.Sprintf("CREATE TABLE %s USING %s TAGS ('%s')", table, stable, table)
+	_, err = s.db.Exec(sql)
+	return
+}
+
+func (s *TdEngine) column(dataType, key, name string, maxLength int) string {
+	column := ""
+	comment := ""
+	if name != "" {
+		comment = "COMMENT '" + name + "'"
+	}
+	tdType := ""
+	switch dataType {
+	case "int64":
+		tdType = "INT"
+	case "long":
+		tdType = "BIGINT"
+	case "float64":
+		tdType = "FLOAT"
+	case "double":
+		tdType = "DOUBLE"
+	case "string":
+		if maxLength == 0 {
+			maxLength = 255
+		}
+		tdType = "NCHAR(" + strconv.Itoa(maxLength) + ")"
+	case "boolean":
+		tdType = "BOOL"
+	case "date":
+		tdType = "TIMESTAMP"
+	default:
+		if maxLength == 0 {
+			maxLength = 255
+		}
+		tdType = "NCHAR(" + strconv.Itoa(maxLength) + ")"
+	}
+	column = fmt.Sprintf("%s %s %s", key, tdType, comment)
+	return column
+}
+
+// 删除超级表
+func (s *TdEngine) DropStable(table string) (err error) {
+	sql := fmt.Sprintf("DROP STABLE IF EXISTS %s.%s", s.dbName, table)
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// 删除子表
+func (s *TdEngine) DropTable(table string) (err error) {
+	sql := fmt.Sprintf("DROP TABLE IF EXISTS %s.%s", s.dbName, table)
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// AddSTableField 添加数据库超级表字段
+func (s *TdEngine) AddSTableField(tableName, fieldName string, dataType string, len int) (err error) {
+	sql := fmt.Sprintf("ALTER STABLE %s.%s ADD COLUMN %s", s.dbName, tableName, s.column(dataType, fieldName, "", len))
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// AddTableField 添加数据库表字段
+func (s *TdEngine) AddTableField(tableName, fieldName string, dataType string, len int) (err error) {
+	sql := fmt.Sprintf("ALTER TABLE %s.%s ADD COLUMN %s", s.dbName, tableName, s.column(dataType, fieldName, "", len))
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// DelTableField 删除数据库表字段
+func (s *TdEngine) DelTableField(tableName, fieldName string) (err error) {
+	sql := fmt.Sprintf("ALTER TABLE %s.%s DROP COLUMN %s", s.dbName, tableName, fieldName)
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// DelSTableField 删除数据库超级表字段
+func (s *TdEngine) DelSTableField(tableName, fieldName string) (err error) {
+	sql := fmt.Sprintf("ALTER STABLE %s.%s DROP COLUMN %s", s.dbName, tableName, fieldName)
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// ModifyDatabaseField 修改数据库指定字段长度
+func (s *TdEngine) ModifyDatabaseField(tableName, fieldName string, dataType string, len int) (err error) {
+	sql := fmt.Sprintf("ALTER STABLE %s.%s MODIFY COLUMN %s", s.dbName, tableName, s.column(dataType, fieldName, "", len))
+	_, err = s.db.Exec(sql)
+	if err != nil {
+		err = errors.New("设置字段长度失败,长度只能增大不能缩小")
+	}
+	return
+}
+
+// AddTag 添加标签
+func (s *TdEngine) AddTag(tableName, tagName string, dataType string, len int) (err error) {
+	sql := fmt.Sprintf("ALTER STABLE %s.%s ADD TAG %s", s.dbName, tableName, s.column(dataType, tagName, "", len))
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// DelTag 删除标签
+func (s *TdEngine) DelTag(tableName, tagName string) (err error) {
+	sql := fmt.Sprintf("ALTER STABLE %s.%s DROP TAG %s", s.dbName, tableName, tagName)
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// ModifyTag 修改标签
+func (s *TdEngine) ModifyTag(tableName, tagName string, dataType string, len int) (err error) {
+	sql := fmt.Sprintf("ALTER STABLE %s.%s MODIFY TAG %s", s.dbName, tableName, s.column(dataType, tagName, "", len))
+	_, err = s.db.Exec(sql)
+	if err != nil {
+		err = errors.New("设置标签长度失败,长度只能增大不能缩小")
+	}
+	return
+}