From adf9716b7ed2cea991f29c9f3d5ba37575aecdb8 Mon Sep 17 00:00:00 2001
From: isinged <isinged@163.com>
Date: Sun, 7 Mar 2021 01:01:08 +0800
Subject: [PATCH] fix exit

---
 srv.go      |  2 +-
 srv_test.go | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/srv.go b/srv.go
index b22daca..99a53b2 100644
--- a/srv.go
+++ b/srv.go
@@ -184,13 +184,13 @@ func (s *Srv) NewContext(server ServerAdapter, sid string, req *Request) *Contex
 	if ok {
 		handlers = make([]HandlerFunc, 0, len(s.middleware)+len(routeHandlers)+len(s.internalMiddleware))
 		handlers = append(handlers, s.middleware...)
+		handlers = append(handlers, s.internalMiddleware...)
 		handlers = append(handlers, routeHandlers...)
 		ctx.OK() // 匹配到了路由，但是 handler 没有设置响应
 	} else {
 		handlers = make([]HandlerFunc, 0, len(s.middleware)+1)
 		handlers = append(handlers, s.middleware...)
 	}
-	handlers = append(handlers, s.internalMiddleware...)
 	ctx.handlers = handlers
 	ctx.handlerIndex = -1
 	return ctx
diff --git a/srv_test.go b/srv_test.go
index 0a2a058..f6e2004 100644
--- a/srv_test.go
+++ b/srv_test.go
@@ -327,13 +327,13 @@ func TestSrv_Exit(t *testing.T) {
 			c.Set("td", c.Get("td").(string)+"d")
 		})
 
-		srv.Run()
+		go srv.Run()
 		time.Sleep(50 * time.Millisecond)
 
-		t.Assert(srv.GetState("test.sid", "ta"), "acb")
-		t.Assert(srv.GetState("test.sid", "tb"), "ac")
-		t.Assert(srv.GetState("test.sid", "tc"), "acdb")
-		t.Assert(srv.GetState("test.sid", "td"), "ac")
+		t.Assert(srv.GetState("", "ta"), "acb")
+		t.Assert(srv.GetState("", "tb"), "acb")
+		t.Assert(srv.GetState("", "tc"), "acdb")
+		t.Assert(srv.GetState("", "td"), "acb")
 	})
 
 }