redis-benchmarks-specification 0.1.339__tar.gz → 0.1.340__tar.gz

{redis_benchmarks_specification-0.1.339 → redis_benchmarks_specification-0.1.340}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: redis-benchmarks-specification
-Version: 0.1.339
+Version: 0.1.340
 Summary: The Redis benchmarks specification describes the cross-language/tools requirements and expectations to foster performance and observability standards around redis related technologies. Members from both industry and academia, including organizations and individuals are encouraged to contribute.
 Author: filipecosta90
 Author-email: filipecosta.90@gmail.com

{redis_benchmarks_specification-0.1.339 → redis_benchmarks_specification-0.1.340}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "redis-benchmarks-specification"
-version = "0.1.339"
+version = "0.1.340"
 description = "The Redis benchmarks specification describes the cross-language/tools requirements and expectations to foster performance and observability standards around redis related technologies. Members from both industry and academia, including organizations and individuals are encouraged to contribute."
 authors = ["filipecosta90 <filipecosta.90@gmail.com>","Redis Performance Group <performance@redis.com>"]
 readme = "Readme.md"

{redis_benchmarks_specification-0.1.339 → redis_benchmarks_specification-0.1.340}/redis_benchmarks_specification/__builder__/builder.py

@@ -47,6 +47,69 @@ from redis_benchmarks_specification.__common__.package import (
 PERFORMANCE_GH_TOKEN = os.getenv("PERFORMANCE_GH_TOKEN", None)
 
 
+def clear_pending_messages_for_builder_consumer(conn, builder_group, builder_id):
+    """Clear all pending messages for a specific builder consumer on startup"""
+    consumer_name = f"{builder_group}-proc#{builder_id}"
+
+    try:
+        # Get pending messages for this specific consumer
+        pending_info = conn.xpending_range(
+            STREAM_KEYNAME_GH_EVENTS_COMMIT,
+            builder_group,
+            min="-",
+            max="+",
+            count=1000,  # Get up to 1000 pending messages
+            consumername=consumer_name,
+        )
+
+        if pending_info:
+            message_ids = [msg["message_id"] for msg in pending_info]
+            logging.info(
+                f"Found {len(message_ids)} pending messages for builder consumer {consumer_name}. Clearing them..."
+            )
+
+            # Acknowledge all pending messages to clear them
+            ack_count = conn.xack(
+                STREAM_KEYNAME_GH_EVENTS_COMMIT, builder_group, *message_ids
+            )
+
+            logging.info(
+                f"Successfully cleared {ack_count} pending messages for builder consumer {consumer_name}"
+            )
+        else:
+            logging.info(
+                f"No pending messages found for builder consumer {consumer_name}"
+            )
+
+    except redis.exceptions.ResponseError as e:
+        if "NOGROUP" in str(e):
+            logging.info(f"Builder consumer group {builder_group} does not exist yet")
+        else:
+            logging.warning(f"Error clearing pending messages: {e}")
+    except Exception as e:
+        logging.error(f"Unexpected error clearing pending messages: {e}")
+
+
+def reset_builder_consumer_group_to_latest(conn, builder_group):
+    """Reset the builder consumer group position to only read new messages (skip old ones)"""
+    try:
+        # Set the consumer group position to '$' (latest) to skip all existing messages
+        conn.xgroup_setid(STREAM_KEYNAME_GH_EVENTS_COMMIT, builder_group, id="$")
+        logging.info(
+            f"Reset builder consumer group {builder_group} position to latest - will only process new messages"
+        )
+
+    except redis.exceptions.ResponseError as e:
+        if "NOGROUP" in str(e):
+            logging.info(f"Builder consumer group {builder_group} does not exist yet")
+        else:
+            logging.warning(f"Error resetting builder consumer group position: {e}")
+    except Exception as e:
+        logging.error(
+            f"Unexpected error resetting builder consumer group position: {e}"
+        )
+
+
 class ZipFileWithPermissions(ZipFile):
     def _extract_member(self, member, targetpath, pwd):
         if not isinstance(member, ZipInfo):
@@ -104,6 +167,12 @@ def main():
     )
     parser.add_argument("--github_token", type=str, default=PERFORMANCE_GH_TOKEN)
     parser.add_argument("--pull-request", type=str, default=None, nargs="?", const="")
+    parser.add_argument(
+        "--skip-clear-pending-on-startup",
+        default=False,
+        action="store_true",
+        help="Skip automatically clearing pending messages and resetting consumer group position on startup. By default, pending messages are cleared and consumer group is reset to latest position to skip old work and recover from crashes.",
+    )
     args = parser.parse_args()
     if args.logname is not None:
         print("Writting log to {}".format(args.logname))
@@ -169,6 +238,19 @@ def main():
         builder_id = "1"
 
     builder_consumer_group_create(conn, builder_group)
+
+    # Clear pending messages and reset consumer group position by default (unless explicitly skipped)
+    if not args.skip_clear_pending_on_startup:
+        logging.info(
+            "Clearing pending messages and resetting builder consumer group position on startup (default behavior)"
+        )
+        clear_pending_messages_for_builder_consumer(conn, builder_group, builder_id)
+        reset_builder_consumer_group_to_latest(conn, builder_group)
+    else:
+        logging.info(
+            "Skipping pending message cleanup and builder consumer group reset as requested"
+        )
+
     if args.github_token is not None:
         logging.info("detected a github token. will update as much as possible!!! =)")
     previous_id = args.consumer_start_id
@@ -268,6 +350,26 @@ def builder_process_stream(
                     build_request_arch, arch
                 )
             )
+            # Acknowledge the message even though we're skipping it
+            ack_reply = conn.xack(
+                STREAM_KEYNAME_GH_EVENTS_COMMIT,
+                STREAM_GH_EVENTS_COMMIT_BUILDERS_CG,
+                streamId,
+            )
+            if type(ack_reply) == bytes:
+                ack_reply = ack_reply.decode()
+            if ack_reply == "1" or ack_reply == 1:
+                logging.info(
+                    "Successfully acknowledged build variation stream with id {} (filtered by arch).".format(
+                        streamId
+                    )
+                )
+            else:
+                logging.error(
+                    "Unable to acknowledge build variation stream with id {}. XACK reply {}".format(
+                        streamId, ack_reply
+                    )
+                )
             return previous_id, new_builds_count, build_stream_fields_arr
 
         home = str(Path.home())

{redis_benchmarks_specification-0.1.339 → redis_benchmarks_specification-0.1.340}/redis_benchmarks_specification/__self_contained_coordinator__/self_contained_coordinator.py

@@ -872,26 +872,34 @@ def self_contained_coordinator_blocking_read(
         )
         num_process_streams = num_process_streams + 1
         num_process_test_suites = num_process_test_suites + total_test_suite_runs
-        if overall_result is True:
-            ack_reply = github_event_conn.xack(
-                STREAM_KEYNAME_NEW_BUILD_EVENTS,
-                get_runners_consumer_group_name(platform_name),
-                stream_id,
-            )
-            if type(ack_reply) == bytes:
-                ack_reply = ack_reply.decode()
-            if ack_reply == "1" or ack_reply == 1:
+
+        # Always acknowledge the message, even if it was filtered out
+        ack_reply = github_event_conn.xack(
+            STREAM_KEYNAME_NEW_BUILD_EVENTS,
+            get_runners_consumer_group_name(platform_name),
+            stream_id,
+        )
+        if type(ack_reply) == bytes:
+            ack_reply = ack_reply.decode()
+        if ack_reply == "1" or ack_reply == 1:
+            if overall_result is True:
                 logging.info(
-                    "Sucessfully acknowledge BENCHMARK variation stream with id {}.".format(
+                    "Successfully acknowledged BENCHMARK variation stream with id {} (processed).".format(
                         stream_id
                     )
                 )
             else:
-                logging.error(
-                    "Unable to acknowledge build variation stream with id {}. XACK reply {}".format(
-                        stream_id, ack_reply
+                logging.info(
+                    "Successfully acknowledged BENCHMARK variation stream with id {} (filtered/skipped).".format(
+                        stream_id
                     )
                 )
+        else:
+            logging.error(
+                "Unable to acknowledge build variation stream with id {}. XACK reply {}".format(
+                    stream_id, ack_reply
+                )
+            )
     return overall_result, stream_id, num_process_streams, num_process_test_suites