diff options
Diffstat (limited to 'source4/cluster/ctdb/tests/ctdb_fetch.c')
-rw-r--r-- | source4/cluster/ctdb/tests/ctdb_fetch.c | 262 |
1 files changed, 0 insertions, 262 deletions
diff --git a/source4/cluster/ctdb/tests/ctdb_fetch.c b/source4/cluster/ctdb/tests/ctdb_fetch.c deleted file mode 100644 index d5ec645757..0000000000 --- a/source4/cluster/ctdb/tests/ctdb_fetch.c +++ /dev/null @@ -1,262 +0,0 @@ -/* - simple ctdb benchmark - - Copyright (C) Andrew Tridgell 2006 - - This program is free software; you can redistribute it and/or modify - it under the terms of the GNU General Public License as published by - the Free Software Foundation; either version 3 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. - - You should have received a copy of the GNU General Public License - along with this program; if not, see <http://www.gnu.org/licenses/>. -*/ - -#include "includes.h" -#include "lib/events/events.h" -#include "system/filesys.h" -#include "popt.h" -#include "cmdline.h" - -#include <sys/time.h> -#include <time.h> - -static struct timeval tp1,tp2; - -static void start_timer(void) -{ - gettimeofday(&tp1,NULL); -} - -static double end_timer(void) -{ - gettimeofday(&tp2,NULL); - return (tp2.tv_sec + (tp2.tv_usec*1.0e-6)) - - (tp1.tv_sec + (tp1.tv_usec*1.0e-6)); -} - - -static int timelimit = 10; -static int num_records = 10; -static int num_nodes; -static int msg_count; - -#define TESTKEY "testkey" - -/* - fetch a record - store a expanded record - send a message to next node to tell it to do the same -*/ -static void bench_fetch_1node(struct ctdb_context *ctdb) -{ - TDB_DATA key, data, nulldata; - struct ctdb_db_context *ctdb_db; - TALLOC_CTX *tmp_ctx = talloc_new(ctdb); - int dest, ret; - struct ctdb_record_handle *h; - - key.dptr = discard_const(TESTKEY); - key.dsize = strlen(TESTKEY); - - ctdb_db = ctdb_db_handle(ctdb, "test.tdb"); - - h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data); - if (h == NULL) { - printf("Failed to fetch record '%s' on node %d\n", - (const char *)key.dptr, ctdb_get_vnn(ctdb)); - talloc_free(tmp_ctx); - return; - } - - if (data.dsize > 1000) { - data.dsize = 0; - } - - if (data.dsize == 0) { - data.dptr = (uint8_t *)talloc_asprintf(tmp_ctx, "Test data\n"); - } - data.dptr = (uint8_t *)talloc_asprintf_append((char *)data.dptr, - "msg_count=%d on node %d\n", - msg_count, ctdb_get_vnn(ctdb)); - data.dsize = strlen((const char *)data.dptr)+1; - - ret = ctdb_record_store(h, data); - talloc_free(h); - if (ret != 0) { - printf("Failed to store record\n"); - } - - talloc_free(tmp_ctx); - - /* tell the next node to do the same */ - nulldata.dptr = NULL; - nulldata.dsize = 0; - - dest = (ctdb_get_vnn(ctdb) + 1) % num_nodes; - ctdb_send_message(ctdb, dest, 0, nulldata); -} - -/* - handler for messages in bench_ring() -*/ -static void message_handler(struct ctdb_context *ctdb, uint64_t srvid, - TDB_DATA data, void *private_data) -{ - msg_count++; - bench_fetch_1node(ctdb); -} - - -/* - benchmark the following: - - fetch a record - store a expanded record - send a message to next node to tell it to do the same - -*/ -static void bench_fetch(struct ctdb_context *ctdb, struct event_context *ev) -{ - int vnn=ctdb_get_vnn(ctdb); - - if (vnn == num_nodes - 1) { - bench_fetch_1node(ctdb); - } - - start_timer(); - - while (end_timer() < timelimit) { - if (vnn == 0 && msg_count % 100 == 0) { - printf("Fetch: %.2f msgs/sec\r", msg_count/end_timer()); - fflush(stdout); - } - if (event_loop_once(ev) != 0) { - printf("Event loop failed!\n"); - break; - } - } - - printf("Fetch: %.2f msgs/sec\n", msg_count/end_timer()); -} - -enum my_functions {FUNC_FETCH=1}; - -/* - ctdb call function to fetch a record -*/ -static int fetch_func(struct ctdb_call_info *call) -{ - call->reply_data = &call->record_data; - return 0; -} - -/* - handler for reconfigure message -*/ -static void reconfigure_handler(struct ctdb_context *ctdb, uint64_t srvid, - TDB_DATA data, void *private_data) -{ - int *ready = (int *)private_data; - *ready = 1; -} - -/* - main program -*/ -int main(int argc, const char *argv[]) -{ - struct ctdb_context *ctdb; - struct ctdb_db_context *ctdb_db; - - struct poptOption popt_options[] = { - POPT_AUTOHELP - POPT_CTDB_CMDLINE - { "timelimit", 't', POPT_ARG_INT, &timelimit, 0, "timelimit", "integer" }, - { "num-records", 'r', POPT_ARG_INT, &num_records, 0, "num_records", "integer" }, - { NULL, 'n', POPT_ARG_INT, &num_nodes, 0, "num_nodes", "integer" }, - POPT_TABLEEND - }; - int opt; - const char **extra_argv; - int extra_argc = 0; - int ret; - poptContext pc; - struct event_context *ev; - struct ctdb_call call; - int cluster_ready=0; - - pc = poptGetContext(argv[0], argc, argv, popt_options, POPT_CONTEXT_KEEP_FIRST); - - while ((opt = poptGetNextOpt(pc)) != -1) { - switch (opt) { - default: - fprintf(stderr, "Invalid option %s: %s\n", - poptBadOption(pc, 0), poptStrerror(opt)); - exit(1); - } - } - - /* talloc_enable_leak_report_full(); */ - - /* setup the remaining options for the main program to use */ - extra_argv = poptGetArgs(pc); - if (extra_argv) { - extra_argv++; - while (extra_argv[extra_argc]) extra_argc++; - } - - ev = s4_event_context_init(NULL); - - ctdb = ctdb_cmdline_client(ev); - - ctdb_set_message_handler(ctdb, CTDB_SRVID_RECONFIGURE, reconfigure_handler, - &cluster_ready); - - /* attach to a specific database */ - ctdb_db = ctdb_attach(ctdb, "test.tdb"); - if (!ctdb_db) { - printf("ctdb_attach failed - %s\n", ctdb_errstr(ctdb)); - exit(1); - } - - ret = ctdb_set_call(ctdb_db, fetch_func, FUNC_FETCH); - - ctdb_set_message_handler(ctdb, 0, message_handler, &msg_count); - - printf("Waiting for cluster\n"); - while (1) { - uint32_t recmode=1; - ctdb_ctrl_getrecmode(ctdb, timeval_zero(), CTDB_CURRENT_NODE, &recmode); - if (recmode == 0) break; - event_loop_once(ev); - } - - bench_fetch(ctdb, ev); - - ZERO_STRUCT(call); - call.key.dptr = discard_const(TESTKEY); - call.key.dsize = strlen(TESTKEY); - - printf("Fetching final record\n"); - - /* fetch the record */ - call.call_id = FUNC_FETCH; - call.call_data.dptr = NULL; - call.call_data.dsize = 0; - - ret = ctdb_call(ctdb_db, &call); - if (ret == -1) { - printf("ctdb_call FUNC_FETCH failed - %s\n", ctdb_errstr(ctdb)); - exit(1); - } - - printf("DATA:\n%s\n", (char *)call.reply_data.dptr); - - return 0; -} |