增加批量导入资产功能

pkg/api/asset.go

@@ -1,6 +1,8 @@
 package api
 
 import (
+	"bufio"
+	"encoding/csv"
 	"encoding/json"
 	"errors"
 	"github.com/labstack/echo/v4"
@@ -44,6 +46,77 @@ func AssetCreateEndpoint(c echo.Context) error {
 	return Success(c, item)
 }
 
+func AssetImportEndpoint(c echo.Context) error {
+	account, _ := GetCurrentAccount(c)
+
+	file, err := c.FormFile("file")
+	if err != nil {
+		return err
+	}
+
+	src, err := file.Open()
+	if err != nil {
+		return err
+	}
+
+	defer src.Close()
+	reader := csv.NewReader(bufio.NewReader(src))
+	records, err := reader.ReadAll()
+	if err != nil {
+		return err
+	}
+
+	total := len(records)
+	if total == 0 {
+		return errors.New("csv数据为空")
+	}
+
+	var successCount = 0
+	var errorCount = 0
+	m := echo.Map{}
+
+	for i := 0; i < total; i++ {
+		record := records[i]
+		if len(record) >= 9 {
+			port, _ := strconv.Atoi(record[3])
+			asset := model.Asset{
+				ID:          utils.UUID(),
+				Name:        record[0],
+				Protocol:    record[1],
+				IP:          record[2],
+				Port:        port,
+				AccountType: model.Custom,
+				Username:    record[4],
+				Password:    record[5],
+				PrivateKey:  record[6],
+				Passphrase:  record[7],
+				Description: record[8],
+				Created:     utils.NowJsonTime(),
+				Owner:       account.ID,
+			}
+
+			err := model.CreateNewAsset(&asset)
+			if err != nil {
+				errorCount++
+				m[strconv.Itoa(i)] = err.Error()
+			} else {
+				successCount++
+				// 创建后自动检测资产是否存活
+				go func() {
+					active := utils.Tcping(asset.IP, asset.Port)
+					model.UpdateAssetActiveById(active, asset.ID)
+				}()
+			}
+		}
+	}
+
+	return Success(c, echo.Map{
+		"successCount": successCount,
+		"errorCount":   errorCount,
+		"data":         m,
+	})
+}
+
 func AssetPagingEndpoint(c echo.Context) error {
 	pageIndex, _ := strconv.Atoi(c.QueryParam("pageIndex"))
 	pageSize, _ := strconv.Atoi(c.QueryParam("pageSize"))
@@ -134,6 +207,10 @@ func AssetUpdateEndpoint(c echo.Context) error {
 func AssetGetAttributeEndpoint(c echo.Context) error {
 
 	assetId := c.Param("id")
+	if err := PreCheckAssetPermission(c, assetId); err != nil {
+		return err
+	}
+
 	attributeMap, err := model.FindAssetAttrMapByAssetId(assetId)
 	if err != nil {
 		return err
@@ -177,6 +254,9 @@ func AssetDeleteEndpoint(c echo.Context) error {
 
 func AssetGetEndpoint(c echo.Context) (err error) {
 	id := c.Param("id")
+	if err := PreCheckAssetPermission(c, id); err != nil {
+		return err
+	}
 
 	var item model.Asset
 	if item, err = model.FindAssetById(id); err != nil {