reuse protobuf message

motis-project · Nov 14, 2024 · 4686575 · 4686575
1 parent e3e83f7
commit 4686575
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/include/nigiri/rt/gtfsrt_update.h b/include/nigiri/rt/gtfsrt_update.h
@@ -41,6 +41,13 @@ statistics gtfsrt_update_msg(timetable const&,
                              std::string_view tag,
                              transit_realtime::FeedMessage const&);
 
+statistics gtfsrt_update_buf(timetable const& tt,
+                             rt_timetable& rtt,
+                             source_idx_t const src,
+                             std::string_view tag,
+                             std::string_view protobuf,
+                             transit_realtime::FeedMessage& msg);
+
 statistics gtfsrt_update_buf(timetable const&,
                              rt_timetable&,
                              source_idx_t const,

diff --git a/src/rt/gtfsrt_update.cc b/src/rt/gtfsrt_update.cc
@@ -367,8 +367,9 @@ statistics gtfsrt_update_buf(timetable const& tt,
                              rt_timetable& rtt,
                              source_idx_t const src,
                              std::string_view tag,
-                             std::string_view protobuf) {
-  gtfsrt::FeedMessage msg;
+                             std::string_view protobuf,
+                             gtfsrt::FeedMessage& msg) {
+  msg.Clear();
 
   auto const success =
       msg.ParseFromArray(reinterpret_cast<void const*>(protobuf.data()),
@@ -383,4 +384,13 @@ statistics gtfsrt_update_buf(timetable const& tt,
   return gtfsrt_update_msg(tt, rtt, src, tag, msg);
 }
 
+statistics gtfsrt_update_buf(timetable const& tt,
+                             rt_timetable& rtt,
+                             source_idx_t const src,
+                             std::string_view tag,
+                             std::string_view protobuf) {
+  auto msg = gtfsrt::FeedMessage{};
+  return gtfsrt_update_buf(tt, rtt, src, tag, protobuf, msg);
+}
+
 }  // namespace nigiri::rt