mirror of
https://github.com/imgproxy/imgproxy.git
synced 2024-11-24 08:12:38 +02:00
30 lines
589 B
Go
30 lines
589 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
var timerSinceCtxKey = ctxKey("timerSince")
|
|
|
|
func startTimer(d time.Duration) (context.Context, context.CancelFunc) {
|
|
return context.WithTimeout(
|
|
context.WithValue(context.Background(), timerSinceCtxKey, time.Now()),
|
|
d,
|
|
)
|
|
}
|
|
|
|
func getTimerSince(ctx context.Context) time.Duration {
|
|
return time.Since(ctx.Value(timerSinceCtxKey).(time.Time))
|
|
}
|
|
|
|
func checkTimeout(ctx context.Context) {
|
|
select {
|
|
case <-ctx.Done():
|
|
panic(newError(503, fmt.Sprintf("Timeout after %v", getTimerSince(ctx)), "Timeout"))
|
|
default:
|
|
// Go ahead
|
|
}
|
|
}
|