ark-member/internal/logic/update_status_logic.go

package logic

import (
	"context"
	"member/internal/domain"
	ers "member/internal/lib/error"
	"member/internal/lib/required"

	"member/gen_result/pb/member"
	"member/internal/svc"

	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateStatusLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewUpdateStatusLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateStatusLogic {
	return &UpdateStatusLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

type updateStatusReq struct {
	UID    string `json:"uid" validate:"required"`
	Status int32  `json:"status" validate:"required,oneof=1 2 3 4 5 6"`
}

// UpdateStatus 修改狀態
func (l *UpdateStatusLogic) UpdateStatus(in *member.UpdateStatusReq) (*member.Response, error) {
	err := required.ValidateAll(l.svcCtx.Validate, &updateStatusReq{
		UID:    in.GetUid(),
		Status: int32(in.GetStatus()),
	})
	if err != nil {
		return nil, ers.InvalidFormat(err.Error())
	}

	err = l.svcCtx.UserModel.UpdateStatus(l.ctx, in.GetUid(), int32(in.GetStatus()))
	if err != nil {
		return nil, ers.DBError(err.Error())
	}

	return &member.Response{
		Status: &member.BaseResp{
			Code:    domain.CodeOk.ToString(),
			Message: "success",
			Error:   "",
		},
	}, nil
}