定时清理不活跃连接

添加页面模板

README.md

@@ -60,12 +60,13 @@ govendor add -tree github.com/spf13/viper
 ```
 
 #### 待办事项
+- gin log日志(请求日志+debug日志)
 - 读取配置文件 完成
-- 定时脚本，清理过期未心跳链接
+- 定时脚本，清理过期未心跳链接 完成
 - http接口，获取登录、链接数量 完成
 - http接口，发送push、查询有多少人在线 完成
 - grpc 程序内部通讯，发送消息
 - appIds 一个用户在多个平台登录
 - 界面，把所有在线的人拉倒一个群里面，发送消息
-- ~~群聊~~、单聊
+- ~~单聊~~、群聊
 - 实现分布式，水平扩张

controllers/home/home_controller.go

@@ -0,0 +1,21 @@
+/**
+* Created by GoLand.
+* User: link1st
+* Date: 2019-07-25
+* Time: 12:11
+ */
+
+package home
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+// 查看用户是否在线
+func Index(c *gin.Context) {
+	data := gin.H{
+		"title": "聊天首页",
+	}
+	c.HTML(http.StatusOK, "index.tpl", data)
+}

main.go

@@ -13,6 +13,7 @@ import (
 	"github.com/spf13/viper"
 	"gowebsocket/lib/redislib"
 	"gowebsocket/routers"
+	"gowebsocket/servers/task"
 	"gowebsocket/servers/websocket"
 	"io"
 	"net/http"
@@ -31,6 +32,9 @@ func main() {
 	routers.Init(router)
 	routers.WebsocketInit()
 
+	// 定时任务
+	task.Init()
+
 	go websocket.StartWebSocket()
 
 	httpPort := viper.GetString("app.httpPort")

routers/web_routers.go

@@ -9,11 +9,13 @@ package routers
 
 import (
 	"github.com/gin-gonic/gin"
+	"gowebsocket/controllers/home"
 	"gowebsocket/controllers/systems"
 	"gowebsocket/controllers/user"
 )
 
 func Init(router *gin.Engine) {
+	router.LoadHTMLGlob("views/**/*")
 
 	// 用户组
 	userRouter := router.Group("/user")
@@ -24,9 +26,15 @@ func Init(router *gin.Engine) {
 	}
 
 	// 系统
-	system := router.Group("/system")
+	systemRouter := router.Group("/system")
 	{
-		system.GET("/state", systems.Status)
+		systemRouter.GET("/state", systems.Status)
+	}
+
+	// home
+	homeRouter := router.Group("/home")
+	{
+		homeRouter.GET("/index", home.Index)
 	}
 
 	// router.POST("/user/online", user.Online)

servers/task/clean_connection _task.go

@@ -0,0 +1,37 @@
+/**
+* Created by GoLand.
+* User: link1st
+* Date: 2019-07-31
+* Time: 15:17
+ */
+
+package task
+
+import (
+	"fmt"
+	"gowebsocket/servers/websocket"
+	"runtime/debug"
+	"time"
+)
+
+func Init() {
+	Timer(3*time.Second, 30*time.Second, cleanConnection, "")
+
+}
+
+// 清理超时连接
+func cleanConnection(param interface{}) (result bool) {
+	result = true
+
+	defer func() {
+		if r := recover(); r != nil {
+			fmt.Println("ClearTimeoutConnections stop", r, string(debug.Stack()))
+		}
+	}()
+
+	fmt.Println("定时任务，清理超时连接", param)
+
+	websocket.ClearTimeoutConnections()
+
+	return
+}

servers/websocket/client.go

@@ -13,6 +13,11 @@ import (
 	"runtime/debug"
 )
 
+const (
+	// 用户连接超时时间
+	heartbeatExpirationTime = 6 * 60
+)
+
 // 用户登录
 type Login struct {
 	AppId  uint32
@@ -64,10 +69,14 @@ func (c *Client) read() {
 	defer func() {
 		if r := recover(); r != nil {
 			fmt.Println("write stop", string(debug.Stack()), r)
-
 		}
 	}()
 
+	defer func() {
+		fmt.Println("读取客户端数据 关闭send", c)
+		close(c.Send)
+	}()
+
 	for {
 		_, message, err := c.Socket.ReadMessage()
 		if err != nil {
@@ -77,12 +86,12 @@ func (c *Client) read() {
 		}
 
 		// 处理程序
-		fmt.Println("读取客户端数据 处理", string(message))
+		fmt.Println("读取客户端数据 处理:", string(message))
 		ProcessData(c, message)
 	}
 }
 
-// 读取客户端数据
+// 向客户端写数据
 func (c *Client) write() {
 	defer func() {
 		if r := recover(); r != nil {
@@ -138,6 +147,15 @@ func (c *Client) Heartbeat(currentTime uint64) {
 	return
 }
 
+// 心跳超时
+func (c *Client) IsHeartbeatTimeout(currentTime uint64) (timeout bool) {
+	if c.HeartbeatTime+heartbeatExpirationTime <= currentTime {
+		timeout = true
+	}
+
+	return
+}
+
 // 是否登录了
 func (c *Client) IsLogin() (isLogin bool) {
 

servers/websocket/client_manager.go

@@ -11,6 +11,7 @@ import (
 	"fmt"
 	"gowebsocket/lib/cache"
 	"sync"
+	"time"
 )
 
 // 连接管理
@@ -139,9 +140,6 @@ func (manager *ClientManager) EventUnregister(client *Client) {
 	if err == nil {
 		userOnline.LogOut()
 		cache.SetUserOnlineInfo(client.GetKey(), userOnline)
-
-		// 退出登录事件
-		// currentTime := uint64(time.Now().Unix())
 	}
 
 	// 关闭 chan
@@ -215,3 +213,17 @@ func GetUserClient(appId uint32, userId string) (client *Client) {
 
 	return
 }
+
+// 定时清理超时连接
+func ClearTimeoutConnections() {
+
+	currentTime := uint64(time.Now().Unix())
+
+	for client := range clientManager.Clients {
+		if client.IsHeartbeatTimeout(currentTime) {
+			fmt.Println("心跳时间超时 关闭连接", client.Addr, client.UserId, client.LoginTime, client.HeartbeatTime)
+
+			client.Socket.Close()
+		}
+	}
+}

views/home/index.tpl

@@ -0,0 +1,9 @@
+<!DOCTYPE html>
+<html>
+<head>
+    <title>{{ .title }}</title>
+</head>
+<body>
+
+</body>
+</html>