optimize: improve chat example (#12)

* optimize: improve chat example * chore: revert * chore: revert
hertz-contrib · Feb 2, 2023 · b7b0e0f · b7b0e0f
1 parent b50c6ca
commit b7b0e0f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/examples/chat/home.html b/examples/chat/home.html
@@ -29,7 +29,7 @@
  };
 
  if (window["WebSocket"]) {
- conn = new WebSocket("ws://" + document.location.host + "/ws");
+ conn = new WebSocket("ws://localhost:8080/ws");
  conn.onclose = function (evt) {
  var item = document.createElement("div");
  item.innerHTML = "<b>Connection closed.</b>";

diff --git a/examples/chat/main.go b/examples/chat/main.go
@@ -20,6 +20,9 @@ import (
 var upgrader = websocket.HertzUpgrader{
  ReadBufferSize: 1024,
  WriteBufferSize: 1024,
+ CheckOrigin: func(ctx *app.RequestContext) bool {
+ return true
+ },
 }
 
 var addr = ":8080"