优化代码

internal/service/addfriend/add_wechat_friend_log.go

@@ -63,29 +63,26 @@ func (l *AddWechatFriendService) AddNewFriend(wechatId, content, message string,
 	if wechatType != 1 && wechatType != 2 {
 		return false
 	}
-	var hookClient *hook.Hook
-	var node *snowflake.Node
-	var err error
-	if wechatType == 1 {
-		hookClient = hook.NewHook("", "", "")
-		node, err = snowflake.NewNode(1) // 1 是节点 ID，根据需要设置
-	} else {
-		hookClient = hook.NewWecomHook("", "", "")
-		node, err = snowflake.NewNode(1)
-	}
+	node, err := snowflake.NewNode(1) // 1 是节点 ID，根据需要设置
 	if err != nil {
 		log.Fatal(err)
+		return false
 	}
+	var (
+		hookClient *hook.Hook
+		result     map[string]interface{}
+	)
 	taskId := node.Generate()
-
-	var result map[string]interface{}
-
-	if wechatType == 1 {
+	switch wechatType {
+	case 1:
+		hookClient = hook.NewHook("", "", "")
 		result, err = hookClient.AddFriendTask(wechatId, content, message, taskId.Int64())
-	} else {
+	case 2:
+		hookClient = hook.NewWecomHook("", "", "")
 		result, err = hookClient.AddWecomCustomerFromSearchTask(wechatId, content, message, taskId.Int64())
+	default:
+		log.Println("Invalid wechat type")
 	}
-
 	if err == nil {
 		result["sendResult"] = "success"
 	} else {
@@ -101,7 +98,7 @@ func (l *AddWechatFriendService) AddNewFriend(wechatId, content, message string,
 	_, err = update.Save(l.ctx)
 	logx.Infof("更新 AddWechatFriendLog-field-add-result成功: %v", result)
 	if err != nil || result["sendResult"] != "success" {
-		logx.Errorf("更新 AddWechatFriendLog 失败 或者: %v", err)
+		logx.Errorf("更新 AddWechatFriendLog 失败 或者: %v,%s", err, result["sendResult"])
 		return false
 	}
 	return true //获取好友列表