110 lines
2.2 KiB
Go
110 lines
2.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/golang/glog"
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
type MonitorType int
|
|
|
|
const (
|
|
Monitor_PORT MonitorType = 1
|
|
Monitor_EXEC MonitorType = 2
|
|
Monitor_CONSUL MonitorType = 3
|
|
)
|
|
|
|
var MonitorMap = map[string]MonitorType{"port": Monitor_PORT, "exec": Monitor_EXEC, "consul": Monitor_CONSUL}
|
|
|
|
func (m MonitorType) String() string {
|
|
for str, mtr := range MonitorMap {
|
|
if m == mtr {
|
|
return str
|
|
}
|
|
}
|
|
return "unknown"
|
|
}
|
|
|
|
type Monitor struct {
|
|
Type MonitorType
|
|
Port string
|
|
Protocol string
|
|
Cmd string
|
|
}
|
|
|
|
func (m *Monitor) Equal(other *Monitor) bool {
|
|
return m.Type == other.Type && m.Port == other.Port && m.Protocol == other.Protocol && m.Cmd == other.Cmd
|
|
}
|
|
|
|
type Monitors []*Monitor
|
|
|
|
func (m Monitors) Contains(elem *Monitor) bool {
|
|
for _, mon := range m {
|
|
if mon.Equal(elem) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
type App struct {
|
|
Name string
|
|
Vip *net.IPNet
|
|
Monitors Monitors
|
|
}
|
|
|
|
func (a *App) Equal(other *App) bool {
|
|
if len(a.Monitors) != len(other.Monitors) {
|
|
return false
|
|
}
|
|
for _, m := range other.Monitors {
|
|
if !a.Monitors.Contains(m) {
|
|
return false
|
|
}
|
|
}
|
|
return a.Name == other.Name && a.Vip.String() == other.Vip.String()
|
|
}
|
|
|
|
func (a *App) needsNatRule() (bool, *Monitor) {
|
|
for _, m := range a.Monitors {
|
|
if m.Type == Monitor_CONSUL && m.Port != "" {
|
|
return true, m
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func NewApp(appName, vip string, monitors []string) (*App, error) {
|
|
if appName == "" {
|
|
return nil, fmt.Errorf("Invalid app name")
|
|
}
|
|
app := &App{Name: appName}
|
|
_, ipnet, err := net.ParseCIDR(vip)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Invalid VIP specified, need ip/mask")
|
|
}
|
|
app.Vip = ipnet
|
|
for _, m := range monitors {
|
|
parts := strings.Split(m, ":")
|
|
if len(parts) != 2 && len(parts) != 3 {
|
|
glog.Errorf("Invalid monitor specified, ignoring")
|
|
continue
|
|
}
|
|
mon := &Monitor{Type: MonitorMap[parts[0]]}
|
|
switch mon.Type.String() {
|
|
case "port":
|
|
mon.Protocol = parts[1]
|
|
mon.Port = parts[2]
|
|
case "exec":
|
|
mon.Cmd = parts[1]
|
|
case "consul":
|
|
glog.V(2).Infof("Using consul health monitor")
|
|
default:
|
|
glog.V(2).Infof("No monitor specified")
|
|
}
|
|
app.Monitors = append(app.Monitors, mon)
|
|
}
|
|
return app, nil
|
|
}
|