MiFaZhan
52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package grpcapi
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-sms/api/pb"
|
|
"git.5iot.com/iotlcm-public/iotlcm-sms/internal/sms"
|
|
)
|
|
|
|
type Sender interface {
|
|
Send(context.Context, sms.Config, sms.Message) (sms.Result, error)
|
|
}
|
|
|
|
type Server struct {
|
|
pb.UnimplementedNoticeServer
|
|
sender Sender
|
|
}
|
|
|
|
func NewServer(sender Sender) *Server {
|
|
return &Server{sender: sender}
|
|
}
|
|
|
|
func (s *Server) Send(ctx context.Context, request *pb.SendReq) (*pb.SendRes, error) {
|
|
if request == nil {
|
|
return nil, status.Error(codes.InvalidArgument, "请求不能为空")
|
|
}
|
|
|
|
var config sms.Config
|
|
if err := json.Unmarshal(request.GetConfig(), &config); err != nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "config JSON 无效: %v", err)
|
|
}
|
|
var message sms.Message
|
|
if err := json.Unmarshal(request.GetParams(), &message); err != nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "params JSON 无效: %v", err)
|
|
}
|
|
|
|
result, _ := s.sender.Send(ctx, config, message)
|
|
responseHead := &pb.HeadRes{
|
|
Code: int32(result.Code),
|
|
Message: result.Message,
|
|
}
|
|
if request.Head != nil {
|
|
responseHead.RouterId = request.Head.RouterId
|
|
responseHead.TraceId = request.Head.TraceId
|
|
}
|
|
return &pb.SendRes{Head: responseHead}, nil
|
|
}
|