Add all methods of output service to the implementation.

PiperOrigin-RevId: 678656758 Change-Id: I36e5efd378f04c7d6f03114c79e1000da0ebc13f
bazelbuild · Sep 25, 2024 · 56af167 · 56af167
1 parent 163160d
commit 56af167
Show file tree

Hide file tree

Showing 3 changed files with 102 additions and 23 deletions.
diff --git a/src/tools/remote/src/main/cpp/testonly_output_service/bazel_output_service_impl.cc b/src/tools/remote/src/main/cpp/testonly_output_service/bazel_output_service_impl.cc
@@ -13,3 +13,70 @@
 // limitations under the License.
 
 #include "src/tools/remote/src/main/cpp/testonly_output_service/bazel_output_service_impl.h"
+
+#include <iostream>
+#include <memory>
+#include <string>
+
+#include "grpcpp/security/server_credentials.h"
+#include "grpcpp/server_builder.h"
+#include "grpcpp/server_context.h"
+#include "grpcpp/support/status.h"
+
+grpc::Status BazelOutputServiceImpl::Clean(
+    grpc::ServerContext* context,
+    const bazel_output_service::CleanRequest* request,
+    bazel_output_service::CleanResponse* response) {
+  return grpc::Status(grpc::StatusCode::UNIMPLEMENTED, "");
+}
+
+grpc::Status BazelOutputServiceImpl::StartBuild(
+    grpc::ServerContext* context,
+    const bazel_output_service::StartBuildRequest* request,
+    bazel_output_service::StartBuildResponse* response) {
+  return grpc::Status(grpc::StatusCode::UNIMPLEMENTED, "");
+}
+
+grpc::Status BazelOutputServiceImpl::StageArtifacts(
+    grpc::ServerContext* context,
+    const bazel_output_service::StageArtifactsRequest* request,
+    bazel_output_service::StageArtifactsResponse* response) {
+  return grpc::Status(grpc::StatusCode::UNIMPLEMENTED, "");
+}
+
+grpc::Status BazelOutputServiceImpl::FinalizeArtifacts(
+    grpc::ServerContext* context,
+    const bazel_output_service::FinalizeArtifactsRequest* request,
+    bazel_output_service::FinalizeArtifactsResponse* response) {
+  return grpc::Status(grpc::StatusCode::UNIMPLEMENTED, "");
+}
+
+grpc::Status BazelOutputServiceImpl::FinalizeBuild(
+    grpc::ServerContext* context,
+    const bazel_output_service::FinalizeBuildRequest* request,
+    bazel_output_service::FinalizeBuildResponse* response) {
+  return grpc::Status(grpc::StatusCode::UNIMPLEMENTED, "");
+}
+
+grpc::Status BazelOutputServiceImpl::BatchStat(
+    grpc::ServerContext* context,
+    const bazel_output_service::BatchStatRequest* request,
+    bazel_output_service::BatchStatResponse* response) {
+  return grpc::Status(grpc::StatusCode::UNIMPLEMENTED, "");
+}
+
+int RunServer(int argc, char** argv) {
+  BazelOutputServiceImpl service;
+
+  std::string server_address = "0.0.0.0:8080";
+
+  grpc::ServerBuilder builder;
+  builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
+  builder.RegisterService(&service);
+  std::unique_ptr<grpc::Server> server(builder.BuildAndStart());
+  std::cerr << "Server listening on " << server_address << std::endl;
+
+  server->Wait();
+
+  return 0;
+}
diff --git a/src/tools/remote/src/main/cpp/testonly_output_service/bazel_output_service_impl.h b/src/tools/remote/src/main/cpp/testonly_output_service/bazel_output_service_impl.h
@@ -16,8 +16,41 @@
 #define BAZEL_SRC_TOOLS_REMOTE_SRC_MAIN_CPP_OUTPUT_SERVICE_BAZEL_OUTPUT_SERVICE_IMPL_H_
 
 #include "src/main/protobuf/bazel_output_service.grpc.pb.h"
+#include "grpcpp/server_context.h"
+#include "grpcpp/support/status.h"
 
 class BazelOutputServiceImpl
-    : public bazel_output_service::BazelOutputService::Service {};
+    : public bazel_output_service::BazelOutputService::Service {
+  grpc::Status Clean(grpc::ServerContext* context,
+                     const bazel_output_service::CleanRequest* request,
+                     bazel_output_service::CleanResponse* response) override;
+
+  grpc::Status StartBuild(
+      grpc::ServerContext* context,
+      const bazel_output_service::StartBuildRequest* request,
+      bazel_output_service::StartBuildResponse* response) override;
+
+  grpc::Status StageArtifacts(
+      grpc::ServerContext* context,
+      const bazel_output_service::StageArtifactsRequest* request,
+      bazel_output_service::StageArtifactsResponse* response) override;
+
+  grpc::Status FinalizeArtifacts(
+      grpc::ServerContext* context,
+      const bazel_output_service::FinalizeArtifactsRequest* request,
+      bazel_output_service::FinalizeArtifactsResponse* response) override;
+
+  grpc::Status FinalizeBuild(
+      grpc::ServerContext* context,
+      const bazel_output_service::FinalizeBuildRequest* request,
+      bazel_output_service::FinalizeBuildResponse* response) override;
+
+  grpc::Status BatchStat(
+      grpc::ServerContext* context,
+      const bazel_output_service::BatchStatRequest* request,
+      bazel_output_service::BatchStatResponse* response) override;
+};
+
+int RunServer(int argc, char** argv);
 
 #endif  // BAZEL_SRC_TOOLS_REMOTE_SRC_MAIN_CPP_OUTPUT_SERVICE_BAZEL_OUTPUT_SERVICE_IMPL_H_
diff --git a/src/tools/remote/src/main/cpp/testonly_output_service/main.cc b/src/tools/remote/src/main/cpp/testonly_output_service/main.cc
@@ -12,29 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-#include <iostream>
-#include <memory>
-#include <string>
-
 #include "src/tools/remote/src/main/cpp/testonly_output_service/bazel_output_service_impl.h"
-#include "grpcpp/security/server_credentials.h"
-#include "grpcpp/server_builder.h"
-
-static void RunServer() {
-  BazelOutputServiceImpl service;
-
-  std::string server_address = "0.0.0.0:8080";
-
-  grpc::ServerBuilder builder;
-  builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
-  builder.RegisterService(&service);
-  std::unique_ptr<grpc::Server> server(builder.BuildAndStart());
-  std::cerr << "Server listening on " << server_address << std::endl;
-
-  server->Wait();
-}
 
 int main(int argc, char** argv) {
-  RunServer();
-  return 0;
+  return RunServer(argc, argv);
 }