fix: store cache

cloudwego · Feb 3, 2025 · df96208 · df96208
1 parent 4903274
commit df96208
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/pkg/generic/httpthrift_codec.go b/pkg/generic/httpthrift_codec.go
@@ -123,7 +123,6 @@ func (c *httpThriftCodec) getMessageReaderWriter() interface{} {
 	rw := thrift.NewHTTPReaderWriter(svcDsc)
 	c.configureHTTPRequestWriter(rw.WriteHTTPRequest)
 	c.configureHTTPResponseReader(rw.ReadHTTPResponse)
-
 	newCache := &cachedHTTPReaderWriterData{
 		svcDsc:       svcDsc,
 		readerWriter: rw,

diff --git a/pkg/generic/jsonthrift_codec.go b/pkg/generic/jsonthrift_codec.go
@@ -122,6 +122,11 @@ func (c *jsonThriftCodec) getMessageReaderWriter() interface{} {
 	rw := thrift.NewJsonReaderWriter(svcDsc)
 	c.configureJSONWriter(rw.WriteJSON)
 	c.configureJSONReader(rw.ReadJSON)
+	newCache := &cachedJSONReaderWriterData{
+		svcDsc:       svcDsc,
+		readerWriter: rw,
+	}
+	c.cachedReaderWriterData.Store(newCache)
 	return rw
 }
 

diff --git a/pkg/generic/mapthrift_codec.go b/pkg/generic/mapthrift_codec.go
@@ -113,6 +113,11 @@ func (c *mapThriftCodec) getMessageReaderWriter() interface{} {
 	}
 	c.configureStructWriter(rw.WriteStruct)
 	c.configureStructReader(rw.ReadStruct)
+	newCache := &cachedStructReaderWriter{
+		svcDsc:       svcDsc,
+		readerWriter: rw,
+	}
+	c.cachedReaderWriterData.Store(newCache)
 	return rw
 }