commit a17c248775442b10bcfbfc9989eb9ee87365e07d
author Protobuf Team Bot <protobuf-github-bot@google.com> Fri Dec 08 13:28:01 2023 -0800
committer Copybara-Service <copybara-worker@google.com> Fri Dec 08 13:32:28 2023 -0800
tree f4b783bca49ea9d5a7ce2b3fa9d42f34548aa3d3
parent 4c32845eb504ca6723f073cf94fa2a84455a0e16

Make LazyField::ByteSizeLong consistent with eager message.

PiperOrigin-RevId: 589227028

src/google/protobuf/extension_set.cc

diff --git a/src/google/protobuf/extension_set.cc b/src/google/protobuf/extension_set.cc
index 0a67b05..5328fa0 100644
--- a/src/google/protobuf/extension_set.cc
+++ b/src/google/protobuf/extension_set.cc
@@ -1452,12 +1452,9 @@
       HANDLE_TYPE(GROUP, Group, *message_value);
 #undef HANDLE_TYPE
       case WireFormatLite::TYPE_MESSAGE: {
-        if (is_lazy) {
-          // LazyField::ByteSizeLong includes sizeof(length).
-          result += lazymessage_value->ByteSizeLong();
-        } else {
-          result += WireFormatLite::MessageSize(*message_value);
-        }
+        result += WireFormatLite::LengthDelimitedSize(
+            is_lazy ? lazymessage_value->ByteSizeLong()
+                    : message_value->ByteSizeLong());
         break;
       }
 
@@ -1919,11 +1916,9 @@
   our_size += io::CodedOutputStream::VarintSize32(number);
 
   // message
-  if (is_lazy) {
-    our_size += lazymessage_value->ByteSizeLong();
-  } else {
-    our_size += WireFormatLite::MessageSize(*message_value);
-  }
+  our_size += WireFormatLite::LengthDelimitedSize(
+      is_lazy ? lazymessage_value->ByteSizeLong()
+              : message_value->ByteSizeLong());
 
   return our_size;
 }