Upload $M[U103]PF3L0]$W6B`8)2P.png by picGo - 2021-09-20 05:06:41

go.mod

@@ -5,4 +5,5 @@ go 1.16
 require (
 	github.com/spf13/cobra v1.2.1
 	github.com/spf13/viper v1.8.1
+	github.com/stretchr/testify v1.7.0
 )

pkg/middle/data.go

@@ -0,0 +1,24 @@
+package middle
+
+// 数据类型
+type data struct {
+	m map[string]interface{}
+}
+
+// 数据初始化，分配内存
+func NewData() *data {
+	return &data{
+		make(map[string]interface{}),
+	}
+}
+
+// 获取数据
+func (d *data) Get(key string) interface{} {
+	return d.m[key]
+}
+
+// 推入数据
+func (d *data) Put(key string, val interface{}) error {
+	d.m[key] = val
+	return nil
+}

pkg/middle/define/define.go

@@ -1 +0,0 @@
-package define

pkg/middle/middle.go

@@ -0,0 +1,62 @@
+package middle
+
+// 获取数据的接口
+type Msg interface {
+	Put(key string, val interface{}) error
+	Get(key string) interface{}
+}
+
+type middleWare interface {
+	Put(msg Msg)
+	Out() Msg
+}
+
+type MiddlewareConf struct {
+	Weight int
+	Middle middleWare
+}
+
+type Driver struct {
+	conf []*MiddlewareConf
+}
+
+func NewDriver() *Driver {
+	return new(Driver)
+}
+
+// 驱动
+func (d *Driver) Start(msg Msg) {
+
+	for _, m := range d.conf {
+
+		go m.Middle.Put(msg)
+
+		msg = m.Middle.Out()
+
+	}
+
+}
+
+func main() {
+	conf := []*MiddlewareConf{
+		{
+			Weight: 1,
+			Middle: NewUpload(),
+		},
+		{
+			Weight: 2,
+			Middle: NewStorage(),
+		},
+		{
+			Weight: 3,
+			Middle: NewSend(),
+		},
+	}
+
+	da := NewDriver()
+	da.conf = conf
+	val := NewData()
+	val.Put("200", 2)
+
+	da.Start(val)
+}

pkg/middle/middle_test.go

@@ -0,0 +1,7 @@
+package middle
+
+import "testing"
+
+func TestMiddle(t *testing.T) {
+	main()
+}

pkg/middle/send.go

@@ -0,0 +1,24 @@
+package middle
+
+import "fmt"
+
+type Send struct {
+	ch chan *data
+}
+
+func NewSend() *Send {
+	return &Send{
+		make(chan *data),
+	}
+}
+
+func (m *Send) Put(msg Msg) {
+	da := NewData()
+	da.Put("1", msg)
+	fmt.Println("传入Send")
+	m.ch <- da
+}
+
+func (m *Send) Out() Msg {
+	return <-m.ch
+}

pkg/middle/storage.go

@@ -0,0 +1,24 @@
+package middle
+
+import "fmt"
+
+type Storage struct {
+	ch chan *data
+}
+
+func NewStorage() *Storage {
+	return &Storage{
+		make(chan *data),
+	}
+}
+
+func (m *Storage) Put(msg Msg) {
+	da := NewData()
+	da.Put("1", msg)
+	fmt.Println("传入Storage")
+	m.ch <- da
+}
+
+func (m *Storage) Out() Msg {
+	return <-m.ch
+}

pkg/middle/upload.go

@@ -0,0 +1,25 @@
+package middle
+
+import "fmt"
+
+type Upload struct {
+	ch chan *data
+}
+
+func NewUpload() *Upload {
+	return &Upload{
+		make(chan *data),
+	}
+}
+
+func (m *Upload) Put(msg Msg) {
+	da := NewData()
+	da.Put("1", msg)
+	fmt.Println("传入Upload")
+
+	m.ch <- da
+}
+
+func (m *Upload) Out() Msg {
+	return <-m.ch
+}