mirror of
https://github.com/protocolbuffers/protobuf
synced 2026-08-26 02:23:14 -04:00
86 lines
3 KiB
C++
86 lines
3 KiB
C++
// 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 "hpb_generator/gen_extensions.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "absl/strings/str_cat.h"
|
|
#include "hpb_generator/context.h"
|
|
#include "hpb_generator/gen_utils.h"
|
|
#include "hpb_generator/keywords.h"
|
|
#include "hpb_generator/names.h"
|
|
#include "google/protobuf/descriptor.h"
|
|
#include "upb_generator/c/names.h"
|
|
|
|
namespace google {
|
|
namespace protobuf {
|
|
namespace hpb_generator {
|
|
|
|
using Sub = google::protobuf::io::Printer::Sub;
|
|
|
|
std::string ExtensionIdentifierBase(const google::protobuf::FieldDescriptor* ext) {
|
|
assert(ext->is_extension());
|
|
std::string ext_scope;
|
|
if (ext->extension_scope()) {
|
|
return upb::generator::CApiMessageType(ext->extension_scope()->full_name());
|
|
} else {
|
|
return ToCIdent(ext->file()->package());
|
|
}
|
|
}
|
|
|
|
std::string ContainingTypeName(const google::protobuf::FieldDescriptor* ext) {
|
|
return ext->containing_type()->file() != ext->file()
|
|
? QualifiedClassName(ext->containing_type())
|
|
: ClassName(ext->containing_type());
|
|
}
|
|
|
|
void WriteExtensionIdentifierHeader(const google::protobuf::FieldDescriptor* ext,
|
|
Context& ctx) {
|
|
std::string mini_table_name =
|
|
absl::StrCat(ExtensionIdentifierBase(ext), "_", ext->name(), "_ext");
|
|
std::string linkage = ext->extension_scope() ? "static" : "";
|
|
std::string ext_type = CppTypeParameterName(ext);
|
|
if (ext->is_repeated()) {
|
|
ext_type = absl::StrCat("::hpb::RepeatedField<", ext_type, ">");
|
|
}
|
|
ctx.Emit(
|
|
{{"containing_type_name", ContainingTypeName(ext)},
|
|
{"extendee_type", ContainingTypeName(ext)},
|
|
{"extension_type", ext_type},
|
|
{"default_value", DefaultValue(ext)},
|
|
{"linkage", linkage},
|
|
{"mini_table_name",
|
|
absl::StrCat(ExtensionIdentifierBase(ext), "_", ext->name(), "_ext")},
|
|
Sub("extension_name", ResolveKeywordConflict(ext->name()))
|
|
.AnnotatedAs(ext),
|
|
{"extension_number", ext->number()}},
|
|
R"cc(
|
|
inline $linkage$ constexpr ::hpb::internal::ExtensionIdentifier<
|
|
$extendee_type$, $extension_type$>
|
|
$extension_name$ =
|
|
::hpb::internal::PrivateAccess::InvokeConstructor<
|
|
::hpb::internal::ExtensionIdentifier<$containing_type_name$,
|
|
$extension_type$>>(
|
|
&$mini_table_name$, $default_value$, $extension_number$);
|
|
)cc");
|
|
}
|
|
|
|
void WriteExtensionIdentifiersHeader(
|
|
const std::vector<const google::protobuf::FieldDescriptor*>& extensions,
|
|
Context& ctx) {
|
|
for (const auto* ext : extensions) {
|
|
if (!ext->extension_scope()) {
|
|
WriteExtensionIdentifierHeader(ext, ctx);
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace hpb_generator
|
|
} // namespace protobuf
|
|
} // namespace google
|