#883 use the rdma build tag

This commit is contained in:
chaoyuepan
2024-12-02 18:28:54 +08:00
parent 60a3812ae7
commit 4ffba56caa
3 changed files with 38 additions and 26 deletions

View File

@@ -1,5 +1,5 @@
//go:build linux
// +build linux
//go:build rdma
// +build rdma
package client

View File

@@ -5,21 +5,17 @@ package server
import (
"net"
"os"
"runtime"
"strconv"
"time"
uringnet "github.com/godzie44/go-uring/net"
"github.com/godzie44/go-uring/reactor"
"github.com/godzie44/go-uring/uring"
"github.com/smallnest/rpcx/log"
"github.com/smallnest/rsocket"
)
func init() {
makeListeners["iouring"] = iouringMakeListener
makeListeners["rdma"] = rdmaMakeListener
}
// iouringMakeListener creates a new listener using io_uring.
@@ -68,23 +64,3 @@ type uringLogger struct {
func (l *uringLogger) Log(keyvals ...interface{}) {
l.Logger.Info(keyvals...)
}
func rdmaMakeListener(s *Server, address string) (ln net.Listener, err error) {
host, port, err := net.SplitHostPort(address)
if err != nil {
return nil, err
}
p, err := strconv.Atoi(port)
if err != nil {
return nil, err
}
backlog := os.Getenv("RDMA_BACKLOG")
if backlog == "" {
backlog = "128"
}
blog, _ := strconv.Atoi(backlog)
if blog == 0 {
blog = 128
}
return rsocket.NewTCPListener(host, p, blog)
}

36
server/listener_rdma.go Normal file
View File

@@ -0,0 +1,36 @@
//go:build rdma
// +build rdma
package server
import (
"net"
"os"
"strconv"
"github.com/smallnest/rsocket"
)
func init() {
makeListeners["rdma"] = rdmaMakeListener
}
func rdmaMakeListener(s *Server, address string) (ln net.Listener, err error) {
host, port, err := net.SplitHostPort(address)
if err != nil {
return nil, err
}
p, err := strconv.Atoi(port)
if err != nil {
return nil, err
}
backlog := os.Getenv("RDMA_BACKLOG")
if backlog == "" {
backlog = "128"
}
blog, _ := strconv.Atoi(backlog)
if blog == 0 {
blog = 128
}
return rsocket.NewTCPListener(host, p, blog)
}