add: 添加子元素与扩展属性动态设置方法

satori-protocol-go · Jun 19, 2024 · 5de5949 · 5de5949
1 parent 0d1f361
commit 5de5949
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pkg/message/message_element.go b/pkg/message/message_element.go
@@ -57,6 +57,11 @@ func (e *ChildrenMessageElement) GetChildren() []MessageElement {
 	return e.children
 }
 
+func (e *ChildrenMessageElement) SetChildren(children []MessageElement) *ChildrenMessageElement {
+	e.children = children
+	return e
+}
+
 type ExtendAttributes struct {
 	attributes map[string]string
 }
@@ -72,6 +77,14 @@ func (e *ExtendAttributes) AddAttribute(key, value string) *ExtendAttributes {
 	return result
 }
 
+func (e *ExtendAttributes) DelAttribute(key string) *ExtendAttributes {
+	if e == nil {
+		return nil
+	}
+	delete(e.attributes, key)
+	return e
+}
+
 func (e *ExtendAttributes) Get(key string) (string, bool) {
 	if e == nil {
 		return "", false