msggateway refactor

internal/msggateway/new/context.go

@@ -1,6 +1,10 @@
 package new
 
-import "net/http"
+import (
+	"Open_IM/pkg/utils"
+	"net/http"
+	"strconv"
+)
 
 type UserConnContext struct {
 	RespWriter http.ResponseWriter
@@ -19,9 +23,32 @@ func newContext(respWriter http.ResponseWriter, req *http.Request) *UserConnCont
 		RemoteAddr: req.RemoteAddr,
 	}
 }
-func (c *UserConnContext) Query(key string) string {
-	return c.Req.URL.Query().Get(key)
+func (c *UserConnContext) Query(key string) (string, bool) {
+	var value string
+	if value = c.Req.URL.Query().Get(key); value == "" {
+		return value, false
+	}
+	return value, true
 }
-func (c *UserConnContext) GetHeader(key string) string {
-	return c.Req.Header.Get(key)
+func (c *UserConnContext) GetHeader(key string) (string, bool) {
+	var value string
+	if value = c.Req.Header.Get(key); value == "" {
+		return value, false
+	}
+	return value, true
+}
+func (c *UserConnContext) SetHeader(key, value string) {
+	c.RespWriter.Header().Set(key, value)
+}
+func (c *UserConnContext) ErrReturn(error string, code int) {
+	http.Error(c.RespWriter, error, code)
+}
+func (c *UserConnContext) GetConnID() string {
+	return c.RemoteAddr + "_" + strconv.Itoa(int(utils.GetCurrentTimestampByMill()))
+}
+func (c *UserConnContext) GetUserID() string {
+	return c.Req.URL.Query().Get(USERID)
+}
+func (c *UserConnContext) GetPlatformID() string {
+	return c.Req.URL.Query().Get(PLATFORM_ID)
 }