cam2mjpeg/mjpeg.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
		}
	}
}