28 lines
463 B
Go
28 lines
463 B
Go
|
package util
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Extracts the port as an integer from an address string.
|
||
|
|
func AddrPort(addr net.Addr) int {
|
||
|
|
_, port, err := net.SplitHostPort(addr.String())
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
i64, err := strconv.ParseInt(port, 0, 0)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return int(i64)
|
||
|
|
}
|
||
|
|
|
||
|
|
func AddrIP(addr net.Addr) net.IP {
|
||
|
|
host, _, err := net.SplitHostPort(addr.String())
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return net.ParseIP(host)
|
||
|
|
}
|