Make the header if it is empty

nats-io · Oct 2, 2024 · f69a1f6 · f69a1f6
1 parent f0c0194
commit f69a1f6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/jetstream/pull.go b/jetstream/pull.go
@@ -205,6 +205,9 @@ func (p *pullConsumer) Consume(handler MessageHandler, opts ...PullConsumeOpt) (
 	p.Unlock()
 
 	internalHandler := func(msg *nats.Msg) {
+		if msg.Header == nil {
+			msg.Header = make(nats.Header)
+		}
 		if sub.hbMonitor != nil {
 			sub.hbMonitor.Stop()
 		}
@@ -781,6 +784,9 @@ func (p *pullConsumer) fetch(req *pullRequest) (MessageBatch, error) {
 			select {
 			case msg := <-msgs:
 				p.Lock()
+				if msg.Header == nil {
+					msg.Header = make(nats.Header)
+				}
 				if hbTimer != nil {
 					hbTimer.Reset(2 * req.Heartbeat)
 				}