Compress all responses if compression is enabled

ClickHouse · May 2, 2024 · a1b3d5d · a1b3d5d
1 parent 18c54b0
commit a1b3d5d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/conn_http.go b/conn_http.go
@@ -424,14 +424,14 @@ func (h *httpConnect) readRawResponse(response *http.Response) (body []byte, err
 	if err != nil {
 		return nil, err
 	}
-	if response.StatusCode == http.StatusOK && (h.compression == CompressionLZ4 || h.compression == CompressionZSTD) {
+	if h.compression == CompressionLZ4 || h.compression == CompressionZSTD {
 		chReader := chproto.NewReader(reader)
 		chReader.EnableCompression()
 		reader = chReader
 	}
 
 	body, err = io.ReadAll(reader)
-	if err != nil {
+	if err != nil && !errors.Is(err, io.EOF) {
 		return nil, err
 	}
 	return body, nil

diff --git a/conn_http_batch.go b/conn_http_batch.go
@@ -199,6 +199,7 @@ func (b *httpBatch) Send() (err error) {
 		headers["Content-Encoding"] = b.conn.compression.String()
 	case CompressionZSTD, CompressionLZ4:
 		options.settings["decompress"] = "1"
+		options.settings["compress"] = "1"
 	}
 
 	go func() {