Adding network structures.

net.go

+// Copyright (c) 2012-2014 José Carlos Nieto, https://menteslibres.net/xiam
+//
+// Permission is hereby granted, free of charge, to any person obtaining
+// a copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Software, and to
+// permit persons to whom the Software is furnished to do so, subject to
+// the following conditions:
+//
+// The above copyright notice and this permission notice shall be
+// included in all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+package db
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+)
+
+// Address is an interface that represents the host part of an URL.
+type Address interface {
+	String() string
+}
+
+// socket is a UNIX address.
+type socket struct {
+	path string
+}
+
+// host is the name or IP of a server coupled with an optional port number.
+type host struct {
+	name string
+	port uint
+}
+
+// cluster is an array of hosts or sockets.
+type cluster struct {
+	address []Address
+}
+
+// ParseCluster parses s into a cluster structure.
+func ParseCluster(s string) (c cluster) {
+	var hosts []string
+	hosts = strings.Split(s, ",")
+	l := len(hosts)
+	for i := 0; i < l; i++ {
+		c.address = append(c.address, ParseAddress(hosts[i]))
+	}
+	return
+}
+
+// ParseAddress parses s into a host or socket structures.
+func ParseAddress(s string) Address {
+	if strings.HasPrefix(s, "/") {
+		// Let's suppose this is a UNIX socket.
+		return socket{path: s}
+	}
+	hp := strings.Split(s, ":")
+	if len(hp) > 1 {
+		p, _ := strconv.Atoi(hp[1])
+		return host{name: hp[0], port: uint(p)}
+	}
+	return host{name: hp[0]}
+}
+
+// Host converts the given name into a host structure.
+func Host(name string) host {
+	return host{name: name}
+}
+
+// HostPort converts the given name and port into a host structure.
+func HostPort(name string, port uint) host {
+	return host{name: name, port: port}
+}
+
+// Cluster converts the given Address structures into a cluster structure.
+func Cluster(addresses ...Address) cluster {
+	return cluster{address: addresses}
+}
+
+// String() returns the string representation of the host struct.
+func (h host) String() string {
+	if h.port > 0 {
+		return fmt.Sprintf("%s:%d", h.name, h.port)
+	}
+	return h.name
+}
+
+// String() returns the string representation of the socket struct.
+func (s socket) String() string {
+	return s.path
+}
+
+// String() returns the string representation of the cluster struct.
+func (c cluster) String() string {
+	l := len(c.address)
+
+	addresses := make([]string, 0, l)
+	for i := 0; i < l; i++ {
+		addresses = append(addresses, c.address[i].String())
+	}
+	return strings.Join(addresses, ",")
+}