mdbx-test: add 'append' testcase.

Change-Id: I71620ea1a019e16b8e3d84a81dcc042961eae5b5
2025-12-17 17:32:24 +08:00 · 2019-02-03 22:37:57 +03:00
parent 73bef80347
commit 0639f54280
13 changed files with 200 additions and 5 deletions
--- a/test/append.cc
+++ b/test/append.cc
@@ -0,0 +1,132 @@
+/*
+ * Copyright 2017-2019 Leonid Yuriev <leo@yuriev.ru>
+ * and other libmdbx authors: please see AUTHORS file.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted only as authorized by the OpenLDAP
+ * Public License.
+ *
+ * A copy of this license is available in the file LICENSE in the
+ * top-level directory of the distribution or, alternatively, at
+ * <http://www.OpenLDAP.org/license.html>.
+ */
+
+#include "test.h"
+
+bool testcase_append::run() {
+  db_open();
+
+  txn_begin(false);
+  MDBX_dbi dbi = db_table_open(true);
+  int rc = mdbx_drop(txn_guard.get(), dbi, false);
+  if (unlikely(rc != MDBX_SUCCESS))
+    failure_perror("mdbx_drop(delete=false)", rc);
+
+  keyvalue_maker.setup(config.params, config.actor_id, 0 /* thread_number */);
+  /* LY: тест наполнения таблиц в append-режиме,
+   * при котором записи добавляются строго в конец (в порядке сортировки) */
+  const unsigned flags = (config.params.table_flags & MDBX_DUPSORT)
+                             ? MDBX_APPEND | MDBX_APPENDDUP
+                             : MDBX_APPEND;
+  keyvalue_maker.make_ordered();
+
+  key = keygen::alloc(config.params.keylen_max);
+  data = keygen::alloc(config.params.datalen_max);
+  keygen::buffer last_key = keygen::alloc(config.params.keylen_max);
+  keygen::buffer last_data = keygen::alloc(config.params.datalen_max);
+  last_key->value.iov_base = last_key->bytes;
+  last_key->value.iov_len = 0;
+  last_data->value.iov_base = last_data->bytes;
+  last_data->value.iov_len = 0;
+
+  simple_checksum inserted_checksum;
+  uint64_t inserted_number = 0;
+  uint64_t serial_count = 0;
+  unsigned txn_nops = 0;
+  while (should_continue()) {
+    const keygen::serial_t serial = serial_count;
+    if (!keyvalue_maker.increment(serial_count, 1)) {
+      // дошли до границы пространства ключей
+      break;
+    }
+
+    log_trace("append: append-a %" PRIu64, serial);
+    generate_pair(serial, key, data);
+    int cmp = inserted_number ? mdbx_cmp(txn_guard.get(), dbi, &key->value,
+                                         &last_key->value)
+                              : 1;
+    if (cmp == 0 && (config.params.table_flags & MDBX_DUPSORT))
+      cmp = mdbx_dcmp(txn_guard.get(), dbi, &data->value, &last_data->value);
+
+    rc = mdbx_put(txn_guard.get(), dbi, &key->value, &data->value, flags);
+    if (cmp > 0) {
+      if (unlikely(rc != MDBX_SUCCESS))
+        failure_perror("mdbx_put(appenda-a)", rc);
+      memcpy(last_key->value.iov_base, key->value.iov_base,
+             last_key->value.iov_len = key->value.iov_len);
+      memcpy(last_data->value.iov_base, data->value.iov_base,
+             last_data->value.iov_len = data->value.iov_len);
+      ++inserted_number;
+      inserted_checksum.push((uint32_t)inserted_number, key->value);
+      inserted_checksum.push(10639, data->value);
+    } else {
+      if (unlikely(rc != MDBX_EKEYMISMATCH))
+        failure_perror("mdbx_put(appenda-a) != MDBX_EKEYMISMATCH", rc);
+    }
+
+    if (++txn_nops >= config.params.batch_write) {
+      txn_restart(false, false);
+      txn_nops = 0;
+    }
+
+    report(1);
+  }
+
+  txn_restart(false, true);
+  //----------------------------------------------------------------------------
+  cursor_open(dbi);
+
+  MDBX_val check_key, check_data;
+  rc = mdbx_cursor_get(cursor_guard.get(), &check_key, &check_data, MDBX_FIRST);
+  if (unlikely(rc != MDBX_SUCCESS))
+    failure_perror("mdbx_cursor_get(MDBX_FIRST)", rc);
+
+  simple_checksum read_checksum;
+  uint64_t read_count = 0;
+  while (rc == MDBX_SUCCESS) {
+    ++read_count;
+    read_checksum.push((uint32_t)read_count, check_key);
+    read_checksum.push(10639, check_data);
+
+    rc =
+        mdbx_cursor_get(cursor_guard.get(), &check_key, &check_data, MDBX_NEXT);
+  }
+
+  if (unlikely(rc != MDBX_NOTFOUND))
+    failure_perror("mdbx_cursor_get(MDBX_NEXT) != EOF", rc);
+
+  if (unlikely(read_count != inserted_number))
+    failure("read_count(%" PRIu64 ") != inserted_number(%" PRIu64 ")",
+            read_count, inserted_number);
+
+  if (unlikely(read_checksum.value != inserted_checksum.value))
+    failure("read_checksum(0x%016" PRIu64 ") "
+            "!= inserted_checksum(0x%016" PRIu64 ")",
+            read_checksum.value, inserted_checksum.value);
+
+  cursor_close();
+  //----------------------------------------------------------------------------
+  if (txn_guard)
+    txn_end(false);
+
+  if (dbi) {
+    if (config.params.drop_table && !mode_readonly()) {
+      txn_begin(false);
+      db_table_drop(dbi);
+      txn_end(false);
+    } else
+      db_table_close(dbi);
+  }
+  return true;
+}