chore: import upstream snapshot with attribution

2026-07-13 13:29:29 +08:00
commit f52cf32e56
1718 changed files with 366293 additions and 0 deletions
@@ -0,0 +1,69 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+// Date: 2016/06/03 13:25:44
+
+#include <gtest/gtest.h>
+#include <bthread/countdown_event.h>
+#include "butil/atomicops.h"
+#include "butil/time.h"
+
+namespace {
+struct Arg {
+    bthread::CountdownEvent event;
+    butil::atomic<int> num_sig;
+};
+
+void *signaler(void *arg) {
+    Arg* a = (Arg*)arg;
+    a->num_sig.fetch_sub(1, butil::memory_order_relaxed);
+    a->event.signal();
+    return NULL;
+}
+
+TEST(CountdonwEventTest, sanity) {
+    std::vector<bthread_t> tids;
+    for (int n = 1; n < 10; ++n) {
+        Arg a;
+        a.num_sig = n;
+        a.event.reset(n);
+        for (int i = 0; i < n; ++i) {
+            bthread_t tid;
+            ASSERT_EQ(0, bthread_start_urgent(&tid, NULL, signaler, &a));
+            tids.push_back(tid);
+        }
+        a.event.wait();
+        ASSERT_EQ(0, a.num_sig.load(butil::memory_order_relaxed));
+    }
+    for (size_t i = 0; i < tids.size(); ++i) {
+        bthread_join(tids[i], NULL);
+    }
+}
+
+TEST(CountdonwEventTest, timed_wait) {
+    bthread::CountdownEvent event;
+    int rc = event.timed_wait(butil::milliseconds_from_now(100));
+    ASSERT_EQ(rc, ETIMEDOUT);
+    event.signal();
+    rc = event.timed_wait(butil::milliseconds_from_now(100));
+    ASSERT_EQ(rc, 0);
+    bthread::CountdownEvent event1;
+    event1.signal();
+    rc = event.timed_wait(butil::milliseconds_from_now(1));
+    ASSERT_EQ(rc, 0);
+}
+} // namespace