chore: 初始开源发布
iotlcm-sms:独立短信 HTTP/gRPC 服务,支持阿里云与腾讯云。 - Go module 迁移至 git.5iot.com/iotlcm-public/iotlcm-sms - 补 Apache-2.0 LICENSE,重写 README(特性/配置/接口/部署/目录/测试) - 配置改为 config.example.yaml + SMS_* 环境变量注入,仓库不含真实凭证 - 清理内部标识(私有镜像仓库 / 公司名 / 客户签名 / 真实模板ID / 真实手机号) - 统一 LF 行尾并加 .gitattributes
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
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
|
||||
}
|
||||
@@ -0,0 +1,88 @@
|
||||
package grpcapi
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net"
|
||||
"testing"
|
||||
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/codes"
|
||||
"google.golang.org/grpc/credentials/insecure"
|
||||
"google.golang.org/grpc/status"
|
||||
"google.golang.org/grpc/test/bufconn"
|
||||
|
||||
"git.5iot.com/iotlcm-public/iotlcm-sms/api/pb"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-sms/internal/sms"
|
||||
)
|
||||
|
||||
type fakeSender struct {
|
||||
config sms.Config
|
||||
message sms.Message
|
||||
result sms.Result
|
||||
}
|
||||
|
||||
func (f *fakeSender) Send(_ context.Context, config sms.Config, message sms.Message) (sms.Result, error) {
|
||||
f.config = config
|
||||
f.message = message
|
||||
return f.result, nil
|
||||
}
|
||||
|
||||
func TestSend(t *testing.T) {
|
||||
fake := &fakeSender{result: sms.Result{Code: 0, Message: "发送成功"}}
|
||||
client, closeServer := newTestClient(t, fake)
|
||||
defer closeServer()
|
||||
|
||||
config, _ := json.Marshal(sms.Config{ServiceType: sms.ServiceAliyun})
|
||||
params, _ := json.Marshal(sms.Message{PhoneNumber: "138****8000", MsgContent: `{"code":"123456"}`})
|
||||
response, err := client.Send(context.Background(), &pb.SendReq{
|
||||
Head: &pb.HeadReq{RouterId: 42, TraceId: "trace-1"},
|
||||
Config: config,
|
||||
Params: params,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if response.Head.Code != 0 || response.Head.RouterId != 42 || response.Head.TraceId != "trace-1" {
|
||||
t.Fatalf("unexpected response: %+v", response.Head)
|
||||
}
|
||||
if fake.config.ServiceType != sms.ServiceAliyun || fake.message.PhoneNumber != "138****8000" {
|
||||
t.Fatalf("unexpected request: config=%+v message=%+v", fake.config, fake.message)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSendRejectsInvalidJSON(t *testing.T) {
|
||||
client, closeServer := newTestClient(t, &fakeSender{})
|
||||
defer closeServer()
|
||||
|
||||
_, err := client.Send(context.Background(), &pb.SendReq{Config: []byte(`{`), Params: []byte(`{}`)})
|
||||
if status.Code(err) != codes.InvalidArgument {
|
||||
t.Fatalf("code = %v, want %v; err=%v", status.Code(err), codes.InvalidArgument, err)
|
||||
}
|
||||
}
|
||||
|
||||
func newTestClient(t *testing.T, sender Sender) (pb.NoticeClient, func()) {
|
||||
t.Helper()
|
||||
listener := bufconn.Listen(1024 * 1024)
|
||||
server := grpc.NewServer()
|
||||
pb.RegisterNoticeServer(server, NewServer(sender))
|
||||
go func() {
|
||||
_ = server.Serve(listener)
|
||||
}()
|
||||
|
||||
conn, err := grpc.NewClient(
|
||||
"passthrough:///bufnet",
|
||||
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
||||
grpc.WithContextDialer(func(context.Context, string) (net.Conn, error) {
|
||||
return listener.Dial()
|
||||
}),
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
return pb.NewNoticeClient(conn), func() {
|
||||
_ = conn.Close()
|
||||
server.Stop()
|
||||
_ = listener.Close()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user