fix: wrong header sent on server/client.go

go-stomp · Mar 23, 2022 · aae3b21 · aae3b21
1 parent b48ba3c
commit aae3b21
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/examples/client_test/client_test b/examples/client_test/client_test
diff --git a/server/client/conn.go b/server/client/conn.go
@@ -415,9 +415,9 @@ func (c *Conn) allocateMessageId(f *frame.Frame, sub *Subscription) {
 		// if there is any requirement by the client to acknowledge, set
 		// the ack header as per STOMP 1.2
 		if sub == nil || sub.ack == frame.AckAuto {
-			f.Header.Del(frame.Id)
+			f.Header.Del(frame.Ack)
 		} else {
-			f.Header.Set(frame.Id, messageId)
+			f.Header.Set(frame.Ack, messageId)
 		}
 	}
 }
@@ -659,7 +659,7 @@ func (c *Conn) handleAck(f *frame.Frame) error {
 	var err error
 	var msgId string
 
-	if ack, ok := f.Header.Contains(frame.Id); ok {
+	if ack, ok := f.Header.Contains(frame.Ack); ok {
 		msgId = ack
 	} else if msgId, ok = f.Header.Contains(frame.MessageId); !ok {
 		return missingHeader(frame.MessageId)
@@ -702,7 +702,7 @@ func (c *Conn) handleNack(f *frame.Frame) error {
 	var err error
 	var msgId string
 
-	if ack, ok := f.Header.Contains(frame.Id); ok {
+	if ack, ok := f.Header.Contains(frame.Ack); ok {
 		msgId = ack
 	} else if msgId, ok = f.Header.Contains(frame.MessageId); !ok {
 		return missingHeader(frame.MessageId)