Changed *tls.Conn by extended connTLSer

valyala · Nov 12, 2018 · 3fb1313 · 3fb1313
1 parent d37102d
commit 3fb1313
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server.go b/server.go
@@ -546,6 +546,7 @@ func (ctx *RequestCtx) VisitUserValues(visitor func([]byte, interface{})) {
 }
 
 type connTLSer interface {
+	Handshake() error
 	ConnectionState() tls.ConnectionState
 }
 
@@ -1268,7 +1269,7 @@ func (s *Server) NextProto(key string, nph ServeHandler) {
 }
 
 func (s *Server) hasNextProto(c net.Conn) (proto string, err error) {
-	tlsConn, ok := c.(*tls.Conn)
+	tlsConn, ok := c.(connTLSer)
 	if ok {
 		err = tlsConn.Handshake()
 		proto = tlsConn.ConnectionState().NegotiatedProtocol