tt

Server/common/net/socket/websocket/wsconn.go

@@ -19,7 +19,7 @@ type WSConn struct {
 	logger    logging.Logger
 	isUpgrade bool
 	isClose   bool
-	param     map[string]interface{}
+	param     map[string]string
 	openTime  int64
 	wsMessageBuf
 }
@@ -163,15 +163,17 @@ func (w *WSConn) OnRequest(u []byte) error {
 	if err != nil {
 		return err
 	}
-	w.SetParam("query", parsedURL.Query())
+	for key, value := range parsedURL.Query() {
+		w.SetParam(key, value[0])
+	}
 	return nil
 }
 
-func (w *WSConn) GetParam(key string) interface{} {
+func (w *WSConn) GetParam(key string) string {
 	return w.param[key]
 }
 
-func (w *WSConn) SetParam(key string, values interface{}) {
+func (w *WSConn) SetParam(key string, values string) {
 	w.param[key] = values
 }