feat(backend): refactor to avoid data race

Author: kidkidkid

backend/modules/observability/domain/trace/entity/loop_span/trans.go

@@ -6,6 +6,7 @@ package loop_span
 import (
 	"context"
 	"encoding/json"
+	"sync"
 	"time"
 
 	"github.com/coze-dev/coze-loop/backend/pkg/lang/slices"
@@ -38,6 +39,7 @@ type SpanTransConfig struct {
 	TagFilter    *TagFilter    `mapstructure:"tag_filter" json:"tag_filter"`
 	InputFilter  *InputFilter  `mapstructure:"input_filter" json:"input_filter"`
 	OutputFilter *OutputFilter `mapstructure:"output_filter" json:"output_filter"`
+	initOnce     sync.Once
 }
 
 type TagFilter struct {
@@ -128,15 +130,17 @@ func (c *OutputFilter) transform(ctx context.Context, span *Span) {
 }
 
 func (c *SpanTransConfig) init() {
-	if c.TagFilter != nil {
-		c.TagFilter.keyBlackListMap = slices.ToMap(c.TagFilter.KeyBlackList, func(e string) (string, bool) { return e, true })
-	}
-	if c.InputFilter != nil {
-		c.InputFilter.keyWhiteListMap = slices.ToMap(c.InputFilter.KeyWhiteList, func(e string) (string, bool) { return e, true })
-	}
-	if c.OutputFilter != nil {
-		c.OutputFilter.keyWhiteListMap = slices.ToMap(c.OutputFilter.KeyWhiteList, func(e string) (string, bool) { return e, true })
-	}
+	c.initOnce.Do(func() {
+		if c.TagFilter != nil {
+			c.TagFilter.keyBlackListMap = slices.ToMap(c.TagFilter.KeyBlackList, func(e string) (string, bool) { return e, true })
+		}
+		if c.InputFilter != nil {
+			c.InputFilter.keyWhiteListMap = slices.ToMap(c.InputFilter.KeyWhiteList, func(e string) (string, bool) { return e, true })
+		}
+		if c.OutputFilter != nil {
+			c.OutputFilter.keyWhiteListMap = slices.ToMap(c.OutputFilter.KeyWhiteList, func(e string) (string, bool) { return e, true })
+		}
+	})
 }
 
 func (c *SpanTransConfig) satisfyFilter(span *Span) bool {

backend/modules/observability/domain/trace/entity/loop_span/trans_test.go

@@ -8,6 +8,7 @@ import (
 	"encoding/json"
 	"math/rand"
 	"sort"
+	"sync"
 	"testing"
 	"time"
 
@@ -520,3 +521,26 @@ func TestParentIdRedirectChaos(t *testing.T) {
 		}
 	}
 }
+
+func TestRace(t *testing.T) {
+	p := SpanTransCfgList{
+		{
+			SpanFilter: &FilterFields{},
+			TagFilter: &TagFilter{
+				KeyBlackList: []string{
+					"custom_a",
+					"custom_c",
+				},
+			},
+		},
+	}
+	var wg sync.WaitGroup
+	for i := 0; i < 100; i++ {
+		wg.Add(1)
+		go func() {
+			p.init()
+			wg.Done()
+		}()
+	}
+	wg.Wait()
+}