From 944e43fa9f1be2bc761b36922304c22e96fbb9e5 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期四, 17 八月 2023 18:11:45 +0800
Subject: [PATCH] 后端添加websocket

---
 server/controllers/plc.go |   27 +++++++++++++++++++++++++++
 1 files changed, 27 insertions(+), 0 deletions(-)

diff --git a/server/controllers/plc.go b/server/controllers/plc.go
index 2d32936..e5c8c7a 100644
--- a/server/controllers/plc.go
+++ b/server/controllers/plc.go
@@ -1 +1,28 @@
 package controllers
+
+import (
+	"fmt"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+)
+
+var upgrader = websocket.Upgrader{}
+
+func WsServer(c *gin.Context) {
+	cn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
+	defer cn.Close()
+
+	if err != nil {
+		// 鍝嶅簲閿欒
+	}
+
+	for {
+		_, message, err := cn.ReadMessage()
+		if err != nil {
+			fmt.Printf("read msg error:%s\n", err.Error())
+		}
+
+		fmt.Printf("read msg: %s\v\n", string(message))
+	}
+}

--
Gitblit v1.8.0