From e8caa377ff3dca86d218a9c58474198b66dbf47b Mon Sep 17 00:00:00 2001
From: samwaf <samwafgo@gmail.com>
Date: Fri, 31 Oct 2025 10:24:03 +0800
Subject: [PATCH] fix: import host error

#502
---
 api/waf_common.go  | 36 ++++++++++++++++++++++++------------
 wafdb/patch_sql.go | 14 ++++++++++++++
 2 files changed, 38 insertions(+), 12 deletions(-)

diff --git a/api/waf_common.go b/api/waf_common.go
index dee9bf5..5596f72 100644
--- a/api/waf_common.go
+++ b/api/waf_common.go
@@ -193,8 +193,10 @@ func processImportData(dataType interface{}, tableName string, rows [][]string,
 		break
 	}
 
+	isBadData := false
 	// 处理数据 获取数据并插入数据库
 	for _, row := range rows {
+		isBadData = false
 		if rowNumber == 0 && needJumpFristCol {
 			rowNumber++
 			continue
@@ -227,8 +229,9 @@ func processImportData(dataType interface{}, tableName string, rows [][]string,
 			// 如果 dataMap 中有匹配的字段
 			if val, exists := dataMap[jsonTag]; exists {
 				//排除一些特定数据
-				if tableName == "hosts" && fieldName == "Host" && val == "全局网站" {
-					continue
+				if tableName == "hosts" && fieldName == "Host" && (val == "全局网站" || val == "") {
+					isBadData = true
+					break
 				}
 				//检查数据是否已经存在
 				if tableName == "hosts" && fieldName == "Host" {
@@ -237,14 +240,16 @@ func processImportData(dataType interface{}, tableName string, rows [][]string,
 						if err != nil {
 							*msg += fmt.Sprintf("行 %d, 检测数据合法性时候 出错: %v |", rowNumber, errMsg)
 							*failInt++
-							continue
+							isBadData = true
+							break
 						}
 					}
 					errMsg, err := checkHostPortData(dataMap["host"], dataMap["port"])
 					if err != nil {
 						*msg += fmt.Sprintf("行 %d, 检测数据合法性时候 出错: %v |", rowNumber, errMsg)
 						*failInt++
-						continue
+						isBadData = true
+						break
 					}
 				}
 				// 将字段值设置到结构体中
@@ -266,12 +271,15 @@ func processImportData(dataType interface{}, tableName string, rows [][]string,
 					if err != nil {
 						*msg += fmt.Sprintf("行 %d, 字段 %s 转换为 int 错误: %v |", rowNumber, fieldName, err)
 						*failInt++
-						continue
+						isBadData = true
+						break
 					}
 					fieldVal.SetInt(int64(intVal))
 				default:
 					*msg += fmt.Sprintf("不支持的字段类型: %s |", fieldVal.Kind())
 					*failInt++
+					isBadData = true
+					break
 				}
 			} else if fieldName == "BaseOrm" {
 				fieldVal := newInstance.Field(fieldIdx)
@@ -288,15 +296,19 @@ func processImportData(dataType interface{}, tableName string, rows [][]string,
 			} else {
 				*msg += fmt.Sprintf("行 %d, 缺少字段 %s 数据 |", rowNumber, fieldName)
 				*failInt++
+				isBadData = true
+				break
 			}
 		}
-		if err := global.GWAF_LOCAL_DB.Create(newInstance.Interface()); err != nil {
-			errGorm := err.Error
-			if errGorm != nil {
-				*msg += fmt.Sprintf("行 %d 插入失败: %v |", rowNumber, err.Error)
-				*failInt++
-			} else {
-				*successInt++
+		if isBadData == false {
+			if err := global.GWAF_LOCAL_DB.Create(newInstance.Interface()); err != nil {
+				errGorm := err.Error
+				if errGorm != nil {
+					*msg += fmt.Sprintf("行 %d 插入失败: %v |", rowNumber, err.Error)
+					*failInt++
+				} else {
+					*successInt++
+				}
 			}
 		}
 
diff --git a/wafdb/patch_sql.go b/wafdb/patch_sql.go
index b75a576..4946fc1 100644
--- a/wafdb/patch_sql.go
+++ b/wafdb/patch_sql.go
@@ -238,6 +238,20 @@ func pathCoreSql(db *gorm.DB) {
 	} else {
 		zlog.Info("db", "hosts :log_only_mode init successfully")
 	}
+
+	// 2025-10-31 删除 主机里面host字段为空的记录加上租户id和用户id
+	//如果存在才删除 否则不删除
+	var hostEmptyCount int64
+	db.Model(&model.Hosts{}).Where("host = ? AND tenant_id = ? AND user_code = ?", "", global.GWAF_TENANT_ID, global.GWAF_USER_CODE).Count(&hostEmptyCount)
+	if hostEmptyCount > 0 {
+		err = db.Exec("DELETE FROM hosts WHERE host = '' AND tenant_id = ? AND user_code = ?", global.GWAF_TENANT_ID, global.GWAF_USER_CODE).Error
+		if err != nil {
+			panic("failed to hosts :delete host empty record " + err.Error())
+		} else {
+			zlog.Info("db", "hosts :delete host empty record successfully")
+		}
+	}
+
 	// 记录结束时间并计算耗时
 	duration := time.Since(startTime)
 	zlog.Info("create core default value completely", "duration", duration.String())