feat: add mx on writing in p

ydb-platform · Jan 16, 2025 · 6c15230 · 6c15230
1 parent a5374f0
commit 6c15230
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/internal/topic/topicwriterinternal/encoders.go b/internal/topic/topicwriterinternal/encoders.go
@@ -44,7 +44,9 @@ func newEncoderPool() *encoderPool {
 
 type EncoderMap struct {
 	m map[rawtopiccommon.Codec]PublicCreateEncoderFunc
-	p map[rawtopiccommon.Codec]*encoderPool
+
+	mx sync.Mutex
+	p  map[rawtopiccommon.Codec]*encoderPool
 }
 
 func NewEncoderMap() *EncoderMap {
@@ -57,7 +59,8 @@ func NewEncoderMap() *EncoderMap {
 				return gzip.NewWriter(writer), nil
 			},
 		},
-		p: make(map[rawtopiccommon.Codec]*encoderPool),
+		mx: sync.Mutex{},
+		p:  make(map[rawtopiccommon.Codec]*encoderPool),
 	}
 }
 
@@ -82,7 +85,9 @@ func (e *EncoderMap) Encode(codec rawtopiccommon.Codec, target io.Writer, data [
 	resetableEnc, ok := enc.(PublicResetableWriter)
 	if ok {
 		if _, ok := e.p[codec]; !ok {
+			e.mx.Lock()
 			e.p[codec] = newEncoderPool()
+			e.mx.Unlock()
 		}
 
 		e.p[codec].Put(resetableEnc)