Skip to content

Commit 4ffba56

Browse files
author
chaoyuepan
committed
#883 use the rdma build tag
1 parent 60a3812 commit 4ffba56

File tree

3 files changed

+38
-26
lines changed

3 files changed

+38
-26
lines changed

โ€Žclient/connection_rdma.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
//go:build linux
2-
// +build linux
1+
//go:build rdma
2+
// +build rdma
33

44
package client
55

โ€Žserver/listener_linux.go

-24
Original file line numberDiff line numberDiff line change
@@ -5,21 +5,17 @@ package server
55

66
import (
77
"net"
8-
"os"
98
"runtime"
10-
"strconv"
119
"time"
1210

1311
uringnet "github.com/godzie44/go-uring/net"
1412
"github.com/godzie44/go-uring/reactor"
1513
"github.com/godzie44/go-uring/uring"
1614
"github.com/smallnest/rpcx/log"
17-
"github.com/smallnest/rsocket"
1815
)
1916

2017
func init() {
2118
makeListeners["iouring"] = iouringMakeListener
22-
makeListeners["rdma"] = rdmaMakeListener
2319
}
2420

2521
// iouringMakeListener creates a new listener using io_uring.
@@ -68,23 +64,3 @@ type uringLogger struct {
6864
func (l *uringLogger) Log(keyvals ...interface{}) {
6965
l.Logger.Info(keyvals...)
7066
}
71-
72-
func rdmaMakeListener(s *Server, address string) (ln net.Listener, err error) {
73-
host, port, err := net.SplitHostPort(address)
74-
if err != nil {
75-
return nil, err
76-
}
77-
p, err := strconv.Atoi(port)
78-
if err != nil {
79-
return nil, err
80-
}
81-
backlog := os.Getenv("RDMA_BACKLOG")
82-
if backlog == "" {
83-
backlog = "128"
84-
}
85-
blog, _ := strconv.Atoi(backlog)
86-
if blog == 0 {
87-
blog = 128
88-
}
89-
return rsocket.NewTCPListener(host, p, blog)
90-
}

โ€Žserver/listener_rdma.go

+36
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
//go:build rdma
2+
// +build rdma
3+
4+
package server
5+
6+
import (
7+
"net"
8+
"os"
9+
"strconv"
10+
11+
"github.com/smallnest/rsocket"
12+
)
13+
14+
func init() {
15+
makeListeners["rdma"] = rdmaMakeListener
16+
}
17+
18+
func rdmaMakeListener(s *Server, address string) (ln net.Listener, err error) {
19+
host, port, err := net.SplitHostPort(address)
20+
if err != nil {
21+
return nil, err
22+
}
23+
p, err := strconv.Atoi(port)
24+
if err != nil {
25+
return nil, err
26+
}
27+
backlog := os.Getenv("RDMA_BACKLOG")
28+
if backlog == "" {
29+
backlog = "128"
30+
}
31+
blog, _ := strconv.Atoi(backlog)
32+
if blog == 0 {
33+
blog = 128
34+
}
35+
return rsocket.NewTCPListener(host, p, blog)
36+
}

0 commit comments

Comments
ย (0)