Skip to content

Commit 9022ebf

Browse files
committed
zstd lib update
1 parent fe27dc1 commit 9022ebf

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

internal/storage/kafka_publisher.go

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ import (
1010
"sync"
1111
"time"
1212

13-
"github.com/DataDog/zstd"
13+
"github.com/klauspost/compress/zstd"
1414
"github.com/rs/zerolog/log"
1515
config "github.com/thirdweb-dev/indexer/configs"
1616
"github.com/thirdweb-dev/indexer/internal/common"
@@ -308,10 +308,13 @@ func (p *KafkaPublisher) createBlockRevertMessage(chainId uint64, blockNumber ui
308308

309309
func (p *KafkaPublisher) createRecord(msgType MessageType, chainId uint64, blockNumber uint64, timestamp time.Time, msgJson []byte) (*kgo.Record, error) {
310310
// Compress the JSON data using zstd
311-
compressedData, err := zstd.Compress(nil, msgJson)
311+
encoder, err := zstd.NewWriter(nil)
312312
if err != nil {
313-
return nil, fmt.Errorf("failed to compress message data: %v", err)
313+
return nil, fmt.Errorf("failed to create zstd encoder: %v", err)
314314
}
315+
defer encoder.Close()
316+
317+
compressedData := encoder.EncodeAll(msgJson, nil)
315318

316319
// Create headers with metadata including compression info
317320
headers := []kgo.RecordHeader{

0 commit comments

Comments
 (0)