fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
laodaming 2023-09-05 14:05:21 +08:00
parent 038ac93db4
commit 290859eaa2
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/websocket/etc/websocket.yaml
View File

@ -3,8 +3,8 @@ Host: 0.0.0.0
Port: 9914 Port: 9914
ReplicaId: 95 ReplicaId: 95
Timeout: 15000 #服务超时时间 Timeout: 15000 #服务超时时间
CertFile: "/opt/server.fusen.3718.cn.pem" #CertFile: "/opt/server.fusen.3718.cn.pem"
KeyFile: "/opt/server.fusen.3718.cn.key" #KeyFile: "/opt/server.fusen.3718.cn.key"
SourceMysql: "fsreaderwriter:XErSYmLELKMnf3Dh@tcp(fusen.cdmigcvz3rle.us-east-2.rds.amazonaws.com:3306)/fusen" SourceMysql: "fsreaderwriter:XErSYmLELKMnf3Dh@tcp(fusen.cdmigcvz3rle.us-east-2.rds.amazonaws.com:3306)/fusen"
Log: Log:
Stat: false Stat: false

10
server/websocket/internal/logic/datatransferlogic.go
View File

@ -116,12 +116,14 @@ type wsConnectItem struct {
// 请求建立连接，升级websocket协议 // 请求建立连接，升级websocket协议
func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) { func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) {
//判断是不是允许的origin origin := r.Header.Get("Origin")
/*if _, ok := mapAllowOrigin[r.Header.Get("Origin")]; !ok { //判断是不是允许的host
upgrader.CheckOrigin = func(r *http.Request) bool { upgrader.CheckOrigin = func(r *http.Request) bool {
if _, ok := mapAllowOrigin[origin]; !ok {
return false return false
} }
}*/ return true
}
//把子协议携带的token设置到标准token头信息中 //把子协议携带的token设置到标准token头信息中
token := r.Header.Get("Sec-Websocket-Protocol") token := r.Header.Get("Sec-Websocket-Protocol")
//有token是正常用户，无则是白板用户，也可以连接 //有token是正常用户，无则是白板用户，也可以连接