>From 0da6911d1918d6ce1cc625704885c89d806b3505 Mon Sep 17 00:00:00 2001 From: eindenbom Date: Thu, 17 Jun 2010 16:09:36 +0400 Subject: [PATCH 02/12] Added an interface to query number of configured (and currently resolved through SRV records) failover servers. --- src/providers/data_provider_fo.c | 21 +++++++++++++++++++++ src/providers/dp_backend.h | 1 + src/providers/fail_over.c | 13 +++++++++++++ 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 1654e22..21059d3 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 61917b9..69872b6 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 c965c0e..f364a8c 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 a31ace2..24b2c1f 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. */ -- 1.6.6.1