add gen_record() method for create a new record

Author: akaila-splunk

splunklib/searchcommands/generating_command.py

@@ -213,7 +213,15 @@ def _execute(self, ifile, process):
 
     def _execute_chunk_v2(self, process, chunk):
         count = 0
+        records = []
         for row in process:
+            records.append(row)
+            count+=1
+            if count == self._record_writer._maxresultrows:
+                break
+
+        count = 0
+        for row in records:
             self._record_writer.write_record(row)
             count += 1
             if count == self._record_writer._maxresultrows:

splunklib/searchcommands/search_command.py

@@ -177,6 +177,10 @@ def add_field(self, current_record, field_name, field_value):
         self._record_writer.custom_fields.add(field_name)
         current_record[field_name] = field_value
 
+    def gen_record(self, **record):
+        self._record_writer.custom_fields |= record.keys()
+        return {**record}
+
     record = Option(doc='''
         **Syntax: record=<bool>