Merge pull request #8 from WindowsSov8forUs/master

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

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

pkg/message/message_element_extend.go

@@ -39,7 +39,7 @@ func (e *MessageElementExtend) Parse(n *html.Node) (MessageElement, error) {
 	return result, nil
 }
 
-func NewMessageElementExtend(tag string, attrs map[string]string, children ...MessageElement) MessageElement {
+func NewMessageElementExtend(tag string, attrs map[string]string, children ...MessageElement) *MessageElementExtend {
 	result := &MessageElementExtend{
 		tag: tag,
 	}