iot init

pkg/tdengine/tdengine_log.go

@@ -0,0 +1,56 @@
+package tdengine
+
+import "time"
+
+// CreateLogStable 添加LOG超级表
+func (s *TdEngine) CreateLogStable() (err error) {
+	var name string
+	err = s.db.QueryRow("SELECT stable_name FROM information_schema.ins_stables WHERE stable_name = 'device_log' LIMIT 1").Scan(&name)
+	if name != "" {
+		return
+	}
+	sql := "CREATE STABLE device_log (ts TIMESTAMP, type VARCHAR(20), content VARCHAR(1000)) TAGS (device VARCHAR(255))"
+	_, err = s.db.Exec(sql)
+	return
+}
+
+// InsertLog 写入数据
+func (s *TdEngine) InsertLog(log *TdLog) (err error) {
+	sql := "INSERT INTO ? USING device_log TAGS ('?') VALUES ('?', '?', '?')"
+	_, err = s.db.Exec(sql, "log_"+log.Device, log.Device, log.Ts, log.Type, log.Content)
+
+	return
+}
+
+// ClearLog 清理过期数据
+func (s *TdEngine) ClearLog() (err error) {
+	ts := time.Now().Add(-7 * 24 * time.Hour).Format("2006-01-02")
+
+	sql := "DELETE FROM device_log WHERE ts < '" + ts + "'"
+	_, err = s.db.Exec(sql)
+
+	return
+}
+
+// GetAllLog 超级表查询，多条数据
+func (s *TdEngine) GetAllLog(sql string, args ...any) (list []TdLog, err error) {
+	rows, err := s.db.Query(sql, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var log TdLog
+
+		err = rows.Scan(&log.Ts, &log.Type, &log.Content, &log.Device)
+		if err != nil {
+			return nil, err
+		}
+		log.Ts = s.Time(log.Ts)
+
+		list = append(list, log)
+	}
+
+	return
+}