chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）
This commit is contained in:
MiFaZhan
@@ -0,0 +1,158 @@
|
||||
package trafficusage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"math"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
||||
)
|
||||
|
||||
type UpstreamClient interface {
|
||||
QuerySimDatausageDaily(context.Context, *model.SimDataUsageDailyInput) ([]model.SimBaseRes[model.SimDataUsageDailyOutputItem], error)
|
||||
QuerySimDatausageMonthly(context.Context, *model.SimDataUsageMonthlyInput) ([]model.SimBaseRes[model.SimDataUsageMonthlyOutputItem], error)
|
||||
}
|
||||
|
||||
type Repository interface {
|
||||
UpdateDaily(context.Context, []model.TrafficUsageDailyItem) error
|
||||
UpdateMonthly(context.Context, []model.TrafficUsageMonthlyItem) error
|
||||
}
|
||||
|
||||
type Service struct {
|
||||
upstream UpstreamClient
|
||||
repository Repository
|
||||
}
|
||||
|
||||
type Failure struct {
|
||||
Cause error
|
||||
UsageMb *float64
|
||||
Permanent bool
|
||||
}
|
||||
|
||||
func (failure *Failure) Error() string { return failure.Cause.Error() }
|
||||
func (failure *Failure) Unwrap() error { return failure.Cause }
|
||||
|
||||
func New(upstream UpstreamClient, repository Repository) *Service {
|
||||
return &Service{upstream: upstream, repository: repository}
|
||||
}
|
||||
|
||||
var usageLocation = time.FixedZone("UTC+8", 8*60*60)
|
||||
|
||||
func validateMessage(simKey *model.SimKey, card model.TrafficUsageSimCard, queryDate, layout string) (time.Time, error) {
|
||||
if simKey == nil || card.Iccid == "" || simKey.ICCID != card.Iccid {
|
||||
return time.Time{}, &Failure{Cause: gerror.New("invalid traffic usage card identity"), Permanent: true}
|
||||
}
|
||||
date, err := time.ParseInLocation(layout, queryDate, usageLocation)
|
||||
if err != nil {
|
||||
return time.Time{}, &Failure{Cause: gerror.Wrap(err, "invalid traffic usage period"), Permanent: true}
|
||||
}
|
||||
return date, nil
|
||||
}
|
||||
|
||||
func validateUsage(usage float64) error {
|
||||
if math.IsNaN(usage) || math.IsInf(usage, 0) || usage < 0 {
|
||||
return &Failure{Cause: gerror.New("invalid traffic usage amount"), Permanent: true}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) RetryDaily(ctx context.Context, message model.TrafficUsageDailyRetryMessage) error {
|
||||
date, err := validateMessage(message.SimKey, message.Card, message.QueryDate, "20060102")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
usage := message.UsageMb
|
||||
if usage == nil {
|
||||
output, err := s.upstream.QuerySimDatausageDaily(ctx, &model.SimDataUsageDailyInput{
|
||||
SimKeys: []*model.SimKey{message.SimKey},
|
||||
ReqData: model.SimDataUsageDailyInputItem{QueryDate: message.QueryDate},
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if len(output) != 1 || output[0].SimKey == nil || output[0].SimKey.ICCID != message.Card.Iccid {
|
||||
return gerror.New("daily usage response missing, failed or card identity mismatched")
|
||||
}
|
||||
if output[0].Code != 0 {
|
||||
return &Failure{Cause: gerror.Newf("daily usage query failed: %s", output[0].Message), Permanent: isPermanentUsageResponse(output[0].Message)}
|
||||
}
|
||||
if queryDate, ok := normalizeDailyQueryDate(output[0].ResData.QueryDate); !ok || queryDate != message.QueryDate {
|
||||
return gerror.New("daily usage response period mismatched")
|
||||
}
|
||||
usage = &output[0].ResData.DataAmount
|
||||
}
|
||||
if err := validateUsage(*usage); err != nil {
|
||||
return err
|
||||
}
|
||||
err = s.repository.UpdateDaily(ctx, []model.TrafficUsageDailyItem{{
|
||||
TrafficUsageBase: model.TrafficUsageBase{Iccid: message.Card.Iccid, UsageMb: *usage, PackageCode: message.Card.PackageCode}, Date: date,
|
||||
}})
|
||||
if err != nil {
|
||||
return &Failure{Cause: err, UsageMb: usage}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) RetryMonthly(ctx context.Context, message model.TrafficUsageMonthlyRetryMessage) error {
|
||||
month, err := validateMessage(message.SimKey, message.Card, message.QueryDate, "200601")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
usage := message.UsageMb
|
||||
if usage == nil {
|
||||
output, err := s.upstream.QuerySimDatausageMonthly(ctx, &model.SimDataUsageMonthlyInput{
|
||||
SimKeys: []*model.SimKey{message.SimKey},
|
||||
ReqData: model.SimDataUsageMonthlyInputItem{QueryDate: message.QueryDate},
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if len(output) != 1 || output[0].SimKey == nil || output[0].SimKey.ICCID != message.Card.Iccid {
|
||||
return gerror.New("monthly usage response missing, failed or card identity mismatched")
|
||||
}
|
||||
if output[0].Code != 0 {
|
||||
return &Failure{Cause: gerror.Newf("monthly usage query failed: %s", output[0].Message), Permanent: isPermanentUsageResponse(output[0].Message)}
|
||||
}
|
||||
if queryMonth, ok := normalizeMonthlyQueryDate(output[0].ResData.DeadLine); !ok || queryMonth != message.QueryDate {
|
||||
return gerror.New("monthly usage response period mismatched")
|
||||
}
|
||||
usage = &output[0].ResData.DataAmount
|
||||
}
|
||||
if err := validateUsage(*usage); err != nil {
|
||||
return err
|
||||
}
|
||||
err = s.repository.UpdateMonthly(ctx, []model.TrafficUsageMonthlyItem{{
|
||||
TrafficUsageBase: model.TrafficUsageBase{Iccid: message.Card.Iccid, UsageMb: *usage, PackageCode: message.Card.PackageCode}, Month: month,
|
||||
}})
|
||||
if err != nil {
|
||||
return &Failure{Cause: err, UsageMb: usage}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func normalizeDailyQueryDate(value string) (string, bool) {
|
||||
for _, layout := range []string{"20060102", "2006-01-02"} {
|
||||
parsed, err := time.ParseInLocation(layout, value, usageLocation)
|
||||
if err == nil {
|
||||
return parsed.Format("20060102"), true
|
||||
}
|
||||
}
|
||||
return "", false
|
||||
}
|
||||
|
||||
func normalizeMonthlyQueryDate(value string) (string, bool) {
|
||||
for _, layout := range []string{"200601", "2006-01", "20060102", "2006-01-02"} {
|
||||
parsed, err := time.ParseInLocation(layout, value, usageLocation)
|
||||
if err == nil {
|
||||
return parsed.Format("200601"), true
|
||||
}
|
||||
}
|
||||
return "", false
|
||||
}
|
||||
|
||||
func isPermanentUsageResponse(message string) bool {
|
||||
message = strings.ToLower(strings.TrimSpace(message))
|
||||
return strings.Contains(message, "unsupported") || strings.Contains(message, "does not support") || strings.Contains(message, "not supported")
|
||||
}
|
||||
@@ -0,0 +1,229 @@
|
||||
package trafficusage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
|
||||
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
||||
)
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
g.Log().SetWriter(io.Discard)
|
||||
os.Exit(m.Run())
|
||||
}
|
||||
|
||||
type upstreamStub struct {
|
||||
dailyCalls int
|
||||
monthlyCalls int
|
||||
dailyOutput []model.SimBaseRes[model.SimDataUsageDailyOutputItem]
|
||||
monthlyOutput []model.SimBaseRes[model.SimDataUsageMonthlyOutputItem]
|
||||
dailyErr error
|
||||
monthlyErr error
|
||||
}
|
||||
|
||||
func (s *upstreamStub) QuerySimDatausageDaily(_ context.Context, _ *model.SimDataUsageDailyInput) ([]model.SimBaseRes[model.SimDataUsageDailyOutputItem], error) {
|
||||
s.dailyCalls++
|
||||
return s.dailyOutput, s.dailyErr
|
||||
}
|
||||
|
||||
func (s *upstreamStub) QuerySimDatausageMonthly(_ context.Context, _ *model.SimDataUsageMonthlyInput) ([]model.SimBaseRes[model.SimDataUsageMonthlyOutputItem], error) {
|
||||
s.monthlyCalls++
|
||||
return s.monthlyOutput, s.monthlyErr
|
||||
}
|
||||
|
||||
type repositoryStub struct {
|
||||
dailyCalls int
|
||||
monthlyCalls int
|
||||
dailyItems []model.TrafficUsageDailyItem
|
||||
monthlyItems []model.TrafficUsageMonthlyItem
|
||||
dailyErr error
|
||||
monthlyErr error
|
||||
}
|
||||
|
||||
func (s *repositoryStub) UpdateDaily(_ context.Context, items []model.TrafficUsageDailyItem) error {
|
||||
s.dailyCalls++
|
||||
s.dailyItems = items
|
||||
return s.dailyErr
|
||||
}
|
||||
|
||||
func (s *repositoryStub) UpdateMonthly(_ context.Context, items []model.TrafficUsageMonthlyItem) error {
|
||||
s.monthlyCalls++
|
||||
s.monthlyItems = items
|
||||
return s.monthlyErr
|
||||
}
|
||||
|
||||
func TestRetryDailySuccess(t *testing.T) {
|
||||
upstream := &upstreamStub{
|
||||
dailyOutput: []model.SimBaseRes[model.SimDataUsageDailyOutputItem]{
|
||||
{
|
||||
SimKey: &model.SimKey{ICCID: "89860001"},
|
||||
ResData: model.SimDataUsageDailyOutputItem{QueryDate: "20260714", DataAmount: 12.5},
|
||||
},
|
||||
},
|
||||
}
|
||||
repository := &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
message := model.TrafficUsageDailyRetryMessage{
|
||||
SimKey: &model.SimKey{ICCID: "89860001"},
|
||||
Card: model.TrafficUsageSimCard{Iccid: "89860001", PackageCode: "PKG-1"},
|
||||
QueryDate: "20260714",
|
||||
}
|
||||
|
||||
if err := service.RetryDaily(context.Background(), message); err != nil {
|
||||
t.Fatalf("RetryDaily returned error: %v", err)
|
||||
}
|
||||
if upstream.dailyCalls != 1 || repository.dailyCalls != 1 {
|
||||
t.Fatalf("unexpected calls: upstream=%d repository=%d", upstream.dailyCalls, repository.dailyCalls)
|
||||
}
|
||||
if len(repository.dailyItems) != 1 {
|
||||
t.Fatalf("expected one daily item, got %d", len(repository.dailyItems))
|
||||
}
|
||||
item := repository.dailyItems[0]
|
||||
if item.Iccid != "89860001" || item.UsageMb != 12.5 || item.PackageCode != "PKG-1" {
|
||||
t.Fatalf("unexpected daily item: %+v", item)
|
||||
}
|
||||
if got := item.Date.Format("20060102"); got != "20260714" {
|
||||
t.Fatalf("unexpected daily date: %s", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRetryDailyPreservesRetryBehavior(t *testing.T) {
|
||||
t.Run("period mismatch", func(t *testing.T) {
|
||||
upstream := &upstreamStub{dailyOutput: []model.SimBaseRes[model.SimDataUsageDailyOutputItem]{{SimKey: &model.SimKey{ICCID: "89860001"}, ResData: model.SimDataUsageDailyOutputItem{QueryDate: "20260713", DataAmount: 1}}}}
|
||||
repository := &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
if err := service.RetryDaily(context.Background(), validDailyMessage()); err == nil {
|
||||
t.Fatal("expected period mismatch failure")
|
||||
}
|
||||
if repository.dailyCalls != 0 {
|
||||
t.Fatal("period mismatch was persisted")
|
||||
}
|
||||
})
|
||||
t.Run("unsupported channel is permanent", func(t *testing.T) {
|
||||
upstream := &upstreamStub{dailyOutput: []model.SimBaseRes[model.SimDataUsageDailyOutputItem]{{SimKey: &model.SimKey{ICCID: "89860001"}, Code: 1, Message: "daily usage query is not supported"}}}
|
||||
service := New(upstream, &repositoryStub{})
|
||||
err := service.RetryDaily(context.Background(), validDailyMessage())
|
||||
var failure *Failure
|
||||
if !errors.As(err, &failure) || !failure.Permanent {
|
||||
t.Fatalf("expected permanent failure, got %v", err)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("upstream failure", func(t *testing.T) {
|
||||
upstream := &upstreamStub{dailyErr: errors.New("upstream failed")}
|
||||
repository := &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
|
||||
if err := service.RetryDaily(context.Background(), validDailyMessage()); err == nil {
|
||||
t.Fatal("expected retryable failure")
|
||||
}
|
||||
if upstream.dailyCalls != 1 || repository.dailyCalls != 0 {
|
||||
t.Fatalf("unexpected calls: upstream=%d repository=%d", upstream.dailyCalls, repository.dailyCalls)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("repository failure", func(t *testing.T) {
|
||||
upstream := &upstreamStub{dailyOutput: validDailyOutput()}
|
||||
repository := &repositoryStub{dailyErr: errors.New("repository failed")}
|
||||
service := New(upstream, repository)
|
||||
|
||||
if err := service.RetryDaily(context.Background(), validDailyMessage()); err == nil {
|
||||
t.Fatal("expected persistence failure")
|
||||
}
|
||||
if upstream.dailyCalls != 1 || repository.dailyCalls != 1 {
|
||||
t.Fatalf("unexpected calls: upstream=%d repository=%d", upstream.dailyCalls, repository.dailyCalls)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("invalid date", func(t *testing.T) {
|
||||
upstream := &upstreamStub{}
|
||||
repository := &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
message := validDailyMessage()
|
||||
message.QueryDate = "invalid"
|
||||
|
||||
if err := service.RetryDaily(context.Background(), message); err == nil {
|
||||
t.Fatal("expected invalid date failure")
|
||||
}
|
||||
if upstream.dailyCalls != 0 || repository.dailyCalls != 0 {
|
||||
t.Fatalf("invalid date reached dependencies")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestRetryMonthlySuccessAndRetry(t *testing.T) {
|
||||
t.Run("success", func(t *testing.T) {
|
||||
upstream := &upstreamStub{
|
||||
monthlyOutput: []model.SimBaseRes[model.SimDataUsageMonthlyOutputItem]{
|
||||
{
|
||||
SimKey: &model.SimKey{ICCID: "89860002"},
|
||||
ResData: model.SimDataUsageMonthlyOutputItem{DeadLine: "202607", DataAmount: 25.5},
|
||||
},
|
||||
},
|
||||
}
|
||||
repository := &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
message := model.TrafficUsageMonthlyRetryMessage{
|
||||
SimKey: &model.SimKey{ICCID: "89860002"},
|
||||
Card: model.TrafficUsageSimCard{Iccid: "89860002", PackageCode: "PKG-2"},
|
||||
QueryDate: "202607",
|
||||
}
|
||||
|
||||
if err := service.RetryMonthly(context.Background(), message); err != nil {
|
||||
t.Fatalf("RetryMonthly returned error: %v", err)
|
||||
}
|
||||
if upstream.monthlyCalls != 1 || repository.monthlyCalls != 1 {
|
||||
t.Fatalf("unexpected calls: upstream=%d repository=%d", upstream.monthlyCalls, repository.monthlyCalls)
|
||||
}
|
||||
if len(repository.monthlyItems) != 1 {
|
||||
t.Fatalf("expected one monthly item, got %d", len(repository.monthlyItems))
|
||||
}
|
||||
item := repository.monthlyItems[0]
|
||||
if item.Iccid != "89860002" || item.UsageMb != 25.5 || item.PackageCode != "PKG-2" {
|
||||
t.Fatalf("unexpected monthly item: %+v", item)
|
||||
}
|
||||
if got := item.Month.Format("200601"); got != "202607" {
|
||||
t.Fatalf("unexpected month: %s", got)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("upstream failure", func(t *testing.T) {
|
||||
upstream := &upstreamStub{monthlyErr: errors.New("upstream failed")}
|
||||
repository := &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
message := model.TrafficUsageMonthlyRetryMessage{
|
||||
SimKey: &model.SimKey{ICCID: "89860002"},
|
||||
Card: model.TrafficUsageSimCard{Iccid: "89860002"},
|
||||
QueryDate: "202607",
|
||||
}
|
||||
|
||||
if err := service.RetryMonthly(context.Background(), message); err == nil {
|
||||
t.Fatal("expected retryable failure")
|
||||
}
|
||||
if upstream.monthlyCalls != 1 || repository.monthlyCalls != 0 {
|
||||
t.Fatalf("unexpected calls: upstream=%d repository=%d", upstream.monthlyCalls, repository.monthlyCalls)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func validDailyMessage() model.TrafficUsageDailyRetryMessage {
|
||||
return model.TrafficUsageDailyRetryMessage{
|
||||
SimKey: &model.SimKey{ICCID: "89860001"},
|
||||
Card: model.TrafficUsageSimCard{Iccid: "89860001"},
|
||||
QueryDate: "20260714",
|
||||
}
|
||||
}
|
||||
|
||||
func validDailyOutput() []model.SimBaseRes[model.SimDataUsageDailyOutputItem] {
|
||||
return []model.SimBaseRes[model.SimDataUsageDailyOutputItem]{
|
||||
{
|
||||
SimKey: &model.SimKey{ICCID: "89860001"},
|
||||
ResData: model.SimDataUsageDailyOutputItem{QueryDate: "20260714", DataAmount: 1},
|
||||
},
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,54 @@
|
||||
package trafficusage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestPersistenceFailureRetainsSnapshot(t *testing.T) {
|
||||
upstream := &upstreamStub{dailyOutput: validDailyOutput()}
|
||||
repository := &repositoryStub{dailyErr: errors.New("database offline")}
|
||||
service := New(upstream, repository)
|
||||
message := validDailyMessage()
|
||||
var failure *Failure
|
||||
if err := service.RetryDaily(context.Background(), message); !errors.As(err, &failure) || failure.UsageMb == nil || *failure.UsageMb != 1 {
|
||||
t.Fatalf("snapshot missing: %v", err)
|
||||
}
|
||||
message.UsageMb = failure.UsageMb
|
||||
repository.dailyErr = nil
|
||||
if err := service.RetryDaily(context.Background(), message); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if upstream.dailyCalls != 1 || repository.dailyCalls != 2 {
|
||||
t.Fatal("persistence retry queried upstream again")
|
||||
}
|
||||
}
|
||||
|
||||
func TestZeroSnapshotBypassesUpstream(t *testing.T) {
|
||||
upstream, repository := &upstreamStub{}, &repositoryStub{}
|
||||
service := New(upstream, repository)
|
||||
zero := 0.0
|
||||
message := validDailyMessage()
|
||||
message.UsageMb = &zero
|
||||
if err := service.RetryDaily(context.Background(), message); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
monthly := model.TrafficUsageMonthlyRetryMessage{SimKey: message.SimKey, Card: message.Card, QueryDate: "202607", UsageMb: &zero}
|
||||
if err := service.RetryMonthly(context.Background(), monthly); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if upstream.dailyCalls != 0 || upstream.monthlyCalls != 0 || repository.dailyCalls != 1 || repository.monthlyCalls != 1 {
|
||||
t.Fatal("zero snapshot was not persisted directly")
|
||||
}
|
||||
}
|
||||
|
||||
func TestRejectMismatchedOrMissingResult(t *testing.T) {
|
||||
for _, output := range [][]model.SimBaseRes[model.SimDataUsageDailyOutputItem]{nil, {{}}, {{SimKey: &model.SimKey{ICCID: "wrong"}}}, {validDailyOutput()[0], validDailyOutput()[0]}} {
|
||||
upstream, repository := &upstreamStub{dailyOutput: output}, &repositoryStub{}
|
||||
if err := New(upstream, repository).RetryDaily(context.Background(), validDailyMessage()); err == nil || repository.dailyCalls != 0 {
|
||||
t.Fatal("invalid result accepted")
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user