From 366111c5c7b407fe0e944ee71de1c51137147805 Mon Sep 17 00:00:00 2001 From: Paul Cristian Sarbu Date: Tue, 10 Oct 2023 17:26:40 +0200 Subject: just-serve: Extend client and API for services Target and Configuration Co-authored-by: Alberto Sartori --- src/buildtool/serve_api/remote/target_client.hpp | 65 ++++++++++++++++++++++++ 1 file changed, 65 insertions(+) create mode 100644 src/buildtool/serve_api/remote/target_client.hpp (limited to 'src/buildtool/serve_api/remote/target_client.hpp') diff --git a/src/buildtool/serve_api/remote/target_client.hpp b/src/buildtool/serve_api/remote/target_client.hpp new file mode 100644 index 00000000..ac52a468 --- /dev/null +++ b/src/buildtool/serve_api/remote/target_client.hpp @@ -0,0 +1,65 @@ +// Copyright 2023 Huawei Cloud Computing Technology Co., Ltd. +// +// Licensed 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. + +#ifndef INCLUDED_SRC_BUILDTOOL_SERVE_API_TARGET_CLIENT_HPP +#define INCLUDED_SRC_BUILDTOOL_SERVE_API_TARGET_CLIENT_HPP + +#include +#include +#include +#include +#include + +#include "justbuild/just_serve/just_serve.grpc.pb.h" +#include "src/buildtool/common/artifact.hpp" +#include "src/buildtool/common/remote/port.hpp" +#include "src/buildtool/execution_api/common/create_execution_api.hpp" +#include "src/buildtool/logging/logger.hpp" +#include "src/buildtool/storage/target_cache_entry.hpp" +#include "src/buildtool/storage/target_cache_key.hpp" + +/// Implements client side for Target service defined in: +/// src/buildtool/serve_api/serve_service/just_serve.proto +class TargetClient { + public: + TargetClient(std::string const& server, Port port) noexcept; + + /// \brief Retrieve the pair of TargetCacheEntry and ObjectInfo associated + /// to the given key. + /// \param[in] key The TargetCacheKey of an export target + /// \returns Pair of cache entry and its object info on success or nullopt. + [[nodiscard]] auto ServeTarget(const TargetCacheKey& key) + -> std::optional>; + + /// \brief Retrieve the flexible config variables of an export target. + /// \param[in] target_root_id Hash of target-level root tree. + /// \param[in] target_file Relative path of the target file. + /// \param[in] target Name of the target to interrogate. + /// \returns The list of flexible config variables, or nullopt on errors. + [[nodiscard]] auto ServeTargetVariables(std::string const& target_root_id, + std::string const& target_file, + std::string const& target) + -> std::optional>; + + private: + std::unique_ptr stub_; + Logger logger_{"RemoteTargetClient"}; + gsl::not_null const remote_api_{ + CreateExecutionApi(RemoteExecutionConfig::RemoteAddress(), + "remote-execution")}; + gsl::not_null const local_api_{ + CreateExecutionApi(std::nullopt)}; +}; + +#endif // INCLUDED_SRC_BUILDTOOL_SERVE_API_TARGET_CLIENT_HPP -- cgit v1.2.3