Added an interface to query number of configured (and currently resolved through SRV records) failover servers.

4 files changed, 40 insertions, 0 deletions

diff --git a/src/providers/data_provider_fo.c b/src/providers/data_provider_fo.c
index 1654e2227..21059d394 100644
--- a/src/providers/data_provider_fo.c
+++ b/src/providers/data_provider_fo.c
@@ -277,6 +277,27 @@ int be_fo_add_srv_server(struct be_ctx *ctx, const char *service_name,
     return EOK;
 }
 
+int be_fo_get_server_count(struct be_ctx *ctx, const char *service_name)
+{
+    struct be_svc_data *svc_data;
+
+    if (!ctx->be_fo) {
+        return 0;
+    }
+
+    DLIST_FOR_EACH(svc_data, ctx->be_fo->svcs) {
+        if (strcmp(svc_data->name, service_name) == 0) {
+            break;
+        }
+    }
+
+    if (!svc_data) {
+        return 0;
+    }
+
+    return fo_get_server_count(svc_data->fo_service);
+}
+
 int be_fo_add_server(struct be_ctx *ctx, const char *service_name,
                      const char *server, int port, void *user_data)
 {
diff --git a/src/providers/dp_backend.h b/src/providers/dp_backend.h
index 61917b99d..69872b6c4 100644
--- a/src/providers/dp_backend.h
+++ b/src/providers/dp_backend.h
@@ -165,6 +165,7 @@ int be_fo_add_service(struct be_ctx *ctx, const char *service_name);
 int be_fo_service_add_callback(TALLOC_CTX *memctx,
                                struct be_ctx *ctx, const char *service_name,
                                be_svc_callback_fn_t *fn, void *private_data);
+int be_fo_get_server_count(struct be_ctx *ctx, const char *service_name);
 int be_fo_add_srv_server(struct be_ctx *ctx, const char *service_name,
                          const char *query_service, const char *proto,
                          void *user_data);
diff --git a/src/providers/fail_over.c b/src/providers/fail_over.c
index c965c0e3a..f364a8c5f 100644
--- a/src/providers/fail_over.c
+++ b/src/providers/fail_over.c
@@ -585,6 +585,19 @@ create_fo_server(struct fo_service *service, const char *name,
 }
 
 int
+fo_get_server_count(struct fo_service *service)
+{
+    struct fo_server *server;
+    int count = 0;
+
+    DLIST_FOR_EACH(server, service->server_list) {
+        count++;
+    }
+
+    return count;
+}
+
+int
 fo_add_server(struct fo_service *service, const char *name, int port,
               void *user_data)
 {
diff --git a/src/providers/fail_over.h b/src/providers/fail_over.h
index a31ace21e..24b2c1f5f 100644
--- a/src/providers/fail_over.h
+++ b/src/providers/fail_over.h
@@ -100,6 +100,11 @@ int fo_get_service(struct fo_ctx *ctx,
                    struct fo_service **_service);
 
 /*
+ * Get number of servers registered for the 'service'.
+ */
+int fo_get_server_count(struct fo_service *service);
+
+/*
  * Adds a server 'name' to the 'service'. Port 'port' will be used for
  * connection. If 'name' is NULL, no server resolution will be done.
  */