Pass messageID to Keepalive messages

Author: iomz

message.go

@@ -1,21 +1,21 @@
 package llrp
 
 // Keepalive generates Keepalive message.
-func Keepalive() []byte {
+func Keepalive(messageID uint32) []byte {
 	var data = []interface{}{
 		uint16(KeepaliveHeader), // Rsvd+Ver+Type=62 (KEEPALIVE)
 		uint32(10),              // Length
-		uint32(0),               // ID
+		messageID,               // ID
 	}
 	return Pack(data)
 }
 
 // KeepaliveAck generates KeepaliveAck message.
-func KeepaliveAck() []byte {
+func KeepaliveAck(messageID uint32) []byte {
 	var data = []interface{}{
 		uint16(KeepaliveAckHeader), // Rsvd+Ver+Type=62 (KEEPALIVE)
 		uint32(10),                 // Length
-		uint32(0),                  // ID
+		messageID,                  // ID
 	}
 	return Pack(data)
 }

message_test.go

@@ -7,7 +7,7 @@ import (
 
 func TestKeepalive(t *testing.T) {
 	var b, out []byte
-	b = Keepalive()
+	b = Keepalive(0)
 	out = []byte{4, 62, 0, 0, 0, 10, 0, 0, 0, 0}
 	if !bytes.Equal(b, out) {
 		t.Errorf("%v, want %v", b, out)
@@ -16,7 +16,7 @@ func TestKeepalive(t *testing.T) {
 
 func TestKeepaliveAck(t *testing.T) {
 	var b, out []byte
-	b = KeepaliveAck()
+	b = KeepaliveAck(0)
 	out = []byte{4, 72, 0, 0, 0, 10, 0, 0, 0, 0}
 	if !bytes.Equal(b, out) {
 		t.Errorf("%v, want %v", b, out)