Added request ID parameter to message constructors

1 files changed, 8 insertions, 6 deletions

diff --git a/message.go b/message.go
index 7314927..2e215f8 100644
--- a/message.go
+++ b/message.go
@@ -73,20 +73,22 @@ type Message struct {
 //
 // Use DefaultVersion as a first argument, if you don't
 // have any specific needs
-func NewRequest(v Version, op Op) *Message {
+func NewRequest(v Version, op Op, id uint32) *Message {
 	return &Message{
-		Version: v,
-		Code:    Code(op),
+		Version:   v,
+		Code:      Code(op),
+		RequestID: id,
 	}
 }
 
 // NewResponse creates a new response message
 //
 // Use DefaultVersion as a first argument, if you don't
-func NewResponse(v Version, status Status) *Message {
+func NewResponse(v Version, status Status, id uint32) *Message {
 	return &Message{
-		Version: v,
-		Code:    Code(status),
+		Version:   v,
+		Code:      Code(status),
+		RequestID: id,
 	}
 }