You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.2 KiB
94 lines
2.2 KiB
package msggateway
|
|
|
|
import (
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type UserConnContext struct {
|
|
RespWriter http.ResponseWriter
|
|
Req *http.Request
|
|
Path string
|
|
Method string
|
|
RemoteAddr string
|
|
ConnID string
|
|
}
|
|
|
|
func (c *UserConnContext) Deadline() (deadline time.Time, ok bool) {
|
|
return
|
|
}
|
|
|
|
func (c *UserConnContext) Done() <-chan struct{} {
|
|
return nil
|
|
}
|
|
|
|
func (c *UserConnContext) Err() error {
|
|
return nil
|
|
}
|
|
|
|
func (c *UserConnContext) Value(key any) any {
|
|
switch key {
|
|
case constant.OpUserID:
|
|
return c.GetUserID()
|
|
case constant.OperationID:
|
|
return c.GetOperationID()
|
|
case constant.ConnID:
|
|
return c.GetConnID()
|
|
case constant.OpUserPlatform:
|
|
return constant.PlatformIDToName(utils.StringToInt(c.GetPlatformID()))
|
|
case constant.RemoteAddr:
|
|
return c.RemoteAddr
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func newContext(respWriter http.ResponseWriter, req *http.Request) *UserConnContext {
|
|
return &UserConnContext{
|
|
RespWriter: respWriter,
|
|
Req: req,
|
|
Path: req.URL.Path,
|
|
Method: req.Method,
|
|
RemoteAddr: req.RemoteAddr,
|
|
ConnID: utils.Md5(req.RemoteAddr + "_" + strconv.Itoa(int(utils.GetCurrentTimestampByMill()))),
|
|
}
|
|
}
|
|
func (c *UserConnContext) GetRemoteAddr() string {
|
|
return c.RemoteAddr
|
|
}
|
|
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, 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.ConnID
|
|
}
|
|
func (c *UserConnContext) GetUserID() string {
|
|
return c.Req.URL.Query().Get(WsUserID)
|
|
}
|
|
func (c *UserConnContext) GetPlatformID() string {
|
|
return c.Req.URL.Query().Get(PlatformID)
|
|
}
|
|
func (c *UserConnContext) GetOperationID() string {
|
|
return c.Req.URL.Query().Get(OperationID)
|
|
}
|