From f6de7550a4c8bca974574c71b7da19826ffa2705 Mon Sep 17 00:00:00 2001 From: Ronnie Sahlberg Date: Wed, 20 Jul 2011 15:43:55 +1000 Subject: [PATCH] ReadOnly: Add a test tool to fetch a record, requesting a readonly delegation and lock the record once (This used to be ctdb commit 29c77c249d0454057ce28003e018e126c7c01513) --- ctdb/Makefile.in | 6 +- ctdb/tests/src/ctdb_fetch_readonly_once.c | 134 ++++++++++++++++++++++ 2 files changed, 139 insertions(+), 1 deletion(-) create mode 100644 ctdb/tests/src/ctdb_fetch_readonly_once.c diff --git a/ctdb/Makefile.in b/ctdb/Makefile.in index 646b4be0dc2..9c5d364dacd 100755 --- a/ctdb/Makefile.in +++ b/ctdb/Makefile.in @@ -67,7 +67,7 @@ CTDB_SERVER_OBJ = server/ctdbd.o server/ctdb_daemon.o server/ctdb_lockwait.o \ $(CTDB_CLIENT_OBJ) $(CTDB_TCP_OBJ) @INFINIBAND_WRAPPER_OBJ@ TEST_BINS=tests/bin/ctdb_bench tests/bin/ctdb_fetch tests/bin/ctdb_fetch_one \ - tests/bin/ctdb_fetch_lock_once \ + tests/bin/ctdb_fetch_lock_once tests/bin/ctdb_fetch_readonly_once \ tests/bin/ctdb_store tests/bin/ctdb_trackingdb_test \ tests/bin/ctdb_randrec tests/bin/ctdb_persistent \ tests/bin/ctdb_traverse tests/bin/rb_test tests/bin/ctdb_transaction \ @@ -172,6 +172,10 @@ tests/bin/ctdb_fetch_lock_once: $(CTDB_CLIENT_OBJ) tests/src/ctdb_fetch_lock_onc @echo Linking $@ @$(CC) $(CFLAGS) -o $@ tests/src/ctdb_fetch_lock_once.o $(CTDB_CLIENT_OBJ) $(LIB_FLAGS) +tests/bin/ctdb_fetch_readonly_once: $(CTDB_CLIENT_OBJ) tests/src/ctdb_fetch_readonly_once.o + @echo Linking $@ + @$(CC) $(CFLAGS) -o $@ tests/src/ctdb_fetch_readonly_once.o $(CTDB_CLIENT_OBJ) $(LIB_FLAGS) + tests/bin/ctdb_trackingdb_test: $(CTDB_CLIENT_OBJ) tests/src/ctdb_trackingdb_test.o @echo Linking $@ @$(CC) $(CFLAGS) -o $@ tests/src/ctdb_trackingdb_test.o $(CTDB_CLIENT_OBJ) $(LIB_FLAGS) diff --git a/ctdb/tests/src/ctdb_fetch_readonly_once.c b/ctdb/tests/src/ctdb_fetch_readonly_once.c new file mode 100644 index 00000000000..662f4704d44 --- /dev/null +++ b/ctdb/tests/src/ctdb_fetch_readonly_once.c @@ -0,0 +1,134 @@ +/* + simple ctdb test tool + This test just fetch_locks a record and releases it once. + + Copyright (C) Ronnie Sahlberg 2009 + + 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 . +*/ + +#include "includes.h" +#include "lib/tevent/tevent.h" +#include "system/filesys.h" +#include "popt.h" +#include "cmdline.h" +#include "ctdb_private.h" + +static struct ctdb_db_context *ctdb_db; + +char *TESTKEY = "testkey"; + + +/* + Just try locking/unlocking a single record once +*/ +static void fetch_lock_once(struct ctdb_context *ctdb, struct event_context *ev) +{ + TALLOC_CTX *tmp_ctx = talloc_new(ctdb); + TDB_DATA key, data; + struct ctdb_record_handle *h; + + key.dptr = discard_const(TESTKEY); + key.dsize = strlen(TESTKEY); + + printf("Trying to fetch lock the record ...\n"); + + h = ctdb_fetch_readonly_lock(ctdb_db, tmp_ctx, key, &data, true); + if (h == NULL) { + printf("Failed to fetch record '%s' on node %d\n", + (const char *)key.dptr, ctdb_get_pnn(ctdb)); + talloc_free(tmp_ctx); + exit(10); + } + + printf("Record fetchlocked.\n"); + printf("Press enter to release the record ...\n"); + (void)getchar(); + + talloc_free(tmp_ctx); + printf("Record released.\n"); +} + +/* + main program +*/ +int main(int argc, const char *argv[]) +{ + struct ctdb_context *ctdb; + TDB_DATA key; + + struct poptOption popt_options[] = { + POPT_AUTOHELP + POPT_CTDB_CMDLINE + { "record", 'r', POPT_ARG_STRING, &TESTKEY, 0, "record", "string" }, + POPT_TABLEEND + }; + int opt, ret; + const char **extra_argv; + int extra_argc = 0; + poptContext pc; + struct event_context *ev; + + 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); + } + } + + /* 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 = event_context_init(NULL); + + ctdb = ctdb_cmdline_client(ev); + + key.dptr = TESTKEY; + key.dsize = strlen(TESTKEY); + + ret = ctdb_ctrl_getvnnmap(ctdb, timeval_zero(), CTDB_CURRENT_NODE, ctdb, &ctdb->vnn_map); + if (ret != 0) { + printf("failed to get vnnmap\n"); + exit(10); + } + printf("Record:%s\n", TESTKEY); + printf("Lmaster : %d\n", ctdb_lmaster(ctdb, &key)); + + /* attach to a specific database */ + ctdb_db = ctdb_attach(ctdb, "test.tdb", false, 0); + if (!ctdb_db) { + printf("ctdb_attach failed - %s\n", ctdb_errstr(ctdb)); + exit(1); + } + + printf("Waiting for cluster\n"); + while (1) { + uint32_t recmode=1; + ctdb_ctrl_getrecmode(ctdb, ctdb, timeval_zero(), CTDB_CURRENT_NODE, &recmode); + if (recmode == 0) break; + event_loop_once(ev); + } + + fetch_lock_once(ctdb, ev); + + return 0; +}