Protocol Buffers - Google's data interchange format (grpc依赖)
https://developers.google.com/protocol-buffers/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.9 KiB
93 lines
2.9 KiB
// Protocol Buffers - Google's data interchange format |
|
// Copyright 2023 Google LLC. All rights reserved. |
|
// |
|
// Use of this source code is governed by a BSD-style |
|
// license that can be found in the LICENSE file or at |
|
// https://developers.google.com/open-source/licenses/bsd |
|
|
|
#include <string> |
|
|
|
#include "absl/log/absl_log.h" |
|
#include "absl/strings/string_view.h" |
|
#include "absl/strings/substitute.h" |
|
#include "upb/base/status.hpp" |
|
#include "upb/base/string_view.h" |
|
#include "upb/reflection/def.hpp" |
|
#include "upb_generator/common.h" |
|
#include "upb_generator/file_layout.h" |
|
#include "upb_generator/plugin.h" |
|
#include "upb_generator/protoc-gen-upb_minitable.h" |
|
|
|
// Must be last. |
|
#include "upb/port/def.inc" |
|
|
|
namespace upb { |
|
namespace generator { |
|
|
|
std::string SourceFilename(upb::FileDefPtr file) { |
|
return StripExtension(file.name()) + ".upb_minitable.c"; |
|
} |
|
|
|
absl::string_view ToStringView(upb_StringView str) { |
|
return absl::string_view(str.data, str.size); |
|
} |
|
|
|
void GenerateFile(const DefPoolPair& pools, upb::FileDefPtr file, |
|
const MiniTableOptions& options, Plugin* plugin) { |
|
Output h_output; |
|
WriteMiniTableHeader(pools, file, options, h_output); |
|
plugin->AddOutputFile(MiniTableHeaderFilename(file, false), |
|
h_output.output()); |
|
|
|
Output c_output; |
|
WriteMiniTableSource(pools, file, options, c_output); |
|
plugin->AddOutputFile(SourceFilename(file), c_output.output()); |
|
|
|
if (options.one_output_per_message) { |
|
WriteMiniTableMultipleSources(pools, file, options, plugin); |
|
} |
|
} |
|
|
|
bool ParseOptions(MiniTableOptions* options, Plugin* plugin) { |
|
for (const auto& pair : ParseGeneratorParameter(plugin->parameter())) { |
|
if (pair.first == "bootstrap_stage") { |
|
options->bootstrap = true; |
|
} else if (pair.first == "experimental_strip_nonfunctional_codegen") { |
|
options->strip_nonfunctional_codegen = true; |
|
} else if (pair.first == "one_output_per_message") { |
|
options->one_output_per_message = true; |
|
} else { |
|
plugin->SetError(absl::Substitute("Unknown parameter: $0", pair.first)); |
|
return false; |
|
} |
|
} |
|
|
|
return true; |
|
} |
|
|
|
int PluginMain(int argc, char** argv) { |
|
DefPoolPair pools; |
|
MiniTableOptions options; |
|
Plugin plugin; |
|
if (!ParseOptions(&options, &plugin)) return 0; |
|
plugin.GenerateFilesRaw( |
|
[&](const UPB_DESC(FileDescriptorProto) * file_proto, bool generate) { |
|
upb::Status status; |
|
upb::FileDefPtr file = pools.AddFile(file_proto, &status); |
|
if (!file) { |
|
absl::string_view name = |
|
ToStringView(UPB_DESC(FileDescriptorProto_name)(file_proto)); |
|
ABSL_LOG(FATAL) << "Couldn't add file " << name |
|
<< " to DefPool: " << status.error_message(); |
|
} |
|
if (generate) GenerateFile(pools, file, options, &plugin); |
|
}); |
|
return 0; |
|
} |
|
|
|
} // namespace generator |
|
} // namespace upb |
|
|
|
int main(int argc, char** argv) { |
|
return upb::generator::PluginMain(argc, argv); |
|
}
|
|
|