database (#4)

Co-authored-by: Evan Burkey <dev@fputs.com>
Reviewed-on: https://git.fputs.com/fputs/protohackers/pulls/4

pkg/conn/conn.go

@@ -8,17 +8,16 @@ import (
 
 const (
 	Port = ":3030"
-	Type = "tcp4"
 )
 
 func StartSimple(handler func(conn net.Conn)) error {
-	l, err := net.Listen(Type, Port)
+	l, err := net.Listen("tcp4", Port)
 	if err != nil {
 		return err
 	}
 	defer l.Close()
 
-	fmt.Println(fmt.Sprintf("Listening on port %s", Port))
+	fmt.Println(fmt.Sprintf("Listening TCP on port %s", Port))
 
 	for {
 		c, err := l.Accept()
@@ -26,9 +25,30 @@ func StartSimple(handler func(conn net.Conn)) error {
 			return err
 		}
 		go handler(c)
+	}
+}
+
+func StartSimpleUDP(handler func(*net.UDPConn, *net.UDPAddr, []byte), buffSize int) error {
+	srv, err := net.ResolveUDPAddr("udp4", Port)
+	if err != nil {
+		return err
+	}
+
+	c, err := net.ListenUDP("udp4", srv)
+	if err != nil {
+		return err
+	}
+	defer c.Close()
+
+	fmt.Println(fmt.Sprintf("Listening UDP on port %s", Port))
+
+	buf := make([]byte, buffSize)
+	for {
+		n, addr, err := c.ReadFromUDP(buf)
 		if err != nil {
 			return err
 		}
+		go handler(c, addr, buf[:n])
 	}
 }