fixed restore build artifacts issue on docker hub triggering

Author: fcostaoliveira

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "redis-benchmarks-specification"
-version = "0.1.210"
+version = "0.1.213"
 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 <[email protected]>","Redis Performance Group <[email protected]>"]
 readme = "Readme.md"

redis_benchmarks_specification/__builder__/builder.py

@@ -521,22 +521,29 @@ def store_airgap_image_redis(conn, docker_client, run_image):
             run_image, airgap_key
         )
     )
-    run_image_binary_stream = io.BytesIO()
-    run_image_docker = docker_client.images.get(run_image)
-    for chunk in run_image_docker.save():
-        run_image_binary_stream.write(chunk)
+
     # 7 days expire
     binary_exp_secs = 24 * 60 * 60 * 7
-    res_airgap = conn.set(
-        airgap_key,
-        run_image_binary_stream.getbuffer(),
-        ex=binary_exp_secs,
-    )
-    logging.info(
-        "DOCKER AIR GAP: result of set bin data to {}: {}".format(
-            airgap_key, res_airgap
+    if conn.exists(airgap_key):
+        logging.info(
+            f"DOCKER AIRGAP KEY ALREADY EXISTS: {airgap_key}. Updating only the expire time"
+        )
+        conn.expire(airgap_key, binary_exp_secs)
+    else:
+        run_image_binary_stream = io.BytesIO()
+        run_image_docker = docker_client.images.get(run_image)
+        for chunk in run_image_docker.save():
+            run_image_binary_stream.write(chunk)
+        res_airgap = conn.set(
+            airgap_key,
+            run_image_binary_stream.getbuffer(),
+            ex=binary_exp_secs,
+        )
+        logging.info(
+            "DOCKER AIR GAP: result of set bin data to {}: {}".format(
+                airgap_key, res_airgap
+            )
         )
-    )
 
 
 def generate_benchmark_stream_request(

redis_benchmarks_specification/__cli__/cli.py

@@ -90,6 +90,7 @@ def trigger_tests_dockerhub_cli_command_logic(args, project_name, project_versio
     )
     build_stream_fields["github_repo"] = args.gh_repo
     build_stream_fields["github_org"] = args.gh_org
+    build_stream_fields["restore_build_artifacts"] = "False"
     server_name = args.gh_repo
     if args.server_name is not None:
         server_name = args.server_name

redis_benchmarks_specification/__self_contained_coordinator__/artifacts.py

@@ -6,19 +6,20 @@ def restore_build_artifacts_from_test_details(
     build_artifacts, conn, temporary_dir, testDetails
 ):
     for build_artifact in build_artifacts:
-        buffer_key = testDetails["{}".format(build_artifact).encode()]
-        logging.info(
-            "Reading artifact binary {} from key {}".format(build_artifact, buffer_key)
-        )
-        buffer = bytes(conn.get(buffer_key))
-        artifact_fname = "{}/{}".format(temporary_dir, build_artifact)
-        with open(artifact_fname, "wb") as fd:
-            fd.write(buffer)
-            os.chmod(artifact_fname, 755)
-        # TODO: re-enable
-        # if build_artifact == "redis-server":
-        #     redis_server_path = artifact_fname
+        build_artifact_key = "{}".format(build_artifact).encode()
+        if build_artifact_key in testDetails:
+            buffer_key = testDetails[build_artifact_key]
+            logging.info(
+                "Reading artifact binary {} from key {}".format(
+                    build_artifact, buffer_key
+                )
+            )
+            buffer = bytes(conn.get(buffer_key))
+            artifact_fname = "{}/{}".format(temporary_dir, build_artifact)
+            with open(artifact_fname, "wb") as fd:
+                fd.write(buffer)
+                os.chmod(artifact_fname, 755)
 
-        logging.info(
-            "Successfully restored {} into {}".format(build_artifact, artifact_fname)
-        )
+            logging.info(
+                "Successfully restored {} into {}".format(build_artifact, artifact_fname)
+            )