[t:5062] Add some additional tests for xids independent from lsns

git-svn-id: file:///svn/toku/tokudb@44669 c7de825b-a66e-492c-adef-691d508d4ae1

ft/tests/xid_lsn_independent.c

@@ -13,35 +13,136 @@
 #define TESTDIR __SRCFILE__ ".dir"
 #define FILENAME "test0.ft"
 
-static void test_it (int N) {
-    FT_HANDLE brt;
+static void test_setup(TOKULOGGER *loggerp, CACHETABLE *ctp) {
+    *loggerp = NULL;
+    *ctp = NULL;
     int r;
     r = system("rm -rf " TESTDIR);
     CKERR(r);
-    r = toku_os_mkdir(TESTDIR, S_IRWXU);                                                                    CKERR(r);
+    r = toku_os_mkdir(TESTDIR, S_IRWXU);
+    CKERR(r);
 
-    TOKULOGGER logger;
-    r = toku_logger_create(&logger);                                                                        CKERR(r);
-    r = toku_logger_open(TESTDIR, logger);                                                                  CKERR(r);
+    r = toku_logger_create(loggerp);
+    CKERR(r);
+    TOKULOGGER logger = *loggerp;
 
+    r = toku_logger_open(TESTDIR, logger);
+    CKERR(r);
 
-    CACHETABLE ct;
-    r = toku_create_cachetable(&ct, 0, ZERO_LSN, logger);                                               CKERR(r);
+    r = toku_create_cachetable(ctp, 0, ZERO_LSN, logger);
+    CKERR(r);
+    CACHETABLE ct = *ctp;
     toku_cachetable_set_env_dir(ct, TESTDIR);
 
     toku_logger_set_cachetable(logger, ct);
 
-    r = toku_logger_open_rollback(logger, ct, TRUE);                                                        CKERR(r);
+    r = toku_logger_open_rollback(logger, ct, TRUE);
+    CKERR(r);
+
+    r = toku_checkpoint(ct, logger, NULL, NULL, NULL, NULL, STARTUP_CHECKPOINT);
+    CKERR(r);
+}
+
+static void
+xid_lsn_keep_cachetable_callback (DB_ENV *env, CACHETABLE cachetable) {
+    CACHETABLE *ctp = (void*)env;
+    *ctp = cachetable;
+}
+
+static void test_setup_and_recover(TOKULOGGER *loggerp, CACHETABLE *ctp) {
+    int r;
+    TOKULOGGER logger = NULL;
+    CACHETABLE ct = NULL;
+    r = toku_logger_create(&logger);
+    CKERR(r);
+
+    r = tokudb_recover((DB_ENV*)&ct,
+                       NULL_prepared_txn_callback,
+                       xid_lsn_keep_cachetable_callback,
+                       logger,
+                       TESTDIR, TESTDIR, 0, 0, 0, NULL, 0);
+    CKERR(r);
+    if (!toku_logger_is_open(logger)) {
+        //Did not need recovery.
+        invariant(ct==NULL);
+        r = toku_logger_open(TESTDIR, logger);
+        CKERR(r);
+        r = toku_create_cachetable(&ct, 0, ZERO_LSN, logger);
+        CKERR(r);
+        toku_logger_set_cachetable(logger, ct);
+    }
+    *ctp = ct;
+    *loggerp = logger;
+}
+
+static void clean_shutdown(TOKULOGGER *loggerp, CACHETABLE *ctp) {
+    int r;
+    r = toku_checkpoint(*ctp, *loggerp, NULL, NULL, NULL, NULL, SHUTDOWN_CHECKPOINT);
+    CKERR(r);
+
+    r = toku_logger_close_rollback(*loggerp, false);
+    CKERR(r);
+
+    r = toku_checkpoint(*ctp, *loggerp, NULL, NULL, NULL, NULL, SHUTDOWN_CHECKPOINT);
+    CKERR(r);
+
+    r = toku_logger_shutdown(*loggerp);
+    CKERR(r);
+
+    r = toku_cachetable_close(ctp);
+    CKERR(r);
+
+    r = toku_logger_close(loggerp);
+    CKERR(r);
+}
+
+static void shutdown_after_recovery(TOKULOGGER *loggerp, CACHETABLE *ctp) {
+    int r;
+    r = toku_logger_close_rollback(*loggerp, false);
+    CKERR(r);
+    r = toku_cachetable_close(ctp);
+    CKERR(r);
+    r = toku_logger_close(loggerp);
+    CKERR(r);
+}
+
+static void do_txn(TOKULOGGER logger, bool readonly) {
+    int r;
+    TOKUTXN txn;
+    r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn, logger, TXN_SNAPSHOT_ROOT);
+    CKERR(r);
+
+    if (!readonly) {
+        toku_maybe_log_begin_txn_for_write_operation(txn);
+    }
+    r = toku_txn_commit_txn(txn, FALSE, NULL, NULL);
+    CKERR(r);
+
+    toku_txn_close_txn(txn);
+}
+
+static void test_xid_lsn_independent(int N) {
+    TOKULOGGER logger;
+    CACHETABLE ct;
+    test_setup(&logger, &ct);
+
+    FT_HANDLE brt;
+
+    int r;
 
     TOKUTXN txn;
-    r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn, logger, TXN_SNAPSHOT_ROOT);                     CKERR(r);
+    r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn, logger, TXN_SNAPSHOT_ROOT);
+    CKERR(r);
 
-    r = toku_open_ft_handle(FILENAME, 1, &brt, 1024, 256, TOKU_DEFAULT_COMPRESSION_METHOD, ct, txn, toku_builtin_compare_fun);                    CKERR(r);
+    r = toku_open_ft_handle(FILENAME, 1, &brt, 1024, 256, TOKU_DEFAULT_COMPRESSION_METHOD, ct, txn, toku_builtin_compare_fun);
+    CKERR(r);
 
-    r = toku_txn_commit_txn(txn, FALSE, NULL, NULL);                                 CKERR(r);
+    r = toku_txn_commit_txn(txn, FALSE, NULL, NULL);
+    CKERR(r);
     toku_txn_close_txn(txn);
 
-    r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn, logger, TXN_SNAPSHOT_ROOT);                 CKERR(r);
+    r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn, logger, TXN_SNAPSHOT_ROOT);
+    CKERR(r);
     TXNID xid_first = txn->txnid64;
     unsigned int rands[N];
     for (int i=0; i<N; i++) {
@@ -56,40 +157,110 @@ static void test_it (int N) {
     }
     {
         TOKUTXN txn2;
-        r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn2, logger, TXN_SNAPSHOT_ROOT);                 CKERR(r);
+        r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn2, logger, TXN_SNAPSHOT_ROOT);
+    CKERR(r);
         // Verify the txnid has gone up only by one (even though many log entries were done)
         invariant(txn2->txnid64 == xid_first + 1);
-        r = toku_txn_commit_txn(txn2, FALSE, NULL, NULL);                                 CKERR(r);
+        r = toku_txn_commit_txn(txn2, FALSE, NULL, NULL);
+    CKERR(r);
         toku_txn_close_txn(txn2);
     }
-    r = toku_txn_commit_txn(txn, FALSE, NULL, NULL);                                 CKERR(r);
+    r = toku_txn_commit_txn(txn, FALSE, NULL, NULL);
+    CKERR(r);
     toku_txn_close_txn(txn);
     {
         //TODO(yoni) #5067 will break this portion of the test. (End ids are also assigned, so it would increase by 4 instead of 2.)
         // Verify the txnid has gone up only by two (even though many log entries were done)
         TOKUTXN txn3;
-        r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn3, logger, TXN_SNAPSHOT_ROOT);                 CKERR(r);
+        r = toku_txn_begin_txn((DB_TXN*)NULL, (TOKUTXN)0, &txn3, logger, TXN_SNAPSHOT_ROOT);
+    CKERR(r);
         invariant(txn3->txnid64 == xid_first + 2);
-        r = toku_txn_commit_txn(txn3, FALSE, NULL, NULL);                                 CKERR(r);
+        r = toku_txn_commit_txn(txn3, FALSE, NULL, NULL);
+    CKERR(r);
         toku_txn_close_txn(txn3);
     }
 
-    r = toku_checkpoint(ct, logger, NULL, NULL, NULL, NULL, CLIENT_CHECKPOINT);                                CKERR(r);
-    r = toku_close_ft_handle_nolsn(brt, NULL);                                                                             CKERR(r);
+    r = toku_checkpoint(ct, logger, NULL, NULL, NULL, NULL, CLIENT_CHECKPOINT);
+    CKERR(r);
+    r = toku_close_ft_handle_nolsn(brt, NULL);
+    CKERR(r);
 
-    r = toku_checkpoint(ct, logger, NULL, NULL, NULL, NULL, CLIENT_CHECKPOINT);                                CKERR(r);
-    r = toku_logger_close_rollback(logger, FALSE);                                                             CKERR(r);
-    r = toku_checkpoint(ct, logger, NULL, NULL, NULL, NULL, CLIENT_CHECKPOINT);                                CKERR(r);
-    r = toku_cachetable_close(&ct);                                                                            CKERR(r);
-    r = toku_logger_close(&logger);                                                        assert(r==0);
+    clean_shutdown(&logger, &ct);
+}
 
+static TXNID
+logger_get_last_xid(TOKULOGGER logger) {
+    TXN_MANAGER mgr = toku_logger_get_txn_manager(logger);
+    return toku_txn_manager_get_last_xid(mgr);
 }
 
+static void test_xid_lsn_independent_crash_recovery(int N) {
+    TOKULOGGER logger;
+    CACHETABLE ct;
+    test_setup(&logger, &ct);
+
+    int r;
+
+    for (int i=0; i < N - 1; i++) {
+        do_txn(logger, true);
+    }
+    do_txn(logger, false);
+
+    TXNID last_xid_before = logger_get_last_xid(logger);
+
+    r = toku_logger_close_rollback(logger, false);
+    CKERR(r);
+
+    r = toku_cachetable_close(&ct);
+    CKERR(r);
+    // "Crash"
+    r = toku_logger_close(&logger);
+    CKERR(r);
+    ct = NULL;
+    logger = NULL;
+
+    // "Recover"
+    test_setup_and_recover(&logger, &ct);
+
+    TXNID last_xid_after = logger_get_last_xid(logger);
+
+    invariant(last_xid_after >= last_xid_before);
+
+    shutdown_after_recovery(&logger, &ct);
+}
+
+static void test_xid_lsn_independent_shutdown_recovery(int N) {
+    TOKULOGGER logger;
+    CACHETABLE ct;
+    test_setup(&logger, &ct);
+
+    for (int i=0; i < N - 1; i++) {
+        do_txn(logger, true);
+    }
+    do_txn(logger, false);
+
+    TXNID last_xid_before = logger_get_last_xid(logger);
+
+    clean_shutdown(&logger, &ct);
+
+    // Did a clean shutdown.
+
+    // "Recover"
+    test_setup_and_recover(&logger, &ct);
+
+    TXNID last_xid_after = logger_get_last_xid(logger);
+
+    invariant(last_xid_after >= last_xid_before);
+
+    shutdown_after_recovery(&logger, &ct);
+}
 
 int test_main (int argc, const char *argv[]) {
     default_parse_args(argc, argv);
-    for (int i=1; i<=64; i++) {
-	test_it(i);
+    for (int i=1; i<=128; i *= 2) {
+	test_xid_lsn_independent(i);
+        test_xid_lsn_independent_crash_recovery(i);
+        test_xid_lsn_independent_shutdown_recovery(i);
     }
     return 0;
 }