Add configuration option publicServer.
This commit is contained in:
1 parent
bb0a01895e
commit
90e2de0b2d
3 files changed
+23
-1
No files matched your search
+19
-1
@@ -479,8 +479,26 @@ var wsUpgrader = websocket.Upgrader{
|
||||
HandshakeTimeout: 30 * time.Second,
|
||||
}
|
||||
|
||||
var wsPublicUpgrader = websocket.Upgrader{
|
||||
HandshakeTimeout: 30 * time.Second,
|
||||
CheckOrigin: func(r *http.Request) bool {
|
||||
return true
|
||||
},
|
||||
}
|
||||
|
||||
func wsHandler(w http.ResponseWriter, r *http.Request) {
|
||||
conn, err := wsUpgrader.Upgrade(w, r, nil)
|
||||
conf, err := group.GetConfiguration()
|
||||
if err != nil {
|
||||
http.Error(w, "Internal server error",
|
||||
http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
upgrader := wsUpgrader
|
||||
if conf.PublicServer {
|
||||
upgrader = wsPublicUpgrader
|
||||
}
|
||||
|
||||
conn, err := upgrader.Upgrade(w, r, nil)
|
||||
if err != nil {
|
||||
log.Printf("Websocket upgrade: %v", err)
|
||||
return
|
||||
|
||||
Reference in new issue
Block a user