mirror of
https://github.com/tiennm99/keepalive.git
synced 2026-10-11 03:13:31 +00:00
tiennm99
0af652953f
- Each connect attempt, including initialization, times out after 1 minute, and PostgreSQL DSNs without connect_timeout get connect_timeout=30, since lib/pq honours the context only while dialing. - A service increments right after connecting, so a restart writes even when the interval outlasts the process. - A config path that is a directory (Docker's stand-in for a missing bind-mount source) fails with a clear message. - Generated service names take the first free suffix, and suffixed names are reserved so an explicit name cannot silently collide with them. - counter_key inside a service's config map is rejected instead of being overwritten. - Unknown keys in the config file log a warning without blocking start. - A .dockerignore keeps local configs and .env out of the build context.
122 lines
2.6 KiB
Go
122 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/tiennm99/keepalive/adapter"
|
|
)
|
|
|
|
// normalizeServiceName returns a unique service name and records it in
|
|
// usedNames. Explicit names must be unique; generated names take the first
|
|
// free numeric suffix.
|
|
func normalizeServiceName(rawName, adapterType string, cfg adapter.Config, usedNames map[string]bool, servicePath string) (string, error) {
|
|
if strings.TrimSpace(rawName) != "" {
|
|
name := slugify(rawName)
|
|
if name == "" {
|
|
return "", fmt.Errorf("%s.name must contain at least one letter or number", servicePath)
|
|
}
|
|
if usedNames[name] {
|
|
return "", fmt.Errorf("%s.name %q duplicates another service name", servicePath, name)
|
|
}
|
|
usedNames[name] = true
|
|
return name, nil
|
|
}
|
|
|
|
base := generatedServiceName(adapterType, cfg)
|
|
name := base
|
|
for n := 2; usedNames[name]; n++ {
|
|
name = fmt.Sprintf("%s-%d", base, n)
|
|
}
|
|
usedNames[name] = true
|
|
return name, nil
|
|
}
|
|
|
|
func generatedServiceName(adapterType string, cfg adapter.Config) string {
|
|
adapterPart := slugify(adapterType)
|
|
if adapterPart == "" {
|
|
adapterPart = "service"
|
|
}
|
|
host := serviceHost(cfg)
|
|
if host == "" {
|
|
return adapterPart
|
|
}
|
|
return adapterPart + "-" + slugify(host)
|
|
}
|
|
|
|
func serviceHost(cfg adapter.Config) string {
|
|
for _, key := range []string{"url", "uri", "connection_string", "dsn"} {
|
|
if host := hostFromEndpoint(cfg[key]); host != "" {
|
|
return host
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func hostFromEndpoint(endpoint string) string {
|
|
endpoint = strings.TrimSpace(endpoint)
|
|
if endpoint == "" {
|
|
return ""
|
|
}
|
|
|
|
if strings.Contains(endpoint, "://") {
|
|
if u, err := url.Parse(endpoint); err == nil {
|
|
if host := u.Hostname(); host != "" {
|
|
return host
|
|
}
|
|
}
|
|
}
|
|
|
|
if host := hostFromMySQLDSN(endpoint); host != "" {
|
|
return host
|
|
}
|
|
|
|
host, _, err := net.SplitHostPort(endpoint)
|
|
if err == nil && host != "" {
|
|
return host
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func hostFromMySQLDSN(dsn string) string {
|
|
start := strings.Index(dsn, "@tcp(")
|
|
if start == -1 {
|
|
return ""
|
|
}
|
|
start += len("@tcp(")
|
|
end := strings.Index(dsn[start:], ")")
|
|
if end == -1 {
|
|
return ""
|
|
}
|
|
address := dsn[start : start+end]
|
|
host, _, err := net.SplitHostPort(address)
|
|
if err == nil && host != "" {
|
|
return host
|
|
}
|
|
return address
|
|
}
|
|
|
|
func slugify(value string) string {
|
|
value = strings.ToLower(strings.TrimSpace(value))
|
|
var b strings.Builder
|
|
lastDash := false
|
|
|
|
for _, r := range value {
|
|
isAlnum := (r >= 'a' && r <= 'z') || (r >= '0' && r <= '9')
|
|
if isAlnum {
|
|
b.WriteRune(r)
|
|
lastDash = false
|
|
continue
|
|
}
|
|
if !lastDash && b.Len() > 0 {
|
|
b.WriteByte('-')
|
|
lastDash = true
|
|
}
|
|
}
|
|
|
|
return strings.Trim(b.String(), "-")
|
|
}
|