wechat-api/internal/service/MessageHandlers/task_result_notice.go

package MessageHandlers

import (
	"context"
	"encoding/json"
	"github.com/zeromicro/go-zero/core/logx"
	"strconv"
	"time"
	"wechat-api/ent/addwechatfriendlog"
	"wechat-api/internal/pkg/wechat_ws"
	"wechat-api/internal/svc"
)

type TaskResultNoticeHandler struct {
	svcCtx *svc.ServiceContext
}

func NewTaskResultNoticeHandler(svcCtx *svc.ServiceContext) *TaskResultNoticeHandler {
	return &TaskResultNoticeHandler{
		svcCtx: svcCtx,
	}
}

// Handle 实现 MessageHandlerStrategy 接口
func (f *TaskResultNoticeHandler) Handle(ctx context.Context, msg *wechat_ws.MsgJsonObject, svcCtx *svc.ServiceContext) error {

	var messageTask TaskResultNoticeMessage
	if err := json.Unmarshal([]byte(msg.Message), &messageTask); err != nil {
		logx.Errorf("预处理 JSON 失败: %v", err)
		return err
	}
	// 如果 TaskId 是字符串，就转换成数字
	var taskIdInt int64
	if messageTask.TaskId != nil {
		var err error
		taskIdInt, err = strconv.ParseInt(*messageTask.TaskId, 10, 64)
		if err != nil {
			logx.Errorf("Unmarshal.fail")
			return err
		}
	}

	if messageTask.TaskType != "AddFriendsTask" {
		logx.Infof("不是添加好友任务的回执")
		return nil
	}
	m, err := ParseJSONStringToMap(msg.Message)
	if err != nil {
		logx.Errorf("解析 JSON 失败: %v", err)
		return err
	}
	update := svcCtx.DB.AddWechatFriendLog.Update().
		Where(addwechatfriendlog.OwnerWxIDEQ(*messageTask.WeChatId)).
		Where(addwechatfriendlog.TaskIDEQ(taskIdInt)).
		//SetFindResult(m).
		SetAddResult(m).
		SetUpdatedAt(time.Now().Unix())
	if *messageTask.Success {
		update.SetIsCanAdd(2)
	}
	if _, err := update.Save(ctx); err != nil {
		logx.Errorf("更新AddWechatFriendLog-field-add-result失败: %v", err)
		return err
	}
	return nil
}

type TaskResultNoticeMessage struct {
	Success  *bool   `json:"Success,omitempty"`  // 指针，便于判断字段是否存在
	Code     *int32  `json:"Code,omitempty"`     // 错误码
	ErrMsg   *string `json:"ErrMsg,omitempty"`   // 错误描述
	TaskId   *string `json:"TaskId,omitempty"`   // 任务ID（因前面问题，保持 string 类型）
	TaskType string  `json:"TaskType,omitempty"` // 枚举类型
	WeChatId *string `json:"WeChatId,omitempty"` // 微信号
}