diff --git a/src/connection_pool.c b/src/connection_pool.c index 779939c..09bb696 100644 --- a/src/connection_pool.c +++ b/src/connection_pool.c @@ -779,6 +779,7 @@ int conn_pool_global_init_for_rdma() LOAD_API(G_COMMON_CONNECTION_CALLBACKS[fc_comm_type_rdma], is_connected); + LOAD_API(G_RDMA_CONNECTION_CALLBACKS, set_busy_polling); LOAD_API(G_RDMA_CONNECTION_CALLBACKS, alloc_pd); LOAD_API(G_RDMA_CONNECTION_CALLBACKS, get_connection_size); LOAD_API(G_RDMA_CONNECTION_CALLBACKS, init_connection); diff --git a/src/connection_pool.h b/src/connection_pool.h index 08af0d3..7f8e29d 100644 --- a/src/connection_pool.h +++ b/src/connection_pool.h @@ -59,7 +59,7 @@ typedef struct { struct fc_server_config; struct ibv_pd; -typedef int (*fc_global_init_callback)(); +typedef void (*fc_set_busy_polling_callback)(const bool busy_polling); typedef struct ibv_pd *(*fc_alloc_pd_callback)(const char **ip_addrs, const int count, const int port); typedef int (*fc_get_connection_size_callback)(); @@ -98,7 +98,7 @@ typedef struct { } CommonConnectionCallbacks; typedef struct { - fc_global_init_callback global_init; + fc_set_busy_polling_callback set_busy_polling; fc_alloc_pd_callback alloc_pd; fc_get_connection_size_callback get_connection_size; fc_init_connection_callback init_connection;