46 lines
884 B
Go
46 lines
884 B
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
"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, batchNo string) error {
|
|
|
|
if batchNo == "" {
|
|
s.logHelper.Errorf("wechat query error: batchNo is empty")
|
|
return nil
|
|
}
|
|
|
|
if err := s.VoucherBiz.WechatQuery(ctx, batchNo); err != nil {
|
|
s.logHelper.Error(err)
|
|
}
|
|
|
|
return nil
|
|
}
|