From bd047b25bb7ed46953fc5cb976454455b931e94a Mon Sep 17 00:00:00 2001 From: LandaMm Date: Sun, 20 Apr 2025 13:53:29 +0200 Subject: [PATCH] feat: set port in addr --- hsp/address.go | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/hsp/address.go b/hsp/address.go index 7215196..a062665 100644 --- a/hsp/address.go +++ b/hsp/address.go @@ -7,6 +7,7 @@ import ( type Adddress struct { Host string + Port string Route string } @@ -24,12 +25,23 @@ func ParseAddress(address string) (*Adddress, error) { addr := parts[0] + port := HSP_PORT + + if strings.Contains(addr, ":") { + p := strings.Split(addr, ":") + if len(p) >= 2 { + port = p[len(p) - 1] + addr = p[0] + } + } + return &Adddress{ Host: addr, + Port: port, Route: route, }, nil } func (a *Adddress) String() string { - return fmt.Sprintf("%s:%s", a.Host, HSP_PORT) + return fmt.Sprintf("%s:%s", a.Host, a.Port) }