1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
// 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.
#include "src/buildtool/execution_api/execution_service/ac_server.hpp"
#include <optional>
#include <string>
#include <utility>
#include "fmt/core.h"
#include "src/buildtool/common/artifact_digest.hpp"
#include "src/buildtool/common/artifact_digest_factory.hpp"
#include "src/buildtool/crypto/hash_function.hpp"
#include "src/buildtool/logging/log_level.hpp"
#include "src/buildtool/storage/garbage_collector.hpp"
#include "src/utils/cpp/expected.hpp"
auto ActionCacheServiceImpl::GetActionResult(
::grpc::ServerContext* /*context*/,
const ::bazel_re::GetActionResultRequest* request,
::bazel_re::ActionResult* response) -> ::grpc::Status {
auto action_digest = ArtifactDigestFactory::FromBazel(
storage_config_.hash_function.GetType(), request->action_digest());
if (not action_digest) {
logger_.Emit(LogLevel::Debug, "{}", action_digest.error());
return ::grpc::Status{::grpc::StatusCode::INVALID_ARGUMENT,
std::move(action_digest).error()};
}
logger_.Emit(LogLevel::Trace, "GetActionResult: {}", action_digest->hash());
auto const lock = GarbageCollector::SharedLock(storage_config_);
if (not lock) {
static constexpr auto kStr = "Could not acquire SharedLock";
logger_.Emit(LogLevel::Error, kStr);
return grpc::Status{grpc::StatusCode::INTERNAL, kStr};
}
auto action_result = storage_.ActionCache().CachedResult(*action_digest);
if (not action_result) {
return grpc::Status{
grpc::StatusCode::NOT_FOUND,
fmt::format("{} missing from AC", action_digest->hash())};
}
*response = *std::move(action_result);
return ::grpc::Status::OK;
}
auto ActionCacheServiceImpl::UpdateActionResult(
::grpc::ServerContext* /*context*/,
const ::bazel_re::UpdateActionResultRequest*
/*request*/,
::bazel_re::ActionResult* /*response*/) -> ::grpc::Status {
static auto constexpr kStr = "UpdateActionResult not implemented";
logger_.Emit(LogLevel::Error, kStr);
return ::grpc::Status{grpc::StatusCode::UNIMPLEMENTED, kStr};
}
|
