diff --git a/handlers/socket.go b/handlers/socket.go index 0ced35f..15439bc 100644 --- a/handlers/socket.go +++ b/handlers/socket.go @@ -1,3 +1,5 @@ +// +build !amd64 + package handlers import ( diff --git a/handlers/socket_amd64.go b/handlers/socket_amd64.go new file mode 100644 index 0000000..7568c18 --- /dev/null +++ b/handlers/socket_amd64.go @@ -0,0 +1,53 @@ +package handlers + +import ( + "bytes" + "log" + "time" + + "golang.org/x/net/websocket" + "goost.org/encoding/base64" + + "github.com/gen2brain/cam2ip/encoder" + "github.com/gen2brain/cam2ip/reader" +) + +// Socket handler. +type Socket struct { + reader reader.ImageReader + delay int +} + +// NewSocket returns new socket handler. +func NewSocket(reader reader.ImageReader, delay int) websocket.Handler { + s := &Socket{reader, delay} + return websocket.Handler(s.write) +} + +// write writes images to socket +func (s *Socket) write(ws *websocket.Conn) { + for { + img, err := s.reader.Read() + if err != nil { + log.Printf("socket: read: %v", err) + break + } + + w := new(bytes.Buffer) + + err = encoder.New(w).Encode(img) + if err != nil { + log.Printf("socket: encode: %v", err) + continue + } + + b64 := base64.StdEncoding.EncodeToString(w.Bytes()) + + _, err = ws.Write([]byte(b64)) + if err != nil { + break + } + + time.Sleep(time.Duration(s.delay) * time.Millisecond) + } +}