huoxue1
/
build-push-action
mirror of https://github.com/docker/build-push-action.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

report cache hit ratio and build size back to backend (#40)

This commit is contained in:
Aayush Shah 2024-11-25 14:08:55 -05:00 committed by GitHub
parent ed90c2b818
commit 03e2c54ea5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main.ts
View File

@ -6,6 +6,7 @@ import * as actionsToolkit from '@docker/actions-toolkit';
import {Buildx} from '@docker/actions-toolkit/lib/buildx/buildx'; import {Buildx} from '@docker/actions-toolkit/lib/buildx/buildx';
import {History as BuildxHistory} from '@docker/actions-toolkit/lib/buildx/history'; import {History as BuildxHistory} from '@docker/actions-toolkit/lib/buildx/history';
import {ExportRecordResponse} from '@docker/actions-toolkit/lib/types/buildx/history';
import {Context} from '@docker/actions-toolkit/lib/context'; import {Context} from '@docker/actions-toolkit/lib/context';
import {Docker} from '@docker/actions-toolkit/lib/docker/docker'; import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
import {Exec} from '@docker/actions-toolkit/lib/exec'; import {Exec} from '@docker/actions-toolkit/lib/exec';
@ -37,12 +38,14 @@ async function getBlacksmithAgentClient(): Promise<AxiosInstance> {
} }
// Reports a successful build to the local sticky disk manager // Reports a successful build to the local sticky disk manager
async function reportBuildCompleted() { async function reportBuildCompleted(exportRes?: ExportRecordResponse) {
if (!stateHelper.blacksmithDockerBuildId) { if (!stateHelper.blacksmithDockerBuildId) {
core.warning('No docker build ID found, skipping build completion report'); core.warning('No docker build ID found, skipping build completion report');
return; return;
} }
core.warning(`reportBuildCompleted: exportRes: ${JSON.stringify(exportRes, null, 2)}`);
try { try {
const client = await getBlacksmithAgentClient(); const client = await getBlacksmithAgentClient();
const formData = new FormData(); const formData = new FormData();
@ -61,6 +64,17 @@ async function reportBuildCompleted() {
runtime_seconds: stateHelper.dockerBuildDurationSeconds runtime_seconds: stateHelper.dockerBuildDurationSeconds
}; };
core.debug(`exportRes: ${JSON.stringify(exportRes, null, 2)}`);
core.debug(`stateHelper.buildRef: ${stateHelper.buildRef}`);
if (exportRes && stateHelper.buildRef) {
const buildRefSummary = exportRes.summaries[stateHelper.buildRef];
const cachedRatio = buildRefSummary.numCachedSteps / buildRefSummary.numTotalSteps;
requestOptions['docker_build_size'] = exportRes.dockerbuildSize;
requestOptions['cached_steps_ratio'] = cachedRatio;
}
await postWithRetryToBlacksmithAPI(`/stickydisks/dockerbuilds/${stateHelper.blacksmithDockerBuildId}`, requestOptions, retryCondition); await postWithRetryToBlacksmithAPI(`/stickydisks/dockerbuilds/${stateHelper.blacksmithDockerBuildId}`, requestOptions, retryCondition);
return; return;
} catch (error) { } catch (error) {
@ -687,6 +701,7 @@ actionsToolkit.run(
}, },
// post // post
async () => { async () => {
let exportRes;
if (stateHelper.isSummarySupported) { if (stateHelper.isSummarySupported) {
await core.group(`Generating build summary`, async () => { await core.group(`Generating build summary`, async () => {
try { try {
@ -697,7 +712,7 @@ actionsToolkit.run(
} }
const buildxHistory = new BuildxHistory(); const buildxHistory = new BuildxHistory();
const exportRes = await buildxHistory.export({ exportRes = await buildxHistory.export({
refs: stateHelper.buildRef ? [stateHelper.buildRef] : [] refs: stateHelper.buildRef ? [stateHelper.buildRef] : []
}); });
core.info(`Build record written to ${exportRes.dockerbuildFilename} (${Util.formatFileSize(exportRes.dockerbuildSize)})`); core.info(`Build record written to ${exportRes.dockerbuildFilename} (${Util.formatFileSize(exportRes.dockerbuildSize)})`);
@ -737,8 +752,10 @@ actionsToolkit.run(
await new Promise(resolve => setTimeout(resolve, 100)); await new Promise(resolve => setTimeout(resolve, 100));
} }
} }
core.warning(`stateHelper.dockerBuildStatus: ${stateHelper.dockerBuildStatus}`);
core.warning(`exportRes: ${JSON.stringify(exportRes, null, 2)}`);
if (stateHelper.dockerBuildStatus == 'success') { if (stateHelper.dockerBuildStatus == 'success') {
await reportBuildCompleted(); await reportBuildCompleted(exportRes);
} else { } else {
await reportBuildFailed(); await reportBuildFailed();
} }