From 19cb3e3ddf604cffc9c887eefb7e962402d91c89 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?=
 <go_developer@163.com>
Date: Wed, 17 Nov 2021 18:54:34 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=9C=8D=E5=8A=A1=E9=80=80?=
 =?UTF-8?q?=E5=87=BA=E6=97=B6,=20finish=20hook=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 construct.go      | 11 +++++++----
 example/server.go |  4 ++--
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/construct.go b/construct.go
index 5d90367..5e5869b 100644
--- a/construct.go
+++ b/construct.go
@@ -60,7 +60,7 @@ var (
 // Author : go_developer@163.com<白茶清欢>
 //
 // Date : 6:49 下午 2021/3/27
-func NewWebsocketServe(wsInstanceList ...abstract.IWebsocket) error {
+func NewWebsocketServe(finishHook func(), wsInstanceList ...abstract.IWebsocket) error {
 	if len(wsInstanceList) == 0 {
 		return errors.WithStack(errors.New("register websocket server list is empty"))
 	}
@@ -70,7 +70,7 @@ func NewWebsocketServe(wsInstanceList ...abstract.IWebsocket) error {
 	for _, wsInstance := range wsInstanceList {
 		initServer(wsInstance)
 	}
-	run()
+	run(finishHook)
 	return nil
 }
 
@@ -321,7 +321,7 @@ func dispatchCommand(ctx *context.WSContext, data []byte) error {
 }
 
 // 启动所有端口的监听
-func run() {
+func run(finishHook func()) {
 	for port, ginInstance := range ginRouterTable {
 		go func(ginInstance *gin.Engine, port int) {
 			if err := ginInstance.Run(fmt.Sprintf(":%d", port)); nil != err {
@@ -331,7 +331,10 @@ func run() {
 	}
 
 	<-sigChan
-	// TODO : 增加后置hook
+	// 增加后置hook
+	if nil != finishHook {
+		finishHook()
+	}
 }
 
 // Stop 停止服务
diff --git a/example/server.go b/example/server.go
index 246fe86..6e56755 100644
--- a/example/server.go
+++ b/example/server.go
@@ -23,7 +23,7 @@ import (
 )
 
 func main() {
-	websocket.NewWebsocketServe(&Example{})
+	_ = websocket.NewWebsocketServe(nil, &Example{})
 }
 
 type Example struct {
@@ -31,7 +31,7 @@ type Example struct {
 
 func (e Example) Connect(ctx *context.WSContext) error {
 	fmt.Println("建立连接成功")
-	message.Response(ctx, map[string]interface{}{"say": "hello world!", "cid": ctx.ConnectionID})
+	_ = message.Response(ctx, map[string]interface{}{"say": "hello world!", "cid": ctx.ConnectionID})
 	return nil
 }