add chan of message test case

protocol/message_chan_test.go

@@ -0,0 +1,94 @@
+package protocol
+
+import (
+    "fmt"
+    "strings"
+    "sync/atomic"
+    "testing"
+    "time"
+)
+
+func TestChanValue(t *testing.T) {
+    var ct uint64
+    ch := make(chan Message, 10000)
+    go func(ch <-chan Message) {
+        for range ch {
+            atomic.AddUint64(&ct, 1)
+        }
+    }(ch)
+
+    go func(ch chan Message) {
+        m := strings.Repeat("_", 100)
+        p := strings.Repeat("_", 100)
+        payload := make([]byte, 1024)
+        for {
+            ch <- Message{ServiceMethod: m, ServicePath: p, Payload: payload}
+        }
+    }(ch)
+    for i := 0; i < 5; i++ {
+        time.Sleep(time.Second)
+        fmt.Println(atomic.LoadUint64(&ct))
+        atomic.StoreUint64(&ct, 0)
+    }
+}
+
+func TestChanPtr(t *testing.T) {
+    var ct uint64
+    ch := make(chan *Message, 10000)
+    go func(ch <-chan *Message) {
+        for range ch {
+            atomic.AddUint64(&ct, 1)
+        }
+    }(ch)
+
+    go func(ch chan *Message) {
+        m := strings.Repeat("_", 100)
+        p := strings.Repeat("_", 100)
+        payload := make([]byte, 1024)
+        for {
+            ch <- &Message{ServiceMethod: m, ServicePath: p,Payload: payload}
+        }
+    }(ch)
+
+    for i := 0; i < 5; i++ {
+        time.Sleep(time.Second)
+        fmt.Println(atomic.LoadUint64(&ct))
+        atomic.StoreUint64(&ct, 0)
+    }
+}
+
+func BenchmarkChanValue(b *testing.B) {
+    ch := make(chan Message, 10000)
+    var ct uint64
+    go func(ch <-chan Message) {
+        for range ch {
+            atomic.AddUint64(&ct, 1)
+        }
+    }(ch)
+    b.ReportAllocs()
+    b.ResetTimer()
+    m := strings.Repeat("_", 100)
+    p := strings.Repeat("_", 100)
+    payload := make([]byte, 1024)
+    for i := 0; i < b.N; i++ {
+        ch <- Message{ServiceMethod: m, ServicePath: p,Payload: payload}
+    }
+}
+
+func BenchmarkChanPtr(b *testing.B) {
+    ch := make(chan *Message, 10000)
+    var ct uint64
+    go func(ch <-chan *Message) {
+        for range ch {
+            atomic.AddUint64(&ct, 1)
+        }
+    }(ch)
+    b.ReportAllocs()
+    b.ResetTimer()
+    m := strings.Repeat("_", 100)
+    p := strings.Repeat("_", 100)
+    payload := make([]byte, 1024)
+    for i := 0; i < b.N; i++ {
+        ch <- &Message{ServiceMethod: m, ServicePath: p,Payload: payload}
+    }
+}