mirror of
https://github.com/hrfee/jfa-go.git
synced 2024-11-09 20:00:12 +00:00
24 lines
456 B
Go
24 lines
456 B
Go
|
package common
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TimeoutHandler recovers from an http timeout.
|
||
|
|
type TimeoutHandler func()
|
||
|
|
|
||
|
|
// NewTimeoutHandler returns a new Timeout handler.
|
||
|
|
func NewTimeoutHandler(name, addr string, noFail bool) TimeoutHandler {
|
||
|
|
return func() {
|
||
|
|
if r := recover(); r != nil {
|
||
|
|
out := fmt.Sprintf("Failed to authenticate with %s @ %s: Timed out", name, addr)
|
||
|
|
if noFail {
|
||
|
|
log.Print(out)
|
||
|
|
} else {
|
||
|
|
log.Fatalf(out)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|