Support SO_REUSEPORT (#12)

fagongzi · Sep 8, 2021 · 565a457 · 565a457
1 parent 2e3871f
commit 565a457
Show file tree

Hide file tree

Showing 14 changed files with 367 additions and 1 deletion.
diff --git a/application.go b/application.go
@@ -10,9 +10,15 @@ import (
 	"sync/atomic"
 	"time"
 
+	"github.com/valyala/tcplisten"
 	"go.uber.org/zap"
 )
 
+var cfg = &tcplisten.Config{
+	ReusePort: true,
+	FastOpen:  true,
+}
+
 var (
 	stateReadyToStart = int32(0)
 	stateStarting     = int32(1)
@@ -75,7 +81,7 @@ func NewApplication(listener net.Listener, handleFunc func(IOSession, interface{
 
 // NewTCPApplication returns a net application
 func NewTCPApplication(addr string, handleFunc func(IOSession, interface{}, uint64) error, opts ...AppOption) (NetApplication, error) {
-	listener, err := net.Listen("tcp", addr)
+	listener, err := cfg.NewListener("tcp4", addr)
 	if err != nil {
 		return nil, err
 	}

diff --git a/go.mod b/go.mod
@@ -5,5 +5,6 @@ go 1.16
 require (
 	github.com/fagongzi/util v0.0.0-20201116094402-221cc40c4593
 	github.com/stretchr/testify v1.7.0
+	github.com/valyala/tcplisten v1.0.0
 	go.uber.org/zap v1.18.1
 )
diff --git a/go.sum b/go.sum
@@ -19,6 +19,8 @@ github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UV
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
 github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/valyala/tcplisten v1.0.0 h1:rBHj/Xf+E1tRGZyWIWwJDiRY0zc1Js+CV5DqwacVSA8=
+github.com/valyala/tcplisten v1.0.0/go.mod h1:T0xQ8SeCZGxckz9qRXTfG43PvQ/mcWh7FwZEA7Ioqkc=
 go.uber.org/atomic v1.7.0 h1:ADUqmZGgLDDfbSL9ZmPxKTybcoEYHgpYfELNoN+7hsw=
 go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
 go.uber.org/goleak v1.1.10 h1:z+mqJhf6ss6BSfSM671tgKyZBFPTTJM+HLxnhPC3wu0=

diff --git a/vendor/github.com/valyala/tcplisten/.travis.yml b/vendor/github.com/valyala/tcplisten/.travis.yml
diff --git a/vendor/github.com/valyala/tcplisten/LICENSE b/vendor/github.com/valyala/tcplisten/LICENSE
diff --git a/vendor/github.com/valyala/tcplisten/README.md b/vendor/github.com/valyala/tcplisten/README.md
diff --git a/vendor/github.com/valyala/tcplisten/go.mod b/vendor/github.com/valyala/tcplisten/go.mod
diff --git a/vendor/github.com/valyala/tcplisten/socket.go b/vendor/github.com/valyala/tcplisten/socket.go
diff --git a/vendor/github.com/valyala/tcplisten/socket_darwin.go b/vendor/github.com/valyala/tcplisten/socket_darwin.go
diff --git a/vendor/github.com/valyala/tcplisten/socket_other.go b/vendor/github.com/valyala/tcplisten/socket_other.go
diff --git a/vendor/github.com/valyala/tcplisten/tcplisten.go b/vendor/github.com/valyala/tcplisten/tcplisten.go
diff --git a/vendor/github.com/valyala/tcplisten/tcplisten_bsd.go b/vendor/github.com/valyala/tcplisten/tcplisten_bsd.go