udp in bg thread

src/device/input/raw/raw.go

@@ -16,7 +16,7 @@ func New(ctx context.Context) Raw {
 		return NewKeyboard()
 	}
 	if port, _ := strconv.Atoi(os.Getenv("RAW_UDP")); port != 0 {
-		return NewUDP(port)
+		return NewUDP(ctx, port)
 	}
 	generator := randomCharFromRange('a', 'g')
 	if p, ok := os.LookupEnv("RAW_RANDOM_WEIGHT_FILE"); ok && len(p) > 0 {

src/device/input/raw/udp.go

@@ -1,6 +1,7 @@
 package raw
 
 import (
+	"context"
 	"log"
 	"net"
 	"os"
@@ -9,29 +10,41 @@ import (
 
 type UDP struct {
 	conn net.PacketConn
+	c    chan []byte
+	ctx  context.Context
 }
 
-func NewUDP(port int) UDP {
+func NewUDP(ctx context.Context, port int) UDP {
 	conn, err := net.ListenPacket("udp", ":"+strconv.Itoa(port))
 	if err != nil {
 		panic(err)
 	}
-	return UDP{
+	result := UDP{
 		conn: conn,
+		c:    make(chan []byte, 8),
+		ctx:  ctx,
 	}
+	go result.listen()
+	return result
 }
 
-func (udp UDP) Read() []byte {
+func (udp UDP) listen() {
-	panic("NEEDS TO BE IN BG THREAD SO WE CAN SHUT DOWN WITHOUT BLOCKING ON READ OR AT LEAST BE RESPONSIVE")
+	debugging := os.Getenv("DEBUG") == "true"
-	buff := make([]byte, 1024)
+	for udp.ctx.Err() == nil {
+		buff := make([]byte, 256)
 		n, _, err := udp.conn.ReadFrom(buff)
 		if err != nil {
 			panic(err)
 		}
-	if os.Getenv("DEBUG") == "true" {
+		if debugging {
 			log.Printf("raw.UDP.Read() => %s", buff[:n])
 		}
-	return buff[:n]
+		udp.c <- buff[:n]
+	}
+}
+
+func (udp UDP) Read() []byte {
+	return <-udp.c
 }
 
 func (udp UDP) Close() {