ClickHouse/tests/queries/0_stateless/02013_zlib_read_after_eof.go

62 lines
1.0 KiB
Go
Raw Normal View History

2021-08-25 17:42:57 +00:00
package main
import (
"compress/gzip"
"fmt"
"io"
"io/ioutil"
"net/http"
"net/url"
"os"
)
func compress(data io.Reader) io.Reader {
pr, pw := io.Pipe()
gw := gzip.NewWriter(pw)
go func() {
_, _ = io.Copy(gw, data)
gw.Close()
pw.Close()
}()
return pr
}
func main() {
2021-08-26 09:44:10 +00:00
database := os.Getenv("CLICKHOUSE_DATABASE")
2021-08-25 17:42:57 +00:00
p, err := url.Parse("http://localhost:8123/")
if err != nil {
panic(err)
}
q := p.Query()
2021-08-26 09:44:10 +00:00
q.Set("query", "INSERT INTO "+database+".graphite FORMAT RowBinary")
2021-08-25 17:42:57 +00:00
p.RawQuery = q.Encode()
queryUrl := p.String()
var req *http.Request
req, err = http.NewRequest("POST", queryUrl, compress(os.Stdin))
req.Header.Add("Content-Encoding", "gzip")
if err != nil {
panic(err)
}
client := &http.Client{
Transport: &http.Transport{DisableKeepAlives: true},
}
resp, err := client.Do(req)
if err != nil {
panic(err)
}
defer resp.Body.Close()
body, _ := ioutil.ReadAll(resp.Body)
if resp.StatusCode != 200 {
panic(fmt.Errorf("clickhouse response status %d: %s", resp.StatusCode, string(body)))
}
}