From ecb7cbdfb2d67029a8f99b2bc7be788a7b3266ea Mon Sep 17 00:00:00 2001
From: Xu Qiaolun <Jamesxql@Gmail.com>
Date: Thu, 30 Aug 2018 22:03:52 +0800
Subject: [PATCH] call the ConnState callbacks in Server.ServeConn

---
 server.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server.go b/server.go
index 9519b47..0e85cf0 100644
--- a/server.go
+++ b/server.go
@@ -1545,11 +1545,13 @@ func (s *Server) ServeConn(c net.Conn) error {
 
 	if err != errHijacked {
 		err1 := c.Close()
+		s.setState(c, StateClosed)
 		if err == nil {
 			err = err1
 		}
 	} else {
 		err = nil
+		s.setState(c, StateHijacked)
 	}
 	return err
 }