преди 4 години · cc81474b37
--- a/.idea/.gitignore
+++ b/.idea/.gitignore
@@ -0,0 +1,8 @@
 
				+# Default ignored files
			
 
				+/shelf/
			
 
				+/workspace.xml
			
 
				+# Datasource local storage ignored files
			
 
				+/dataSources/
			
 
				+/dataSources.local.xml
			
 
				+# Editor-based HTTP Client requests
			
 
				+/httpRequests/
			
--- a/.idea/inspectionProfiles/Project_Default.xml
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -0,0 +1,14 @@
 
				+<component name="InspectionProjectProfileManager">
			
 
				+  <profile version="1.0">
			
 
				+    <option name="myName" value="Project Default" />
			
 
				+    <inspection_tool class="GoCommentStart" enabled="false" level="WEAK WARNING" enabled_by_default="false" />
			
 
				+    <inspection_tool class="GoSnakeCaseUsage" enabled="false" level="WEAK WARNING" enabled_by_default="false" />
			
 
				+    <inspection_tool class="SpellCheckingInspection" enabled="false" level="TYPO" enabled_by_default="false">
			
 
				+      <option name="processCode" value="true" />
			
 
				+      <option name="processLiterals" value="true" />
			
 
				+      <option name="processComments" value="true" />
			
 
				+    </inspection_tool>
			
 
				+    <inspection_tool class="SqlDialectInspection" enabled="false" level="WARNING" enabled_by_default="false" />
			
 
				+    <inspection_tool class="SqlNoDataSourceInspection" enabled="false" level="WARNING" enabled_by_default="false" />
			
 
				+  </profile>
			
 
				+</component>
			
--- a/.idea/modules.xml
+++ b/.idea/modules.xml
@@ -0,0 +1,8 @@
 
				+<?xml version="1.0" encoding="UTF-8"?>
			
 
				+<project version="4">
			
 
				+  <component name="ProjectModuleManager">
			
 
				+    <modules>
			
 
				+      <module fileurl="file://$PROJECT_DIR$/.idea/ws_test.iml" filepath="$PROJECT_DIR$/.idea/ws_test.iml" />
			
 
				+    </modules>
			
 
				+  </component>
			
 
				+</project>
			
--- a/.idea/ws_test.iml
+++ b/.idea/ws_test.iml
@@ -0,0 +1,9 @@
 
				+<?xml version="1.0" encoding="UTF-8"?>
			
 
				+<module type="WEB_MODULE" version="4">
			
 
				+  <component name="Go" enabled="true" />
			
 
				+  <component name="NewModuleRootManager">
			
 
				+    <content url="file://$MODULE_DIR$" />
			
 
				+    <orderEntry type="inheritedJdk" />
			
 
				+    <orderEntry type="sourceFolder" forTests="false" />
			
 
				+  </component>
			
 
				+</module>
			
--- a/go.mod
+++ b/go.mod
@@ -0,0 +1,5 @@
 
				+module ws_test
			
 
				+
			
 
				+go 1.16
			
 
				+
			
 
				+require github.com/gorilla/websocket v1.4.2
			
--- a/go.sum
+++ b/go.sum
@@ -0,0 +1,2 @@
 
				+github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
			
 
				+github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
			
--- a/http.go
+++ b/http.go
@@ -0,0 +1,49 @@
 
				+package main
			
 
				+
			
 
				+import (
			
 
				+	"flag"
			
 
				+	"fmt"
			
 
				+	"log"
			
 
				+	"net/http"
			
 
				+
			
 
				+	"github.com/gorilla/websocket"
			
 
				+)
			
 
				+
			
 
				+var addr = flag.String("addr", "localhost:8080", "http service address")
			
 
				+
			
 
				+var upgrader = websocket.Upgrader{
			
 
				+	// 解决跨域问题
			
 
				+	CheckOrigin: func(r *http.Request) bool {
			
 
				+		return true
			
 
				+	},
			
 
				+} // use default options
			
 
				+
			
 
				+func ws(w http.ResponseWriter, r *http.Request) {
			
 
				+	c, err := upgrader.Upgrade(w, r, nil)
			
 
				+	if err != nil {
			
 
				+		log.Print("upgrade:", err)
			
 
				+		return
			
 
				+	}
			
 
				+	defer c.Close()
			
 
				+	for {
			
 
				+		mt, message, err := c.ReadMessage()
			
 
				+		if err != nil {
			
 
				+			log.Println("read:", err)
			
 
				+			break
			
 
				+		}
			
 
				+		log.Printf("recv: %s", message)
			
 
				+		err = c.WriteMessage(mt, message)
			
 
				+		if err != nil {
			
 
				+			log.Println("write:", err)
			
 
				+			break
			
 
				+		}
			
 
				+	}
			
 
				+}
			
 
				+
			
 
				+func main() {
			
 
				+	flag.Parse()
			
 
				+	log.SetFlags(0)
			
 
				+	http.HandleFunc("/ws", ws)
			
 
				+	fmt.Println(*addr)
			
 
				+	log.Fatal(http.ListenAndServe(*addr, nil))
			
 
				+}
			
--- a/test.html
+++ b/test.html
@@ -0,0 +1,53 @@
 
				+<!DOCTYPE HTML>
			
 
				+<html>
			
 
				+<head>
			
 
				+    <meta charset="utf-8">
			
 
				+    <title>菜鸟教程(runoob.com)</title>
			
 
				+
			
 
				+    <script type="text/javascript">
			
 
				+        function WebSocketTest()
			
 
				+        {
			
 
				+            if ("WebSocket" in window)
			
 
				+            {
			
 
				+                alert("您的浏览器支持 WebSocket!");
			
 
				+
			
 
				+                // 打开一个 web socket
			
 
				+                var ws = new WebSocket("ws://localhost:8080/ws");
			
 
				+
			
 
				+                ws.onopen = function()
			
 
				+                {
			
 
				+                    // Web Socket 已连接上，使用 send() 方法发送数据
			
 
				+                    ws.send("发送数据");
			
 
				+                    alert("数据发送中...");
			
 
				+                };
			
 
				+
			
 
				+                ws.onmessage = function (evt)
			
 
				+                {
			
 
				+                    var received_msg = evt.data;
			
 
				+                    alert("数据已接收...");
			
 
				+                };
			
 
				+
			
 
				+                ws.onclose = function()
			
 
				+                {
			
 
				+                    // 关闭 websocket
			
 
				+                    alert("连接已关闭...");
			
 
				+                };
			
 
				+            }
			
 
				+
			
 
				+            else
			
 
				+            {
			
 
				+                // 浏览器不支持 WebSocket
			
 
				+                alert("您的浏览器不支持 WebSocket!");
			
 
				+            }
			
 
				+        }
			
 
				+    </script>
			
 
				+
			
 
				+</head>
			
 
				+<body>
			
 
				+
			
 
				+<div id="sse">
			
 
				+    <a href="javascript:WebSocketTest()">运行 WebSocket</a>
			
 
				+</div>
			
 
				+
			
 
				+</body>
			
 
				+</html>