[coro_rpc][go example]go example enable rdma (#961)

Author: qicosmos

src/coro_rpc/examples/base_examples/coro_rpc_lib/coro_rpc.cpp

@@ -57,10 +57,14 @@ rpc_result wait_response_finish(void *p) {
   return rpc_result{0, buf, 0};
 }
 
-void *start_rpc_server(char *addr, int parallel) {
+void *start_rpc_server(char *addr, int parallel, bool enable_ib) {
   auto server = std::make_unique<coro_rpc::coro_rpc_server>(
       parallel, addr, std::chrono::seconds(600));
   server->register_handler<ylt_load_service>();
+#ifdef YLT_ENABLE_IBV
+  if (enable_ib)
+    server->init_ibv();
+#endif
   auto res = server->async_start();
   if (res.hasResult()) {
     ELOG_ERROR << "start server failed";
@@ -79,12 +83,20 @@ void stop_rpc_server(void *server) {
 }
 
 // rpc client
-void *create_client_pool(char *addr, int req_timeout_sec) {
+void *create_client_pool(char *addr, int req_timeout_sec, bool enable_ib) {
   std::vector<std::string_view> hosts{std::string_view(addr)};
+  coro_io::client_pool<coro_rpc::coro_rpc_client>::pool_config pool_conf{};
+#ifdef YLT_ENABLE_IBV
+  if (enable_ib) {
+    coro_io::ibverbs_config ib_conf{};
+    pool_conf.client_config.socket_config = ib_conf;
+  }
+#endif
+  pool_conf.client_config.request_timeout_duration =
+      std::chrono::seconds{req_timeout_sec};
+
   auto ld = coro_io::load_balancer<coro_rpc::coro_rpc_client>::create(
-      hosts, {.pool_config{.client_config = {
-                               .request_timeout_duration =
-                                   std::chrono::seconds{req_timeout_sec}}}});
+      hosts, {pool_conf});
   auto ld_ptr = std::make_unique<decltype(ld)>(std::move(ld));
   return ld_ptr.release();
 }

src/coro_rpc/examples/base_examples/coro_rpc_lib/coro_rpc.h

@@ -17,10 +17,11 @@ extern void load_service(void *ctx, uint64_t req_id);
 extern void *response_msg(void *ctx, char *msg, uint64_t size);
 extern void response_error(void *ctx, uint16_t err_code, const char *err_msg);
 extern rpc_result wait_response_finish(void *p);
-extern void *start_rpc_server(char *addr, int parallel);
+extern void *start_rpc_server(char *addr, int parallel, bool enable_ib);
 extern void stop_rpc_server(void *server);
 
-extern void *create_client_pool(char *addr, int req_timeout_sec);
+extern void *create_client_pool(char *addr, int req_timeout_sec,
+                                bool enable_ib);
 extern void free_client_pool(void *pool);
 extern rpc_result load(void *pool, uint64_t req_id, char *dest,
                        uint64_t dest_len);

src/coro_rpc/examples/base_examples/go_example/test_rpc.go

@@ -9,7 +9,7 @@ import(
 /*
 #cgo CXXFLAGS: -std=c++20
 #cgo CFLAGS: -I../coro_rpc_lib
-#cgo LDFLAGS: -L./ -lcoro_rpc -lm -lstdc++
+#cgo LDFLAGS: -L./ -lcoro_rpc -lm -lstdc++ -libverbs
 #include "coro_rpc.h"
 */
 import "C"
@@ -54,7 +54,7 @@ func load_service(ctx unsafe.Pointer, req_id C.uint64_t) {
 
 func test_client(host string, len int) {
   peer := C.CString(host)
-  pool := C.create_client_pool(peer, C.int(30))
+  pool := C.create_client_pool(peer, C.int(30), C.bool(true))
 
   outbuf := make([]byte, len)
   for i := 0; i < 3; i++ {
@@ -90,7 +90,7 @@ func main()  {
   flag.Parse()
   fmt.Println(*addr, "thread number,", *thds, ", response data len:", *resp_len)
   g_resp_buf = make([]byte, *resp_len)
-  var server = C.start_rpc_server(C.CString(*addr), 96)
+  var server = C.start_rpc_server(C.CString(*addr), 96, C.bool(true))
   init_response_buf()
 
   // var quit1 string