init push

backend/utils/processor.go

@@ -0,0 +1,75 @@
+package utils
+
+import (
+	"bytes"
+	"errors"
+	"io"
+	"sync"
+)
+
+type Node struct {
+	buf *bytes.Buffer
+	son []*Node
+}
+
+func newNode() *Node {
+	return &Node{son: []*Node{}, buf: bytes.NewBufferString("")}
+}
+
+type ProcessorTree struct {
+	mu     *sync.Mutex
+	root   *Node
+	result *bytes.Buffer
+}
+
+func NewProcessorTree() *ProcessorTree {
+	return &ProcessorTree{
+		root:   newNode(),
+		mu:     &sync.Mutex{},
+		result: bytes.NewBufferString(""),
+	}
+}
+
+// 获取一个father下的节点
+func (t *ProcessorTree) GetNode(farther *Node) (*Node, error) {
+	if farther == nil {
+		return nil, errors.New("father is nil")
+	}
+	t.mu.Lock()
+	defer t.mu.Unlock()
+	temp := newNode()
+	farther.son = append(farther.son, temp)
+	return temp, nil
+}
+
+func (t *ProcessorTree) Add(node *Node, data []byte) error {
+	if node == nil {
+		return errors.New("node is nil")
+	}
+	t.mu.Lock()
+	defer t.mu.Unlock()
+	node.buf.Write(data)
+	return nil
+}
+
+func (t *ProcessorTree) GetResult() ([]byte, error) {
+	if err := t.getRes(t.root); err != nil {
+		return nil, err
+	}
+	return t.result.Bytes(), nil
+}
+
+func (t *ProcessorTree) getRes(node *Node) error {
+	if node == nil {
+		return nil
+	}
+	if _, err := io.Copy(t.result, node.buf); err != nil {
+		return err
+	}
+	for _, son := range node.son {
+		if err := t.getRes(son); err != nil {
+			return err
+		}
+	}
+	return nil
+}