mirror of
https://github.com/Luzifer/cam2mjpeg.git
synced 2024-09-19 15:23:05 +00:00
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/textproto"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func handleMJPEG(res http.ResponseWriter, r *http.Request, imgs chan []byte, uid string) {
|
|
if r.Method != "GET" {
|
|
http.Error(res, "405 Method Not Allowed", http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
logger := log.WithField("id", uid)
|
|
|
|
mimeWriter := multipart.NewWriter(res)
|
|
mimeWriter.SetBoundary("--boundary")
|
|
defer mimeWriter.Close()
|
|
|
|
res.Header().Add("Connection", "close")
|
|
res.Header().Add("Cache-Control", "no-store, no-cache")
|
|
res.Header().Add("Content-Type", fmt.Sprintf("multipart/x-mixed-replace;boundary=%s", mimeWriter.Boundary()))
|
|
|
|
cn := res.(http.CloseNotifier).CloseNotify()
|
|
errC := 0
|
|
|
|
for {
|
|
select {
|
|
case <-cn:
|
|
return
|
|
|
|
case img := <-imgs:
|
|
partHeader := make(textproto.MIMEHeader)
|
|
partHeader.Add("Content-Type", "image/jpeg")
|
|
|
|
partWriter, err := mimeWriter.CreatePart(partHeader)
|
|
if err != nil {
|
|
logger.WithError(err).Error("Unable to create mime part")
|
|
continue
|
|
}
|
|
|
|
_, err = partWriter.Write(img)
|
|
if err != nil {
|
|
logger.WithError(err).Error("Unable to write image")
|
|
errC++
|
|
|
|
if errC > 5 {
|
|
logger.Error("Too many errors, killing connection")
|
|
return
|
|
}
|
|
continue
|
|
}
|
|
|
|
errC = 0
|
|
}
|
|
}
|
|
}
|