From 3b9bbe43ad0fa322ac7bf1cc072d854cdc14c712 Mon Sep 17 00:00:00 2001 From: FogDong Date: Tue, 13 Dec 2022 19:49:05 +0800 Subject: [PATCH] test Signed-off-by: FogDong --- pkg/backup/sls/producer.go | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/pkg/backup/sls/producer.go b/pkg/backup/sls/producer.go index 6531a82..0e66600 100644 --- a/pkg/backup/sls/producer.go +++ b/pkg/backup/sls/producer.go @@ -13,9 +13,9 @@ import ( // Handler is sls config. type Handler struct { - LogStoreName string - ProjectName string - Producer *producer.Producer + LogStoreName string + ProjectName string + ProducerConfig *producer.ProducerConfig } type Callback struct { @@ -38,9 +38,9 @@ func NewSLSHandler(config map[string][]byte) (*Handler, error) { producerConfig.AccessKeySecret = accessKeySecret return &Handler{ - Producer: producer.InitProducer(producerConfig), - LogStoreName: logStoreName, - ProjectName: projectName, + ProducerConfig: producerConfig, + LogStoreName: logStoreName, + ProjectName: projectName, }, nil } @@ -54,8 +54,9 @@ func (callback *Callback) Success(result *producer.Result) { // Store is store workflowRun to sls func (s *Handler) Store(ctx monitorContext.Context, run *v1alpha1.WorkflowRun) error { ctx.Info("Start Send workflow record to SLS") - s.Producer.Start() - defer s.Producer.SafeClose() + p := producer.InitProducer(s.ProducerConfig) + p.Start() + defer p.SafeClose() data, err := json.Marshal(run) if err != nil { @@ -65,7 +66,7 @@ func (s *Handler) Store(ctx monitorContext.Context, run *v1alpha1.WorkflowRun) e callback := &Callback{ctx: ctx} log := producer.GenerateLog(uint32(time.Now().Unix()), map[string]string{"content": string(data)}) - err = s.Producer.SendLogWithCallBack(s.ProjectName, s.LogStoreName, "topic", "", log, callback) + err = p.SendLogWithCallBack(s.ProjectName, s.LogStoreName, "topic", "", log, callback) if err != nil { ctx.Error(err, "Send WorkflowRun Content to SLS fail") return err