51 lines
1.0 KiB
Go
51 lines
1.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
"time"
|
|
"voucher/internal/pkg/lock"
|
|
"voucher/internal/pkg/rdsmq"
|
|
)
|
|
|
|
func (s *VoucherService) GetConfig() *rdsmq.ConsumeConfig {
|
|
|
|
queue := s.bc.RdsMQ.GetWechatQuery()
|
|
if queue == nil {
|
|
return nil
|
|
}
|
|
|
|
if !queue.GetIsOpen() {
|
|
log.Warn(fmt.Sprintf("[%s]RdsMQ is not open", queue.Name))
|
|
return nil
|
|
}
|
|
|
|
return &rdsmq.ConsumeConfig{
|
|
Rdb: s.rdb.Rdb,
|
|
QueueName: queue.Name,
|
|
NumWorkers: queue.NumWorkers,
|
|
WaitTime: queue.GetWaitTime().AsDuration(),
|
|
RetryNum: queue.RetryNum,
|
|
Fn: s.Handle,
|
|
Logger: s.logHelper,
|
|
}
|
|
}
|
|
|
|
func (s *VoucherService) Handle(ctx context.Context, msg string) error {
|
|
|
|
if msg == "" {
|
|
s.logHelper.Errorf("RdsMQ keySend error: batchNo is empty")
|
|
return nil
|
|
}
|
|
|
|
return lock.NewMutex(s.rdb.Rdb, time.Hour*10).Lock(ctx, fmt.Sprintf("cmb_wechant_query_%s", msg), func(ctx context.Context) error {
|
|
|
|
if err := s.VoucherBiz.WechatQuery(ctx, msg); err != nil {
|
|
s.logHelper.Error(err)
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|