unity-builder/src/model/cli/shared-workspace-locking.ts

63 lines
2.2 KiB
TypeScript
Raw Normal View History

2022-09-16 22:51:19 +00:00
import { CloudRunnerSystem } from '../cloud-runner/services/cloud-runner-system';
import * as fs from 'fs';
import CloudRunner from '../cloud-runner/cloud-runner';
2022-09-17 03:38:00 +00:00
import CloudRunnerLogger from '../cloud-runner/services/cloud-runner-logger';
2022-09-17 03:41:41 +00:00
import CloudRunnerOptions from '../cloud-runner/cloud-runner-options';
2022-09-16 21:24:23 +00:00
export class SharedWorkspaceLocking {
2022-09-16 22:51:19 +00:00
public static async GetLockedWorkspace() {
2022-09-17 03:41:41 +00:00
if (!CloudRunnerOptions.retainWorkspaces) {
return;
}
2022-09-16 21:24:23 +00:00
const workspaces = SharedWorkspaceLocking.GetFreeWorkspaces();
for (const element of workspaces) {
2022-09-16 22:51:19 +00:00
if (await SharedWorkspaceLocking.LockWorkspace(element)) {
2022-09-16 21:24:23 +00:00
return element;
}
}
2022-09-17 04:44:07 +00:00
return await SharedWorkspaceLocking.CreateLockableWorkspace(CloudRunner.buildParameters.buildGuid);
2022-09-16 21:24:23 +00:00
}
public static GetFreeWorkspaces(): string[] {
2022-09-17 04:09:36 +00:00
return [];
2022-09-16 21:24:23 +00:00
}
public static GetAllWorkspaces(): string[] {
return [];
}
2022-09-16 22:51:19 +00:00
public static async LockWorkspace(workspace: string): Promise<boolean> {
2022-09-17 03:49:44 +00:00
const file = `${Date.now()}_${CloudRunner.buildParameters.buildGuid}_lock`;
2022-09-16 22:51:19 +00:00
fs.writeFileSync(file, '');
2022-09-17 03:25:29 +00:00
await CloudRunnerSystem.Run(`aws s3 cp ./${file} s3://game-ci-test-storage/locks/${workspace}/${file}`);
2022-09-16 22:51:19 +00:00
fs.rmSync(file);
return SharedWorkspaceLocking.HasWorkspaceLock(workspace);
}
public static async HasWorkspaceLock(workspace: string): Promise<boolean> {
2022-09-17 03:38:00 +00:00
CloudRunnerLogger.log(
(await CloudRunnerSystem.Run(`aws s3 ls s3://game-ci-test-storage/locks/${workspace}/`))
.split('\n')
.map((x) => {
return x.split(' ');
})
.length.toString(),
);
2022-09-16 22:51:19 +00:00
2022-09-16 21:24:23 +00:00
return true;
}
// eslint-disable-next-line no-unused-vars
public static IsWorkspaceLocked(workspace: string) {}
2022-09-17 04:09:36 +00:00
public static async CreateLockableWorkspace(workspace: string) {
const file = `${Date.now()}_${CloudRunner.buildParameters.buildGuid}_workspace`;
fs.writeFileSync(file, '');
await CloudRunnerSystem.Run(`aws s3 cp ./${file} s3://game-ci-test-storage/locks/${workspace}/${file}`);
2022-09-17 04:44:07 +00:00
return CloudRunner.buildParameters.buildGuid;
2022-09-17 04:09:36 +00:00
}
2022-09-16 21:24:23 +00:00
// eslint-disable-next-line no-unused-vars
public static ReleaseLock(workspace: string) {}
}
2022-09-16 18:48:40 +00:00
export default SharedWorkspaceLocking;