It ended up being odd to construct the AccessorGenerators ahead of time given some specific Context<>: since each of the three methods on AccessorGenerators demands a different Context than the initial one so they shouldn't hold that Context anyway. It ended up just being a spooky parallel array to the Fields with an unenforced invariant that all methods would be called with a new Context<> but the same underlying FieldDescriptor later. PiperOrigin-RevId: 558770643pull/13617/head
Protobuf Team Bot
1 year ago
committed by
Copybara-Service
parent
da38af7191
commit
34585e9d2f
11 changed files with 370 additions and 295 deletions
@ -0,0 +1,123 @@ |
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC. nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#ifndef GOOGLE_PROTOBUF_COMPILER_RUST_ACCESSORS_ACCESSOR_GENERATOR_H__ |
||||
#define GOOGLE_PROTOBUF_COMPILER_RUST_ACCESSORS_ACCESSOR_GENERATOR_H__ |
||||
|
||||
#include <memory> |
||||
|
||||
#include "absl/log/absl_check.h" |
||||
#include "google/protobuf/compiler/rust/context.h" |
||||
#include "google/protobuf/descriptor.h" |
||||
|
||||
namespace google { |
||||
namespace protobuf { |
||||
namespace compiler { |
||||
namespace rust { |
||||
|
||||
class AccessorGenerator { |
||||
public: |
||||
AccessorGenerator() = default; |
||||
virtual ~AccessorGenerator() = default; |
||||
|
||||
AccessorGenerator(const AccessorGenerator &) = delete; |
||||
AccessorGenerator(AccessorGenerator &&) = delete; |
||||
AccessorGenerator &operator=(const AccessorGenerator &) = delete; |
||||
AccessorGenerator &operator=(AccessorGenerator &&) = delete; |
||||
|
||||
// Constructs a generator for the given field.
|
||||
//
|
||||
// Returns `nullptr` if there is no known generator for this field.
|
||||
static std::unique_ptr<AccessorGenerator> For(Context<FieldDescriptor> field); |
||||
|
||||
void GenerateMsgImpl(Context<FieldDescriptor> field) const { |
||||
InMsgImpl(field); |
||||
} |
||||
void GenerateExternC(Context<FieldDescriptor> field) const { |
||||
InExternC(field); |
||||
} |
||||
void GenerateThunkCc(Context<FieldDescriptor> field) const { |
||||
ABSL_CHECK(field.is_cpp()); |
||||
InThunkCc(field); |
||||
} |
||||
|
||||
private: |
||||
// Note: the virtual functions are duplicated as non-virtual public functions,
|
||||
// so that we can customize prologue and epilogue behavior for these
|
||||
// functions. For example, consider calling `field.printer.WithVars()` as a
|
||||
// prologue to inject variables automatically.
|
||||
|
||||
// Called inside the main inherent `impl Msg {}` block.
|
||||
virtual void InMsgImpl(Context<FieldDescriptor> field) const {} |
||||
|
||||
// Called inside of a message's `extern "C" {}` block.
|
||||
virtual void InExternC(Context<FieldDescriptor> field) const {} |
||||
|
||||
// Called inside of an `extern "C" {}` block in the `.thunk.cc` file, if such
|
||||
// a file is being generated.
|
||||
virtual void InThunkCc(Context<FieldDescriptor> field) const {} |
||||
}; |
||||
|
||||
class SingularScalar final : public AccessorGenerator { |
||||
public: |
||||
~SingularScalar() override = default; |
||||
void InMsgImpl(Context<FieldDescriptor> field) const override; |
||||
void InExternC(Context<FieldDescriptor> field) const override; |
||||
void InThunkCc(Context<FieldDescriptor> field) const override; |
||||
}; |
||||
|
||||
class SingularBytes final : public AccessorGenerator { |
||||
public: |
||||
~SingularBytes() override = default; |
||||
void InMsgImpl(Context<FieldDescriptor> field) const override; |
||||
void InExternC(Context<FieldDescriptor> field) const override; |
||||
void InThunkCc(Context<FieldDescriptor> field) const override; |
||||
}; |
||||
|
||||
class SingularMessage final : public AccessorGenerator { |
||||
public: |
||||
~SingularMessage() override = default; |
||||
void InMsgImpl(Context<FieldDescriptor> field) const override; |
||||
void InExternC(Context<FieldDescriptor> field) const override; |
||||
void InThunkCc(Context<FieldDescriptor> field) const override; |
||||
}; |
||||
|
||||
class UnsupportedField final : public AccessorGenerator { |
||||
public: |
||||
~UnsupportedField() override = default; |
||||
void InMsgImpl(Context<FieldDescriptor> field) const override; |
||||
}; |
||||
|
||||
} // namespace rust
|
||||
} // namespace compiler
|
||||
} // namespace protobuf
|
||||
} // namespace google
|
||||
|
||||
#endif // GOOGLE_PROTOBUF_COMPILER_RUST_ACCESSORS_ACCESSOR_GENERATOR_H__
|
||||
@ -0,0 +1,51 @@ |
||||
// Protocol Buffers - Google's data interchange format
|
||||
// Copyright 2023 Google LLC. All rights reserved.
|
||||
// https://developers.google.com/protocol-buffers/
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without
|
||||
// modification, are permitted provided that the following conditions are
|
||||
// met:
|
||||
//
|
||||
// * Redistributions of source code must retain the above copyright
|
||||
// notice, this list of conditions and the following disclaimer.
|
||||
// * Redistributions in binary form must reproduce the above
|
||||
// copyright notice, this list of conditions and the following disclaimer
|
||||
// in the documentation and/or other materials provided with the
|
||||
// distribution.
|
||||
// * Neither the name of Google LLC. nor the names of its
|
||||
// contributors may be used to endorse or promote products derived from
|
||||
// this software without specific prior written permission.
|
||||
//
|
||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
#include "absl/strings/string_view.h" |
||||
#include "google/protobuf/compiler/rust/accessors/accessor_generator.h" |
||||
#include "google/protobuf/compiler/rust/context.h" |
||||
#include "google/protobuf/descriptor.h" |
||||
|
||||
namespace google { |
||||
namespace protobuf { |
||||
namespace compiler { |
||||
namespace rust { |
||||
|
||||
void UnsupportedField::InMsgImpl(Context<FieldDescriptor> field) const { |
||||
field.Emit(R"rs( |
||||
// Unsupported! :(
|
||||
)rs"); |
||||
field.printer().PrintRaw("\n"); |
||||
} |
||||
|
||||
} // namespace rust
|
||||
} // namespace compiler
|
||||
} // namespace protobuf
|
||||
} // namespace google
|
||||
Loading…
Reference in new issue