From 90eecba0d2797bc3cd90b29aafb6a2f719475174 Mon Sep 17 00:00:00 2001 From: guyinyou <36399867+guyinyou@users.noreply.github.com> Date: Fri, 31 Mar 2023 10:51:09 +0800 Subject: [PATCH] add 'AddProperty' (#432) Co-authored-by: guyinyou --- golang/message.go | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/golang/message.go b/golang/message.go index 00586b18c..9d1007f16 100644 --- a/golang/message.go +++ b/golang/message.go @@ -103,10 +103,21 @@ func (msg *Message) SetKeys(keys ...string) { msg.keys = keys } -func (msg *Message) GetProperties() map[string]string { +func (msg *Message) getOrNewProperties() map[string]string { + if msg.properties == nil { + msg.properties = make(map[string]string) + } return msg.properties } +func (msg *Message) GetProperties() map[string]string { + return msg.getOrNewProperties() +} + +func (msg *Message) AddProperty(key, value string) { + msg.getOrNewProperties()[key] = value +} + func (msg *Message) SetDelayTimestamp(deliveryTimestamp time.Time) { msg.deliveryTimestamp = &deliveryTimestamp }