github-mirrors/VictoriaMetrics
1
0
Fork 0
mirror of https://github.com/VictoriaMetrics/VictoriaMetrics.git synced 2024-11-21 14:44:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

lib/httpserver: properly set status code for empty response

Browse source
This commit is contained in:
Aliaksandr Valialkin 2020-05-24 23:55:28 +03:00
parent 947009f459
commit f5a4731412
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/httpserver/httpserver.go
View file

@ -365,10 +365,7 @@ func (zrw *gzipResponseWriter) Write(p []byte) (int, error) {
h.Set("Content-Type", "text/html")
}
}
if zrw.statusCode == 0 {
zrw.statusCode = http.StatusOK
}
zrw.ResponseWriter.WriteHeader(zrw.statusCode)
zrw.writeHeader()
zrw.firstWriteDone = true
}
if zrw.disableCompression {
@ -378,12 +375,16 @@ func (zrw *gzipResponseWriter) Write(p []byte) (int, error) {
}
func (zrw *gzipResponseWriter) WriteHeader(statusCode int) {
if zrw.statusCode != 0 {
return
}
zrw.statusCode = statusCode
}
func (zrw *gzipResponseWriter) writeHeader() {
if zrw.statusCode == 0 {
zrw.statusCode = http.StatusOK
}
zrw.ResponseWriter.WriteHeader(zrw.statusCode)
}
// Implements http.Flusher
func (zrw *gzipResponseWriter) Flush() {
if err := zrw.bw.Flush(); err != nil && !isTrivialNetworkError(err) {
@ -399,6 +400,7 @@ func (zrw *gzipResponseWriter) Flush() {
func (zrw *gzipResponseWriter) Close() error {
if !zrw.firstWriteDone {
zrw.writeHeader()
return nil
}
zrw.Flush()