actions/toolkit
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/actions/toolkit synced 2026-04-02 07:43:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Bethany
2023-08-07 16:26:07 -07:00
parent 6552cb9722
commit af1621025d
2 changed files with 51 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
packages/artifact/src/internal/shared/artifact-twirp-client.ts
View File

@@ -21,17 +21,13 @@ class ArtifactHttpClient implements Rpc {
private baseRetryIntervalMilliseconds: number = 3000
private retryMultiplier: number = 1.5
constructor(userAgent: string, httpClient?: HttpClient) {
constructor(userAgent: string) {
const token = getRuntimeToken()
this.baseUrl = getResultsServiceUrl()
if (httpClient) {
this.httpClient = httpClient
} else {
this.httpClient = new HttpClient(
userAgent,
[new BearerCredentialHandler(token)],
)
}
this.httpClient = new HttpClient(
userAgent,
[new BearerCredentialHandler(token)],
)
}
// This function satisfies the Rpc interface. It is compatible with the JSON
@@ -79,11 +75,18 @@ class ArtifactHttpClient implements Rpc {
throw new Error(errorMessage)
}
if (attempt + 1 === this.maxAttempts) {
info(`Final attempt failed with error: ${errorMessage}`)
break
}
const retryTimeMilliseconds = this.getExponentialRetryTimeMilliseconds(attempt)
info(
`Attempt ${attempt + 1} of ${this.maxAttempts} failed with error: ${errorMessage}. Retrying request...`
`Attempt ${attempt + 1} of ${this.maxAttempts} failed with error: ${errorMessage}. Retrying request in ${retryTimeMilliseconds} ms...`
)
await this.sleep(this.getExponentialRetryTimeMilliseconds(attempt))
await this.sleep(retryTimeMilliseconds)
attempt++
}
@@ -131,7 +134,7 @@ class ArtifactHttpClient implements Rpc {
}
}
export function createArtifactTwirpClient(type: "upload" | "download", httpClient?: HttpClient): ArtifactServiceClientJSON {
const client = new ArtifactHttpClient(`@actions/artifact-${type}`, httpClient)
export function createArtifactTwirpClient(type: "upload" | "download"): ArtifactServiceClientJSON {
const client = new ArtifactHttpClient(`@actions/artifact-${type}`)
return new ArtifactServiceClientJSON(client)
}