Use asm implementation of base64 on amd64

handlers/socket.go

@@ -1,3 +1,5 @@
+// +build !amd64
+
 package handlers
 
 import (

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)
+	}
+}