From 659a30c1a1653858f328e91db5c8277ab6e86130 Mon Sep 17 00:00:00 2001
From: hardy <luohf@infinilabs.com>
Date: Wed, 8 May 2024 08:58:56 +0800
Subject: [PATCH] fix: alert rule import patch for v1.6.0 (#346)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修复[344](https://git.infini.ltd/infini/console/issues/344)

Reviewed-on: https://git.infini.ltd/infini/console/pulls/346
Co-authored-by: hardy <luohf@infinilabs.com>
Co-committed-by: hardy <luohf@infinilabs.com>
---
 plugin/api/data/export.go | 31 +++++++++++++++++++++++--------
 plugin/api/data/model.go  | 13 +++++++------
 2 files changed, 30 insertions(+), 14 deletions(-)

diff --git a/plugin/api/data/export.go b/plugin/api/data/export.go
index 36f415f8..44dec3e4 100644
--- a/plugin/api/data/export.go
+++ b/plugin/api/data/export.go
@@ -6,13 +6,15 @@ package data
 
 import (
 	"fmt"
+	"net/http"
+
 	log "github.com/cihub/seelog"
 	"infini.sh/console/model"
 	"infini.sh/console/model/alerting"
 	httprouter "infini.sh/framework/core/api/router"
+	"infini.sh/framework/core/global"
 	"infini.sh/framework/core/orm"
 	"infini.sh/framework/core/util"
-	"net/http"
 )
 
 func (h *DataAPI) exportData(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
@@ -32,8 +34,9 @@ func (h *DataAPI) exportData(w http.ResponseWriter, req *http.Request, ps httpro
 			return
 		}
 		resBody = append(resBody, ExportData{
-			Type: meta.Type,
-			Data: result.Result,
+			Version: global.Env().GetVersion(),
+			Type:    meta.Type,
+			Data:    result.Result,
 		})
 	}
 	h.WriteJSON(w, resBody, http.StatusOK)
@@ -48,7 +51,11 @@ func (h *DataAPI) importData(w http.ResponseWriter, req *http.Request, ps httpro
 		h.WriteError(w, err.Error(), http.StatusInternalServerError)
 		return
 	}
-	err = indexExportData(reqBody)
+	needPatch := true
+	if len(reqBody) > 0 && len(reqBody[0].Version) > 0 {
+		needPatch = false
+	}
+	err = indexExportData(reqBody, needPatch)
 	if err != nil {
 		log.Error(err)
 		h.WriteError(w, err.Error(), http.StatusInternalServerError)
@@ -57,7 +64,7 @@ func (h *DataAPI) importData(w http.ResponseWriter, req *http.Request, ps httpro
 	h.WriteAckOKJSON(w)
 }
 
-func indexExportData(eds []ExportData) error {
+func indexExportData(eds []ExportData, patch bool) error {
 	for _, ed := range eds {
 		for _, row := range ed.Data {
 			var obj interface{}
@@ -73,6 +80,15 @@ func indexExportData(eds []ExportData) error {
 			}
 			buf := util.MustToJSONBytes(row)
 			err := util.FromJSONBytes(buf, obj)
+			// 当导出无版本号，并且为告警规则，并且导出数据无分类时
+			if patch && ed.Type == DataTypeAlertRule {
+				if rule, ok := obj.(*alerting.Rule); ok {
+					if len(obj.(*alerting.Rule).Category) == 0 {
+						rule.Category = "Platform"
+						obj = rule
+					}
+				}
+			}
 			if err != nil {
 				return err
 			}
@@ -81,7 +97,6 @@ func indexExportData(eds []ExportData) error {
 				return err
 			}
 		}
-
 	}
 	return nil
 }
@@ -103,7 +118,7 @@ func getExportData(meta ExportMetadata) (*orm.Result, error) {
 	}
 	if meta.Filter != nil {
 		query := util.MapStr{
-			"size": 1000,
+			"size":  1000,
 			"query": meta.Filter,
 		}
 		q.RawQuery = util.MustToJSONBytes(query)
@@ -113,4 +128,4 @@ func getExportData(meta ExportMetadata) (*orm.Result, error) {
 		return nil, err
 	}
 	return &result, err
-}
\ No newline at end of file
+}
diff --git a/plugin/api/data/model.go b/plugin/api/data/model.go
index 1635285f..5546523e 100644
--- a/plugin/api/data/model.go
+++ b/plugin/api/data/model.go
@@ -9,17 +9,18 @@ type ExportDataRequest struct {
 }
 
 type ExportMetadata struct {
-	Type string `json:"type"`
+	Type   string      `json:"type"`
 	Filter interface{} `json:"filter,omitempty"`
 }
 
 type ExportData struct {
-	Type string `json:"type"`
-	Data []interface{} `json:"data"`
+	Version string        `json:"version,omitempty"`
+	Type    string        `json:"type"`
+	Data    []interface{} `json:"data"`
 }
 
 const (
-	DataTypeAlertRule = "AlertRule"
-	DataTypeAlertChannel = "AlertChannel"
+	DataTypeAlertRule        = "AlertRule"
+	DataTypeAlertChannel     = "AlertChannel"
 	DataTypeAlertEmailServer = "EmailServer"
-)
\ No newline at end of file
+)