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.
58 lines
1.9 KiB
58 lines
1.9 KiB
// Protocol Buffers - Google's data interchange format |
|
// Copyright 2025 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 "upb_generator/plugin.h" |
|
|
|
#include <string> |
|
|
|
#include "google/protobuf/descriptor.pb.h" |
|
#include "absl/container/flat_hash_set.h" |
|
#include "absl/log/absl_log.h" |
|
#include "absl/strings/string_view.h" |
|
#include "google/protobuf/descriptor.h" |
|
#include "upb/base/status.hpp" |
|
#include "upb/base/string_view.h" |
|
#include "upb/mem/arena.hpp" |
|
#include "upb/reflection/def.hpp" |
|
#include "upb_generator/file_layout.h" |
|
|
|
// Must be last. |
|
#include "upb/port/def.inc" |
|
|
|
namespace upb { |
|
namespace generator { |
|
|
|
absl::string_view ToStringView(upb_StringView str) { |
|
return absl::string_view(str.data, str.size); |
|
} |
|
|
|
void PopulateDefPool(const google::protobuf::FileDescriptor* file, upb::Arena* arena, |
|
DefPoolPair* pools, |
|
absl::flat_hash_set<std::string>* files_seen) { |
|
bool new_file = files_seen->insert(std::string(file->name())).second; |
|
if (new_file) { |
|
for (int i = 0; i < file->dependency_count(); ++i) { |
|
PopulateDefPool(file->dependency(i), arena, pools, files_seen); |
|
} |
|
google::protobuf::FileDescriptorProto raw_proto; |
|
file->CopyTo(&raw_proto); |
|
std::string serialized = raw_proto.SerializeAsString(); |
|
auto* file_proto = UPB_DESC(FileDescriptorProto_parse)( |
|
serialized.data(), serialized.size(), arena->ptr()); |
|
upb::Status status; |
|
upb::FileDefPtr upb_file = pools->AddFile(file_proto, &status); |
|
if (!upb_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(); |
|
} |
|
} |
|
} |
|
|
|
} // namespace generator |
|
} // namespace upb
|
|
|