server.go 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  1. // Copyright 2015 The Gorilla WebSocket Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. // +build ignore
  5. package main
  6. import (
  7. "flag"
  8. "html/template"
  9. "log"
  10. "net/http"
  11. "github.com/gorilla/websocket"
  12. )
  13. var addr = flag.String("addr", "localhost:8080", "http service address")
  14. var upgrader = websocket.Upgrader{} // use default options
  15. func echo(w http.ResponseWriter, r *http.Request) {
  16. c, err := upgrader.Upgrade(w, r, nil)
  17. if err != nil {
  18. log.Print("upgrade:", err)
  19. return
  20. }
  21. defer c.Close()
  22. for {
  23. mt, message, err := c.ReadMessage()
  24. if err != nil {
  25. log.Println("read:", err)
  26. break
  27. }
  28. log.Printf("recv: %s", message)
  29. err = c.WriteMessage(mt, message)
  30. if err != nil {
  31. log.Println("write:", err)
  32. break
  33. }
  34. }
  35. }
  36. func home(w http.ResponseWriter, r *http.Request) {
  37. homeTemplate.Execute(w, "ws://"+r.Host+"/echo")
  38. }
  39. func main() {
  40. flag.Parse()
  41. log.SetFlags(0)
  42. http.HandleFunc("/echo", echo)
  43. http.HandleFunc("/", home)
  44. log.Fatal(http.ListenAndServe(*addr, nil))
  45. }
  46. var homeTemplate = template.Must(template.New("").Parse(`
  47. <!DOCTYPE html>
  48. <head>
  49. <meta charset="utf-8">
  50. <script>
  51. window.addEventListener("load", function(evt) {
  52. var output = document.getElementById("output");
  53. var input = document.getElementById("input");
  54. var ws;
  55. var print = function(message) {
  56. var d = document.createElement("div");
  57. d.innerHTML = message;
  58. output.appendChild(d);
  59. };
  60. document.getElementById("open").onclick = function(evt) {
  61. if (ws) {
  62. return false;
  63. }
  64. ws = new WebSocket("{{.}}");
  65. ws.onopen = function(evt) {
  66. print("OPEN");
  67. }
  68. ws.onclose = function(evt) {
  69. print("CLOSE");
  70. ws = null;
  71. }
  72. ws.onmessage = function(evt) {
  73. print("RESPONSE: " + evt.data);
  74. }
  75. ws.onerror = function(evt) {
  76. print("ERROR: " + evt.data);
  77. }
  78. return false;
  79. };
  80. document.getElementById("send").onclick = function(evt) {
  81. if (!ws) {
  82. return false;
  83. }
  84. print("SEND: " + input.value);
  85. ws.send(input.value);
  86. return false;
  87. };
  88. document.getElementById("close").onclick = function(evt) {
  89. if (!ws) {
  90. return false;
  91. }
  92. ws.close();
  93. return false;
  94. };
  95. });
  96. </script>
  97. </head>
  98. <body>
  99. <table>
  100. <tr><td valign="top" width="50%">
  101. <p>Click "Open" to create a connection to the server,
  102. "Send" to send a message to the server and "Close" to close the connection.
  103. You can change the message and send multiple times.
  104. <p>
  105. <form>
  106. <button id="open">Open</button>
  107. <button id="close">Close</button>
  108. <p><input id="input" type="text" value="Hello world!">
  109. <button id="send">Send</button>
  110. </form>
  111. </td><td valign="top" width="50%">
  112. <div id="output"></div>
  113. </td></tr></table>
  114. </body>
  115. </html>
  116. `))