You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.2 KiB
43 lines
1.2 KiB
package common
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
// Connect dials the given address and returns a net.Conn. The protoAddr argument should be prefixed with the protocol,
|
|
// eg. "tcp://127.0.0.1:8080" or "unix:///tmp/test.sock"
|
|
func Connect(protoAddr string) (net.Conn, error) {
|
|
proto, address := ProtocolAndAddress(protoAddr)
|
|
conn, err := net.Dial(proto, address)
|
|
return conn, err
|
|
}
|
|
|
|
// ProtocolAndAddress splits an address into the protocol and address components.
|
|
// For instance, "tcp://127.0.0.1:8080" will be split into "tcp" and "127.0.0.1:8080".
|
|
// If the address has no protocol prefix, the default is "tcp".
|
|
func ProtocolAndAddress(listenAddr string) (string, string) {
|
|
protocol, address := "tcp", listenAddr
|
|
parts := strings.SplitN(address, "://", 2)
|
|
if len(parts) == 2 {
|
|
protocol, address = parts[0], parts[1]
|
|
}
|
|
return protocol, address
|
|
}
|
|
|
|
// GetFreePort gets a free port from the operating system.
|
|
// Ripped from https://github.com/phayes/freeport.
|
|
// BSD-licensed.
|
|
func GetFreePort() (int, error) {
|
|
addr, err := net.ResolveTCPAddr("tcp", "localhost:0")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
l, err := net.ListenTCP("tcp", addr)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
defer l.Close()
|
|
return l.Addr().(*net.TCPAddr).Port, nil
|
|
}
|