mirror of
https://github.com/php-win-ext/grpc.git
synced 2026-03-24 09:02:15 +01:00
1ab62ba748
- Use a print function instead of stdout/stderr - Made existing main() into a library accessible via an API. - Enabled multiple executions of the current main(). - Removed absl flags for better flag management when using the API. - Consolidated SLEUTH_TOOL and absl flags (only behavioral change). PiperOrigin-RevId: 823083768
85 lines
2.7 KiB
C++
85 lines
2.7 KiB
C++
// Copyright 2025 gRPC authors.
|
|
//
|
|
// 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 "test/cpp/sleuth/tool.h"
|
|
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "absl/container/flat_hash_map.h"
|
|
#include "absl/status/status.h"
|
|
#include "absl/status/statusor.h"
|
|
#include "absl/strings/numbers.h"
|
|
#include "absl/strings/str_cat.h"
|
|
#include "absl/strings/str_split.h"
|
|
#include "absl/strings/string_view.h"
|
|
|
|
namespace grpc_sleuth {
|
|
|
|
absl::StatusOr<std::unique_ptr<ToolArgs>> ToolArgs::TryCreate(
|
|
const std::vector<std::string>& args) {
|
|
absl::flat_hash_map<std::string, std::string> map;
|
|
for (const auto& arg : args) {
|
|
std::vector<std::string> parts =
|
|
absl::StrSplit(arg, absl::MaxSplits('=', 1));
|
|
if (parts.size() != 2) {
|
|
return absl::InvalidArgumentError(absl::StrCat(
|
|
"Invalid argument format: ", arg, "; expected key=value"));
|
|
}
|
|
const std::string& key = parts[0];
|
|
const std::string& value = parts[1];
|
|
if (key.empty()) {
|
|
return absl::InvalidArgumentError(
|
|
absl::StrCat("Key cannot be empty in argument: ", arg));
|
|
}
|
|
if (!map.insert({key, value}).second) {
|
|
return absl::InvalidArgumentError(absl::StrCat("Duplicate key: ", key));
|
|
}
|
|
}
|
|
return std::unique_ptr<ToolArgs>(new ToolArgs(std::move(map)));
|
|
}
|
|
|
|
template <>
|
|
absl::StatusOr<std::string> ToolArgs::ConvertValue<std::string>(
|
|
absl::string_view /*key*/, absl::string_view value) {
|
|
return std::string(value);
|
|
}
|
|
|
|
template <>
|
|
absl::StatusOr<double> ToolArgs::ConvertValue<double>(absl::string_view key,
|
|
absl::string_view value) {
|
|
double result;
|
|
if (!absl::SimpleAtod(value, &result)) {
|
|
return absl::InvalidArgumentError(absl::StrCat(
|
|
"Invalid value for ", key, ": '", value, "' is not a double"));
|
|
}
|
|
return result;
|
|
}
|
|
|
|
template <>
|
|
absl::StatusOr<int64_t> ToolArgs::ConvertValue<int64_t>(
|
|
absl::string_view key, absl::string_view value) {
|
|
int64_t result;
|
|
if (!absl::SimpleAtoi(value, &result)) {
|
|
return absl::InvalidArgumentError(absl::StrCat(
|
|
"Invalid value for ", key, ": '", value, "' is not an integer"));
|
|
}
|
|
return result;
|
|
}
|
|
|
|
} // namespace grpc_sleuth
|