1 files changed, 103 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/utility/tiered.c b/src/third_party/wiredtiger/test/utility/tiered.c
new file mode 100644
index 00000000000..47bab6adf57
--- /dev/null
+++ b/src/third_party/wiredtiger/test/utility/tiered.c
@@ -0,0 +1,103 @@
+/*-
+ * Public Domain 2014-present MongoDB, Inc.
+ * Public Domain 2008-2014 WiredTiger, Inc.
+ *
+ * This is free and unencumbered software released into the public domain.
+ *
+ * Anyone is free to copy, modify, publish, use, compile, sell, or
+ * distribute this software, either in source code form or as a compiled
+ * binary, for any purpose, commercial or non-commercial, and by any
+ * means.
+ *
+ * In jurisdictions that recognize copyright laws, the author or authors
+ * of this software dedicate any and all copyright interest in the
+ * software to the public domain. We make this dedication for the benefit
+ * of the public at large and to the detriment of our heirs and
+ * successors. We intend this dedication to be an overt act of
+ * relinquishment in perpetuity of all present and future rights to this
+ * software under copyright law.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ */
+#include "test_util.h"
+
+/*
+ * testutil_tiered_begin --
+ *     Begin processing for a test program that supports tiered storage.
+ */
+void
+testutil_tiered_begin(TEST_OPTS *opts)
+{
+    WT_SESSION *session;
+
+    testutil_assert(!opts->tiered_begun);
+    testutil_assert(opts->conn != NULL);
+
+    if (opts->tiered_storage && opts->tiered_flush_interval_us != 0) {
+        /*
+         * Initialize the time of the next flush_tier. We need a temporary session to do that.
+         */
+        testutil_check(opts->conn->open_session(opts->conn, NULL, NULL, &session));
+        testutil_tiered_flush_complete(opts, session, NULL);
+        testutil_check(session->close(session, NULL));
+    }
+
+    opts->tiered_begun = true;
+}
+
+/*
+ * testutil_tiered_sleep --
+ *     Sleep for a number of seconds, or until it is time to flush_tier, or the process wants to
+ *     exit.
+ */
+void
+testutil_tiered_sleep(TEST_OPTS *opts, WT_SESSION *session, uint32_t seconds, bool *do_flush_tier)
+{
+    uint64_t now, wake_time;
+    bool do_flush;
+
+    now = testutil_time_us(session);
+    wake_time = now + 1000000 * seconds;
+    do_flush = false;
+    if (do_flush_tier != NULL && opts->tiered_flush_next_us != 0 &&
+      opts->tiered_flush_next_us < wake_time) {
+        wake_time = opts->tiered_flush_next_us;
+        do_flush = true;
+    }
+    *do_flush_tier = false;
+
+    while (now < wake_time && opts->running) {
+        /* Sleep a maximum of one second, so we can make sure we should still be running. */
+        if (now + 1000000 < wake_time)
+            __wt_sleep(1, 0);
+        else
+            __wt_sleep(0, wake_time - now);
+        now = testutil_time_us(session);
+    }
+    if (opts->running && do_flush) {
+        /* Don't flush again until we know this flush is complete. */
+        opts->tiered_flush_next_us = 0;
+        *do_flush_tier = true;
+    }
+}
+
+/*
+ * testutil_tiered_flush_complete --
+ *     Notification that a flush_tier has completed, with the given argument.
+ */
+void
+testutil_tiered_flush_complete(TEST_OPTS *opts, WT_SESSION *session, void *arg)
+{
+    uint64_t now;
+
+    (void)arg;
+
+    now = testutil_time_us(session);
+    opts->tiered_flush_next_us = now + opts->tiered_flush_interval_us;
+}