openreplay/backend/internal/http/router/handlers.go

package router

import (
	gzip "github.com/klauspost/pgzip"
	"io"
	"io/ioutil"
	"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 := ioutil.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)
}