mirror of
https://github.com/protocolbuffers/protobuf
synced 2026-08-26 02:23:14 -04:00
e1c5f3ac47
It is common to want to match only a subset of fields in large protobufs, and doing that in rust right now requires manually clearing out the fields in the `actual` message that we don't care about. This is a pain, and a worse experience when compared to C++. Like in C++, when doing partial matching, only fields set in the expected protobuf message are compared, while extra fields set only in the actual message are ignored. When using the C++ kernel we leverage `proto2::util::MessageDifferencer`, and for the UPB kernel we use the `kUpb_CompareOption_Partial` option. PiperOrigin-RevId: 964125682
44 lines
1.7 KiB
C++
44 lines
1.7 KiB
C++
#include "rust/cpp_kernel/compare.h"
|
|
|
|
#include <string>
|
|
|
|
#include "absl/log/absl_check.h"
|
|
#include "google/protobuf/io/zero_copy_stream_impl_lite.h"
|
|
#include "google/protobuf/message.h"
|
|
#include "google/protobuf/message_lite.h"
|
|
#include "google/protobuf/util/message_differencer.h"
|
|
|
|
static std::string SerializeDeterministically(const google::protobuf::MessageLite& m) {
|
|
std::string serialized;
|
|
{
|
|
google::protobuf::io::StringOutputStream output_stream(&serialized);
|
|
google::protobuf::io::CodedOutputStream coded_stream(&output_stream);
|
|
coded_stream.SetSerializationDeterministic(true);
|
|
// TODO: Remove this suppression.
|
|
(void)m.SerializePartialToCodedStream(&coded_stream);
|
|
}
|
|
return serialized;
|
|
}
|
|
|
|
extern "C" {
|
|
|
|
bool proto2_rust_messagelite_equals(const google::protobuf::MessageLite* msg1,
|
|
const google::protobuf::MessageLite* msg2) {
|
|
return SerializeDeterministically(*msg1) == SerializeDeterministically(*msg2);
|
|
}
|
|
|
|
bool proto2_rust_messagelite_partially_equals(
|
|
const google::protobuf::MessageLite* actual, const google::protobuf::MessageLite* expected) {
|
|
const google::protobuf::Message* actual_message =
|
|
google::protobuf::DynamicCastMessage<google::protobuf::Message>(actual);
|
|
const google::protobuf::Message* expected_message =
|
|
google::protobuf::DynamicCastMessage<google::protobuf::Message>(expected);
|
|
ABSL_CHECK(actual_message != nullptr);
|
|
ABSL_CHECK(expected_message != nullptr);
|
|
|
|
google::protobuf::util::MessageDifferencer differencer;
|
|
differencer.set_scope(google::protobuf::util::MessageDifferencer::PARTIAL);
|
|
return differencer.Compare(*expected_message, *actual_message);
|
|
}
|
|
|
|
} // extern "C"
|