blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-11-02 17:32:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: enable listen TIME_WAIT port

Browse Source
This commit is contained in:
icey-yu 2024-09-04 19:23:39 +08:00
parent 076205a290
commit 5555f31961
3 changed files with 53 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/common/startrpc/socket_posix.go Normal file
View File

@ -0,0 +1,26 @@
//go:build linux || darwin
package startrpc
import (
"context"
"github.com/openimsdk/tools/log"
"net"
"syscall"
)
func createListener() net.ListenConfig {
lc := net.ListenConfig{
Control: func(network, address string, conn syscall.RawConn) error {
return conn.Control(func(fd uintptr) {
err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1)
if err != nil {
log.ZError(context.Background(), "Failed to set socket flag to SO_REUSEADDR", err)
return
}
})
},
}
return lc
}

26
pkg/common/startrpc/socket_windows.go Normal file
View File

@ -0,0 +1,26 @@
//go:build windows
package startrpc
import (
"context"
"github.com/openimsdk/tools/log"
"net"
"syscall"
)
func createListener() net.ListenConfig {
lc := net.ListenConfig{
Control: func(network, address string, conn syscall.RawConn) error {
return conn.Control(func(fd uintptr) {
err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1)
if err != nil {
log.ZError(context.Background(), "Failed to set socket flag to SO_REUSEADDR", err)
return
}
})
},
}
return lc
}

8
pkg/common/startrpc/start.go
View File

@ -55,13 +55,7 @@ func Start[T any](ctx context.Context, discovery *config.Discovery, prometheusCo
"prometheusPorts", prometheusConfig.Ports)
rpcTcpAddr := net.JoinHostPort(network.GetListenIP(listenIP), strconv.Itoa(rpcPort))
lc := net.ListenConfig{
Control: func(network, address string, conn syscall.RawConn) error {
return conn.Control(func(fd uintptr) {
_ = syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1)
})
},
}
lc := createListener()
listener, err := lc.Listen(
ctx,
"tcp",