From 9339a9afeb032a01316203014b4f3b83b4462b7f Mon Sep 17 00:00:00 2001
From: williamlardier <william.lardier@scality.com>
Date: Mon, 23 Dec 2024 14:38:48 +0100
Subject: [PATCH] Use a file with wx flag to control concurrency

Issue: ZENKO-4941
---
 tests/ctst/steps/utils/kubernetes.ts | 38 +++++++++++++---------------
 1 file changed, 18 insertions(+), 20 deletions(-)

diff --git a/tests/ctst/steps/utils/kubernetes.ts b/tests/ctst/steps/utils/kubernetes.ts
index 636353c50..62bfdb0a2 100644
--- a/tests/ctst/steps/utils/kubernetes.ts
+++ b/tests/ctst/steps/utils/kubernetes.ts
@@ -1,6 +1,5 @@
 import fs from 'fs';
 import * as path from 'path';
-import lockFile from 'proper-lockfile';
 import { KubernetesHelper, Utils } from 'cli-testing';
 import Zenko from 'world/Zenko';
 import {
@@ -83,24 +82,27 @@ export async function createJobAndWaitForCompletion(
     const watchClient = createKubeWatchClient(world);
 
     const lockFilePath = path.join('/tmp', `${jobName}.lock`);
-    let releaseLock: (() => Promise<void>) | false = false;
 
-    if (!fs.existsSync(lockFilePath)) {
-        fs.writeFileSync(lockFilePath, '');
+    let lockAquired = false;
+    let tries = 600;
+    while (!lockAquired && tries > 0) {
+        try {
+            fs.writeFileSync(lockFilePath, 'lock', {
+                flag: 'wx',
+            });
+            lockAquired = true;
+        } catch {
+            world.logger.debug(`Failed to acquire lock for job: ${jobName}`, {
+                tries,
+            });            
+        }
+        tries--;
+        if (!lockAquired) {
+            await Utils.sleep(1000);
+        }
     }
 
     try {
-        releaseLock = await lockFile.lock(lockFilePath, {
-            // Expect the jobs in the queue does not take more than 5 minutes to complete
-            stale: 10 * 60 * 1000,
-            // use a linear backoff strategy
-            retries: {
-                retries: 610,
-                factor: 1,
-                minTimeout: 1000,
-                maxTimeout: 1000,
-            },
-        });
         world.logger.debug(`Acquired lock for job: ${jobName}`);
 
         // Read the cron job and prepare the job spec
@@ -159,11 +161,7 @@ export async function createJobAndWaitForCompletion(
         });
         throw err;
     } finally {
-        // Ensure the lock is released
-        if (releaseLock) {
-            await releaseLock();
-            world.logger.debug(`Released lock for job: ${jobName}`);
-        }
+        fs.unlinkSync(lockFilePath);
     }
 }