浏览代码

add state string

Joyit 1 月之前
父节点
当前提交
f34f9c7813
共有 2 个文件被更改,包括 28 次插入0 次删除
  1. 14 0
      conn/conn.go
  2. 14 0
      type.go

+ 14 - 0
conn/conn.go

@@ -1,6 +1,7 @@
 package conn
 
 import (
+	"fmt"
 	"net"
 )
 
@@ -13,6 +14,19 @@ const (
 	ResponseMsg
 )
 
+func (t MsgType) String() string {
+	switch t {
+	case PingMsg:
+		return "PingMsg"
+	case RequestMsg:
+		return "RequestMsg"
+	case ResponseMsg:
+		return "ResponseMsg"
+	default:
+		return fmt.Sprintf("Unknown MsgType (%d)", t)
+	}
+}
+
 // 连接接口,代表一个原始的连接
 type Connect interface {
 	WriteAuthInfo(channel string, auth []byte) (err error)

+ 14 - 0
type.go

@@ -1,6 +1,7 @@
 package tinymq
 
 import (
+	"fmt"
 	"regexp"
 	"time"
 )
@@ -46,6 +47,19 @@ const (
 	Closed
 )
 
+func (t ConnectState) String() string {
+	switch t {
+	case Disconnected:
+		return "Disconnected"
+	case Connected:
+		return "Connected"
+	case Closed:
+		return "Closed"
+	default:
+		return fmt.Sprintf("Unknown ConnectState (%d)", t)
+	}
+}
+
 // 请求数据包
 type RequestData struct {
 	Id   uint16