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,97 @@
+// 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.
+
+
+#include <gflags/gflags.h>
+#include "butil/atomicops.h"
+#include "butil/logging.h"
+#include "butil/time.h"
+#include "brpc/server.h"
+#include "bvar/variable.h"
+#include "test.pb.h"
+
+#ifdef BRPC_WITH_RDMA
+
+DEFINE_int32(port, 8002, "TCP Port of this server");
+DEFINE_bool(use_rdma, true, "Use RDMA or not");
+
+butil::atomic<uint64_t> g_last_time(0);
+
+namespace test {
+class PerfTestServiceImpl : public PerfTestService {
+public:
+    PerfTestServiceImpl() {}
+    ~PerfTestServiceImpl() {}
+
+    void Test(google::protobuf::RpcController* cntl_base,
+              const PerfTestRequest* request,
+              PerfTestResponse* response,
+              google::protobuf::Closure* done) {
+        brpc::ClosureGuard done_guard(done);
+        uint64_t last = g_last_time.load(butil::memory_order_relaxed);
+        uint64_t now = butil::monotonic_time_us();
+        if (now > last && now - last > 100000) {
+            if (g_last_time.exchange(now, butil::memory_order_relaxed) == last) {
+                response->set_cpu_usage(bvar::Variable::describe_exposed("process_cpu_usage"));
+            } else {
+                response->set_cpu_usage("");
+            }
+        } else {
+            response->set_cpu_usage("");
+        }
+        if (request->echo_attachment()) {
+            brpc::Controller* cntl =
+                static_cast<brpc::Controller*>(cntl_base);
+            cntl->response_attachment().append(cntl->request_attachment());
+        }
+    }
+};
+}
+
+int main(int argc, char* argv[]) {
+    GFLAGS_NAMESPACE::ParseCommandLineFlags(&argc, &argv, true);
+
+    brpc::Server server;
+    test::PerfTestServiceImpl perf_test_service_impl;
+
+    if (server.AddService(&perf_test_service_impl, 
+                          brpc::SERVER_DOESNT_OWN_SERVICE) != 0) {
+        LOG(ERROR) << "Fail to add service";
+        return -1;
+    }
+    g_last_time.store(0, butil::memory_order_relaxed);
+
+    brpc::ServerOptions options;
+    options.socket_mode = FLAGS_use_rdma? brpc::SOCKET_MODE_RDMA : brpc::SOCKET_MODE_TCP;
+    if (server.Start(FLAGS_port, &options) != 0) {
+        LOG(ERROR) << "Fail to start EchoServer";
+        return -1;
+    }
+
+    server.RunUntilAskedToQuit();
+    return 0;
+}
+
+#else
+
+
+int main(int argc, char* argv[]) {
+    LOG(ERROR) << " brpc is not compiled with rdma. To enable it, please refer to https://github.com/apache/brpc/blob/master/docs/en/rdma.md";
+    return 0;
+}
+
+#endif