71 lines
1.3 KiB
Go
71 lines
1.3 KiB
Go
|
package server_details
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"github.com/hashicorp/serf/serf"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ServerDetails is used to return details of a consul server
|
||
|
|
type ServerDetails struct {
|
||
|
|
Name string
|
||
|
|
Datacenter string
|
||
|
|
Port int
|
||
|
|
Bootstrap bool
|
||
|
|
Expect int
|
||
|
|
Version int
|
||
|
|
Addr net.Addr
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ServerDetails) String() string {
|
||
|
|
return fmt.Sprintf("%s (Addr: %s) (DC: %s)", s.Name, s.Addr, s.Datacenter)
|
||
|
|
}
|
||
|
|
|
||
|
|
// IsConsulServer returns true if a serf member is a consul server. Returns a
|
||
|
|
// bool and a pointer to the ServerDetails.
|
||
|
|
func IsConsulServer(m serf.Member) (bool, *ServerDetails) {
|
||
|
|
if m.Tags["role"] != "consul" {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
datacenter := m.Tags["dc"]
|
||
|
|
_, bootstrap := m.Tags["bootstrap"]
|
||
|
|
|
||
|
|
expect := 0
|
||
|
|
expect_str, ok := m.Tags["expect"]
|
||
|
|
var err error
|
||
|
|
if ok {
|
||
|
|
expect, err = strconv.Atoi(expect_str)
|
||
|
|
if err != nil {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
port_str := m.Tags["port"]
|
||
|
|
port, err := strconv.Atoi(port_str)
|
||
|
|
if err != nil {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
vsn_str := m.Tags["vsn"]
|
||
|
|
vsn, err := strconv.Atoi(vsn_str)
|
||
|
|
if err != nil {
|
||
|
|
return false, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
addr := &net.TCPAddr{IP: m.Addr, Port: port}
|
||
|
|
|
||
|
|
parts := &ServerDetails{
|
||
|
|
Name: m.Name,
|
||
|
|
Datacenter: datacenter,
|
||
|
|
Port: port,
|
||
|
|
Bootstrap: bootstrap,
|
||
|
|
Expect: expect,
|
||
|
|
Addr: addr,
|
||
|
|
Version: vsn,
|
||
|
|
}
|
||
|
|
return true, parts
|
||
|
|
}
|