diff options
-rw-r--r-- | src/providers/data_provider_fo.c | 21 | ||||
-rw-r--r-- | src/providers/dp_backend.h | 1 | ||||
-rw-r--r-- | src/providers/fail_over.c | 13 | ||||
-rw-r--r-- | src/providers/fail_over.h | 5 |
4 files changed, 40 insertions, 0 deletions
diff --git a/src/providers/data_provider_fo.c b/src/providers/data_provider_fo.c index 1654e222..21059d39 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 61917b99..69872b6c 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 c965c0e3..f364a8c5 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 a31ace21..24b2c1f5 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. */ |