From 05033550051add5decb5f3edbcef90e34c7b914a Mon Sep 17 00:00:00 2001
From: Milan Nikolic <gen2brain@gmail.com>
Date: Sun, 28 Oct 2018 13:44:47 +0100
Subject: [PATCH] Use asm implementation of base64 on amd64

---
 handlers/socket.go       |  2 ++
 handlers/socket_amd64.go | 53 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 handlers/socket_amd64.go

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