Fix msg not found error. (#13)

iamazeem · Sep 16, 2023 · d943273 · d943273
1 parent 3dfd208
commit d943273
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/fluent/plugin/in_protobuf_http.rb b/lib/fluent/plugin/in_protobuf_http.rb
@@ -116,9 +116,10 @@ def get_msg_types(compiled_proto)
         log.debug("Extracting message types [#{compiled_proto}]...")
         msg_types = []
         File.foreach(compiled_proto) do |line|
-          if line.lstrip.start_with?('add_message')
-            msg_type = line[/"([^"]*)"/, 1] # regex: <add_message> 'msg_type' <do>
-            msg_types.push(msg_type) unless msg_type.nil?
+          line.strip!
+          if line.include?('::Google::Protobuf::DescriptorPool.generated_pool.lookup') && line.end_with?('.msgclass')
+            extracted_msg_type = line[/"([^"]*)"/, 1].freeze
+            msg_types.push(extracted_msg_type) unless extracted_msg_type.nil?
           end
         end