MiFaZhan
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package rabbitmq
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
mq "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/library/rabbitmq"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/service/trafficusage"
|
|
)
|
|
|
|
var trafficUsagePublish = mq.PublishBytesWithHeaders
|
|
var trafficUsagePublishDelayed = mq.PublishDelayedBytesWithHeaders
|
|
|
|
func validateTrafficUsageIdentity(simKey *model.SimKey, card model.TrafficUsageSimCard) error {
|
|
if simKey == nil || card.Iccid == "" || simKey.ICCID != card.Iccid {
|
|
return &trafficusage.Failure{Cause: gerror.New("invalid traffic usage card identity"), Permanent: true}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func trafficUsageFailureSnapshot(err error, usage **float64) bool {
|
|
var failure *trafficusage.Failure
|
|
if errors.As(err, &failure) {
|
|
if failure.UsageMb != nil {
|
|
*usage = failure.UsageMb
|
|
}
|
|
return failure.Permanent
|
|
}
|
|
return false
|
|
}
|
|
|
|
func publishTrafficUsageFailure(ctx context.Context, queue string, message any, attempt int, permanent bool) error {
|
|
ctx, cancel := context.WithTimeout(ctx, 15*time.Second)
|
|
defer cancel()
|
|
body, err := json.Marshal(message)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if permanent || attempt >= 10 {
|
|
g.Log().Errorf(ctx, "traffic usage retry exhausted or invalid, queue:%s, attempt:%d, payload:%s", queue, attempt, body)
|
|
err = trafficUsagePublish(ctx, queue+".dlq", body, nil)
|
|
return waitTrafficUsagePublishFailure(ctx, err)
|
|
}
|
|
delay := 30 * time.Second
|
|
if attempt >= 3 {
|
|
delay = 2 * time.Minute
|
|
}
|
|
if attempt >= 6 {
|
|
delay = 5 * time.Minute
|
|
}
|
|
err = trafficUsagePublishDelayed(ctx, queue+".delay."+delay.String(), queue, delay, body, nil)
|
|
return waitTrafficUsagePublishFailure(ctx, err)
|
|
}
|
|
|
|
func waitTrafficUsagePublishFailure(ctx context.Context, err error) error {
|
|
if err != nil {
|
|
select {
|
|
case <-ctx.Done():
|
|
case <-time.After(time.Second):
|
|
}
|
|
}
|
|
return err
|
|
}
|