feat: add consumer and producer messages

message/message.go

@@ -6,12 +6,14 @@ import (
 )
 
 const (
-	OperationPut     = "PUT"
-	OperationGet     = "GET"
-	OperationEmpty   = "EMP"
-	OperationError   = "ERR"
-	OperationOK      = "OK"
-	OperationMessage = "MSG"
+	OperationPut      = "PUT"
+	OperationGet      = "GET"
+	OperationEmpty    = "EMP"
+	OperationError    = "ERR"
+	OperationOK       = "OK"
+	OperationMessage  = "MSG"
+	OperationConsumer = "CON"
+	OperationProducer = "PRO"
 
 	MessageEmpty = ""
 	MessageError = "Operation failed!"
@@ -71,6 +73,20 @@ func NewGetMessage() *Message {
 	}
 }
 
+func NewConsumerMessage() *Message {
+	return &Message{
+		Operation: OperationConsumer,
+		Data:      MessageEmpty,
+	}
+}
+
+func NewProducerMessage() *Message {
+	return &Message{
+		Operation: OperationProducer,
+		Data:      MessageEmpty,
+	}
+}
+
 func (message *Message) ToUnpack(data string) {
 	if err := json.Unmarshal([]byte(data), message); err != nil {
 		log.Panicf("Some unpack error: %s.\n", err)

message/message_test.go

@@ -98,6 +98,32 @@ func TestGetMessage(t *testing.T) {
 	assert.Exactly(t, getMessage, message)
 }
 
+func TestConsumerMessage(t *testing.T) {
+	t.Parallel()
+
+	message := NewConsumerMessage()
+
+	getMessage := &Message{
+		Operation: OperationConsumer,
+		Data:      MessageEmpty,
+	}
+
+	assert.Exactly(t, getMessage, message)
+}
+
+func TestProducerMessage(t *testing.T) {
+	t.Parallel()
+
+	message := NewProducerMessage()
+
+	getMessage := &Message{
+		Operation: OperationProducer,
+		Data:      MessageEmpty,
+	}
+
+	assert.Exactly(t, getMessage, message)
+}
+
 func TestToUnpack(t *testing.T) {
 	t.Parallel()