fix: supplement all the formal protocol formats of TCP and UDP

panjf2000 · Jun 6, 2020 · 258253c · 258253c
1 parent 7862f3e
commit 258253c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/gnet.go b/gnet.go
@@ -241,7 +241,7 @@ func Serve(eventHandler EventHandler, addr string, opts ...Option) (err error) {
 
 	ln.network, ln.addr = parseAddr(addr)
 	switch ln.network {
-	case "udp":
+	case "udp", "udp4", "udp6":
 		if options.ReusePort {
 			ln.pconn, err = netpoll.ReusePortListenPacket(ln.network, ln.addr)
 		} else {
@@ -254,7 +254,7 @@ func Serve(eventHandler EventHandler, addr string, opts ...Option) (err error) {
 			break
 		}
 		fallthrough
-	case "tcp":
+	case "tcp", "tcp4", "tcp6":
 		if options.ReusePort {
 			ln.ln, err = netpoll.ReusePortListen(ln.network, ln.addr)
 		} else {
@@ -282,7 +282,7 @@ func Serve(eventHandler EventHandler, addr string, opts ...Option) (err error) {
 
 func parseAddr(addr string) (network, address string) {
 	network = "tcp"
-	address = addr
+	address = strings.ToLower(addr)
 	if strings.Contains(address, "://") {
 		pair := strings.Split(address, "://")
 		network = pair[0]

diff --git a/gnet_test.go b/gnet_test.go
@@ -567,7 +567,7 @@ func TestDefaultGnetServer(t *testing.T) {
 }
 
 func TestTick(t *testing.T) {
-	testTick("tcp", ":9991", t)
+	testTick("tcp4", ":9991", t)
 }
 
 type testTickServer struct {
@@ -885,7 +885,7 @@ func testShutdownActionOnOpen(network, addr string) {
 }
 
 func TestUDPShutdown(t *testing.T) {
-	testUDPShutdown("udp", ":9000")
+	testUDPShutdown("udp4", ":9000")
 }
 
 type testUDPShutdownServer struct {