origin checker refactor (#7889)

This commit is contained in:
Chris
2017-11-22 15:58:03 -06:00
committed by Christopher Speller
parent 77a1dc1f2f
commit 1ccf093803
4 changed files with 17 additions and 18 deletions

View File

@@ -19,12 +19,10 @@ func (api *API) InitWebSocket() {
}
func connectWebSocket(c *Context, w http.ResponseWriter, r *http.Request) {
originChecker := utils.GetOriginChecker(r)
upgrader := websocket.Upgrader{
ReadBufferSize: model.SOCKET_MAX_MESSAGE_SIZE_KB,
WriteBufferSize: model.SOCKET_MAX_MESSAGE_SIZE_KB,
CheckOrigin: originChecker,
CheckOrigin: c.App.OriginChecker(),
}
ws, err := upgrader.Upgrade(w, r, nil)