1 files changed, 4 insertions, 0 deletions
diff --git a/src/net.h b/src/net.h
index 8f44ab3..19fdb35 100644
--- a/src/net.h
+++ b/src/net.h
@@ -80,6 +80,8 @@ typedef enum perf_socket_event {
 /* Callback for socket events related to connection oriented protocols, for statistics */
 typedef void (*perf_net_event_cb_t)(struct perf_net_socket* sock, perf_socket_event_t event, uint64_t elapsed_time);
 
+typedef void (*perf_net_num_queries_per_conn_t)(struct perf_net_socket* sock, size_t num_queries_per_conn, size_t timeout);
+
 struct perf_net_socket {
     void* data; /* user data */
 
@@ -91,6 +93,8 @@ struct perf_net_socket {
     perf_net_sockready_t sockready;
     perf_net_have_more_t have_more;
 
+    perf_net_num_queries_per_conn_t num_queries_per_conn;
+
     /*
      * Not set by protocol, set by caller.
      * May be 0 if caller don't care.