Add executable
This commit is contained in:
parent
d28e60d19f
commit
931d21b0c8
42
cmd/main.go
Normal file
42
cmd/main.go
Normal file
@ -0,0 +1,42 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"git.fuyu.moe/Fuyu/waitfor"
|
||||
)
|
||||
|
||||
var settings = struct {
|
||||
Endpoints []string
|
||||
Timeout time.Duration
|
||||
}{
|
||||
Endpoints: strings.Split(os.Getenv("WAITFOR"), `,`),
|
||||
Timeout: time.Second * 10,
|
||||
}
|
||||
|
||||
func init() {
|
||||
d, err := time.ParseDuration(os.Getenv("WAITFOR_TIMEOUT"))
|
||||
if err == nil {
|
||||
settings.Timeout = d
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
for _, endpoint := range settings.Endpoints {
|
||||
endpoint = strings.TrimSpace(endpoint)
|
||||
if endpoint == `` {
|
||||
continue
|
||||
}
|
||||
|
||||
// Validate endpoint
|
||||
_, _, err := net.SplitHostPort(endpoint)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
waitfor.Try(`tcp`, endpoint, time.Second, settings.Timeout)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user