76f0f3ff3d
The io/ioutil package has been deprecated as of Go 1.16 [1]. This commit replaces the existing io/ioutil functions with their new definitions in io and os packages. [1]: https://golang.org/doc/go1.16#ioutil Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package router
|
|
|
|
import (
|
|
gzip "github.com/klauspost/pgzip"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
func (e *Router) pushMessages(w http.ResponseWriter, r *http.Request, sessionID uint64, topicName string) {
|
|
start := time.Now()
|
|
body := http.MaxBytesReader(w, r.Body, e.cfg.BeaconSizeLimit)
|
|
defer body.Close()
|
|
|
|
var reader io.ReadCloser
|
|
var err error
|
|
|
|
switch r.Header.Get("Content-Encoding") {
|
|
case "gzip":
|
|
log.Println("Gzip", reader)
|
|
|
|
reader, err = gzip.NewReader(body)
|
|
if err != nil {
|
|
ResponseWithError(w, http.StatusInternalServerError, err, start, r.URL.Path, 0)
|
|
return
|
|
}
|
|
log.Println("Gzip reader init", reader)
|
|
defer reader.Close()
|
|
default:
|
|
reader = body
|
|
}
|
|
log.Println("Reader after switch:", reader)
|
|
buf, err := io.ReadAll(reader)
|
|
if err != nil {
|
|
ResponseWithError(w, http.StatusInternalServerError, err, start, r.URL.Path, 0)
|
|
return
|
|
}
|
|
log.Println("Produce message: ", buf, string(buf))
|
|
if err := e.services.Producer.Produce(topicName, sessionID, buf); err != nil {
|
|
ResponseWithError(w, http.StatusInternalServerError, err, start, r.URL.Path, 0)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|