diff --git a/server.go b/server.go index 6bb4d97..7c5522a 100644 --- a/server.go +++ b/server.go @@ -144,9 +144,11 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) { if err != nil { fields[log.FnError] = err.Error() _ = s.logger.Warn("peekHTTP failed", fields) - } else { - if err == nil && host != "" { + } else if host != "" { + if !strings.Contains(host, ":") { addr = host + addr[strings.Index(addr, ":"):] + } else { + addr = host } } reader = reader_n3