mirror of https://github.com/actions/checkout
62 lines
1.6 KiB
TypeScript
62 lines
1.6 KiB
TypeScript
import * as core from '@actions/core'
|
|
|
|
const defaultMaxAttempts = 3
|
|
const defaultMinSeconds = 10
|
|
const defaultMaxSeconds = 20
|
|
|
|
export class RetryHelper {
|
|
private maxAttempts: number
|
|
private minSeconds: number
|
|
private maxSeconds: number
|
|
|
|
constructor(
|
|
maxAttempts: number = defaultMaxAttempts,
|
|
minSeconds: number = defaultMinSeconds,
|
|
maxSeconds: number = defaultMaxSeconds
|
|
) {
|
|
this.maxAttempts = maxAttempts
|
|
this.minSeconds = Math.floor(minSeconds)
|
|
this.maxSeconds = Math.floor(maxSeconds)
|
|
if (this.minSeconds > this.maxSeconds) {
|
|
throw new Error('min seconds should be less than or equal to max seconds')
|
|
}
|
|
}
|
|
|
|
async execute<T>(action: () => Promise<T>): Promise<T> {
|
|
let attempt = 1
|
|
while (attempt < this.maxAttempts) {
|
|
// Try
|
|
try {
|
|
return await action()
|
|
} catch (err) {
|
|
core.info((err as any)?.message)
|
|
}
|
|
|
|
// Sleep
|
|
const seconds = this.getSleepAmount()
|
|
core.info(`Waiting ${seconds} seconds before trying again`)
|
|
await this.sleep(seconds)
|
|
attempt++
|
|
}
|
|
|
|
// Last attempt
|
|
return await action()
|
|
}
|
|
|
|
private getSleepAmount(): number {
|
|
return (
|
|
Math.floor(Math.random() * (this.maxSeconds - this.minSeconds + 1)) +
|
|
this.minSeconds
|
|
)
|
|
}
|
|
|
|
private async sleep(seconds: number): Promise<void> {
|
|
return new Promise(resolve => setTimeout(resolve, seconds * 1000))
|
|
}
|
|
}
|
|
|
|
export async function execute<T>(action: () => Promise<T>): Promise<T> {
|
|
const retryHelper = new RetryHelper()
|
|
return await retryHelper.execute(action)
|
|
}
|