feat(udp): Added udp server

udp/server_test.go

@@ -0,0 +1,62 @@
+package udp
+
+import (
+	"context"
+	"net"
+	"sync"
+	"testing"
+	"time"
+)
+
+func TestServer(t *testing.T) {
+	var (
+		server *Server
+		wg     sync.WaitGroup
+		done   = make(chan []byte, 1)
+	)
+
+	wg.Add(3)
+
+	go func() {
+		defer wg.Done()
+
+		server = NewServer(":12190", WithHandler(func(conn net.PacketConn, buf []byte, addr net.Addr) {
+			done <- buf
+		}), WithRecoveryHandler(func(conn net.PacketConn, buf []byte, addr net.Addr, err interface{}) {
+			t.Log(err)
+		}), WithBufSize(1024))
+
+		server.Start(context.Background())
+	}()
+
+	go func() {
+		defer wg.Done()
+
+		c, err := net.Dial("udp", ":12190")
+		if err != nil {
+			t.Error(err)
+			return
+		}
+		defer c.Close()
+
+		_, err = c.Write([]byte("test"))
+		if err != nil {
+			t.Error(err)
+			return
+		}
+	}()
+
+	go func() {
+		defer wg.Done()
+
+		time.Sleep(time.Second * 1)
+		server.Stop(context.Background())
+	}()
+
+	wg.Wait()
+
+	buf := <-done
+	if string(buf) != "test" {
+		t.Fatal("buf not equal test")
+	}
+}