package logic import ( "context" "lone-services/pkg/utils" "lone-services/services/message/internal/dao" "lone-services/services/message/internal/model" "lone-services/rpc/message/pb" "lone-services/services/message/internal/svc" "github.com/zeromicro/go-zero/core/logx" "google.golang.org/grpc/codes" "google.golang.org/grpc/status" ) type MessageAddLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger } func NewMessageAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *MessageAddLogic { return &MessageAddLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *MessageAddLogic) MessageAdd(in *message.AddRequest) (*message.AddResponse, error) { add := dao.MessageCreate{ Content: in.Content, TargetType: in.TargetType, TargetId: in.TargetId, TypeStatus: in.TypeStatus, Type: in.Type, ObjId: in.ObjId, } err := model.MessageModel{}.Init().Create(&add) if err != nil { l.Logger.Error("Failed to create message:", err) return nil, status.Error(codes.Internal, utils.Fail.Msg) } return &message.AddResponse{Num: utils.NumberOne}, nil }