Open-IM-Server/internal/rpc/msg/msg_status.go

47 lines
1.5 KiB

3 years ago
package msg
import (
"Open_IM/pkg/common/constant"
"Open_IM/pkg/common/db"
"Open_IM/pkg/common/log"
pbMsg "Open_IM/pkg/proto/msg"
"Open_IM/pkg/utils"
"context"
3 years ago
3 years ago
goRedis "github.com/go-redis/redis/v8"
)
3 years ago
func (rpc *rpcChat) SetSendMsgStatus(_ context.Context, req *pbMsg.SetSendMsgStatusReq) (resp *pbMsg.SetSendMsgStatusResp, err error) {
resp = &pbMsg.SetSendMsgStatusResp{}
3 years ago
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), req.String())
3 years ago
if err := db.DB.SetSendMsgStatus(req.Status, req.OperationID); err != nil {
log.NewError(req.OperationID, utils.GetSelfFuncName(), err.Error())
3 years ago
resp.ErrCode = constant.ErrDB.ErrCode
resp.ErrMsg = err.Error()
return resp, nil
}
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), resp.String())
return resp, nil
}
3 years ago
func (rpc *rpcChat) GetSendMsgStatus(_ context.Context, req *pbMsg.GetSendMsgStatusReq) (resp *pbMsg.GetSendMsgStatusResp, err error) {
3 years ago
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), req.String())
3 years ago
resp = &pbMsg.GetSendMsgStatusResp{}
3 years ago
status, err := db.DB.GetSendMsgStatus(req.OperationID)
if err != nil {
resp.Status = constant.MsgStatusNotExist
3 years ago
if err == goRedis.Nil {
3 years ago
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), req.OperationID, "not exist")
3 years ago
return resp, nil
3 years ago
} else {
log.NewError(req.OperationID, utils.GetSelfFuncName(), err.Error())
resp.ErrMsg = err.Error()
resp.ErrCode = constant.ErrDB.ErrCode
3 years ago
return resp, nil
3 years ago
}
}
3 years ago
resp.Status = int32(status)
3 years ago
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), resp.String())
return resp, nil
}