chore: 并入 iotlcm-sms 历史到 sms/(原提交 1d38bc4)

This commit is contained in:
MiFaZhan
2026-11-02 13:28:02 +08:00
34 changed files with 2776 additions and 0 deletions
+71
View File
@@ -0,0 +1,71 @@
package httpapi
import (
"context"
"encoding/json"
"errors"
"io"
"net/http"
"git.5iot.com/iotlcm-public/iotlcm-sms/internal/sms"
)
const maxRequestBody = 1 << 20
type Sender interface {
Send(ctx context.Context, config sms.Config, message sms.Message) (sms.Result, error)
}
type SendRequest struct {
Config sms.Config `json:"config"`
Params sms.Message `json:"params"`
}
type handler struct {
sender Sender
}
func NewHandler(sender Sender) http.Handler {
h := &handler{sender: sender}
mux := http.NewServeMux()
mux.HandleFunc("GET /healthz", h.health)
mux.HandleFunc("POST /api/v1/sms/send", h.send)
return mux
}
func (h *handler) health(w http.ResponseWriter, _ *http.Request) {
writeJSON(w, http.StatusOK, map[string]string{"status": "ok"})
}
func (h *handler) send(w http.ResponseWriter, r *http.Request) {
r.Body = http.MaxBytesReader(w, r.Body, maxRequestBody)
decoder := json.NewDecoder(r.Body)
decoder.DisallowUnknownFields()
var request SendRequest
if err := decoder.Decode(&request); err != nil {
writeJSON(w, http.StatusBadRequest, sms.Result{Code: 400, Message: "请求 JSON 无效: " + err.Error()})
return
}
if err := decoder.Decode(&struct{}{}); !errors.Is(err, io.EOF) {
writeJSON(w, http.StatusBadRequest, sms.Result{Code: 400, Message: "请求体只能包含一个 JSON 对象"})
return
}
result, err := h.sender.Send(r.Context(), request.Config, request.Params)
if err != nil {
status := http.StatusBadGateway
if errors.Is(err, sms.ErrInvalidRequest) || errors.Is(err, sms.ErrProviderNotFound) || errors.Is(err, sms.ErrSceneNotConfigured) {
status = http.StatusBadRequest
}
writeJSON(w, status, result)
return
}
writeJSON(w, http.StatusOK, result)
}
func writeJSON(w http.ResponseWriter, status int, value any) {
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.WriteHeader(status)
_ = json.NewEncoder(w).Encode(value)
}
+77
View File
@@ -0,0 +1,77 @@
package httpapi
import (
"context"
"encoding/json"
"errors"
"net/http"
"net/http/httptest"
"strings"
"testing"
"git.5iot.com/iotlcm-public/iotlcm-sms/internal/sms"
)
type fakeSender struct {
config sms.Config
message sms.Message
result sms.Result
err error
}
func (f *fakeSender) Send(_ context.Context, config sms.Config, message sms.Message) (sms.Result, error) {
f.config = config
f.message = message
return f.result, f.err
}
func TestSendSuccess(t *testing.T) {
fake := &fakeSender{result: sms.Result{Code: 0, Message: "发送成功"}}
server := httptest.NewServer(NewHandler(fake))
defer server.Close()
body := `{"config":{"serviceType":"1","accessKeyId":"id","accessKeySecret":"secret","signName":"测试签名","templateId":"SMS_1"},"params":{"phoneNumber":"13800008000","msgContent":"{\"code\":\"123456\"}"}}`
response, err := http.Post(server.URL+"/api/v1/sms/send", "application/json", strings.NewReader(body))
if err != nil {
t.Fatal(err)
}
defer response.Body.Close()
if response.StatusCode != http.StatusOK {
t.Fatalf("status = %d, want %d", response.StatusCode, http.StatusOK)
}
if fake.config.ServiceType != sms.ServiceAliyun || fake.message.PhoneNumber != "13800008000" {
t.Fatalf("unexpected request: config=%+v message=%+v", fake.config, fake.message)
}
}
func TestSendRejectsInvalidJSON(t *testing.T) {
request := httptest.NewRequest(http.MethodPost, "/api/v1/sms/send", strings.NewReader(`{"unknown":true}`))
recorder := httptest.NewRecorder()
NewHandler(&fakeSender{}).ServeHTTP(recorder, request)
if recorder.Code != http.StatusBadRequest {
t.Fatalf("status = %d, want %d", recorder.Code, http.StatusBadRequest)
}
}
func TestSendMapsProviderError(t *testing.T) {
fake := &fakeSender{
result: sms.Result{Code: 502, Message: "provider failed"},
err: errors.New("provider failed"),
}
request := httptest.NewRequest(http.MethodPost, "/api/v1/sms/send", strings.NewReader(`{"config":{},"params":{}}`))
recorder := httptest.NewRecorder()
NewHandler(fake).ServeHTTP(recorder, request)
if recorder.Code != http.StatusBadGateway {
t.Fatalf("status = %d, want %d", recorder.Code, http.StatusBadGateway)
}
var result sms.Result
if err := json.NewDecoder(recorder.Body).Decode(&result); err != nil {
t.Fatal(err)
}
if result.Code != 502 {
t.Fatalf("code = %d, want 502", result.Code)
}
}