func writeStateConfig() {
configMutex.Lock()
defer configMutex.Unlock()
if stateConfig == "" {
log.Debug("DEBUG: No state file. Not saving changes")
return
}
cfg := marshal(Registry.Config())
if len(cfg) == 0 {
return
}
lastCfg, _ := ioutil.ReadFile(stateConfig)
if bytes.Equal(cfg, lastCfg) {
log.Println("INFO: No change in config")
return
}
// We should probably write a temp file and mv for atomic update.
err := ioutil.WriteFile(stateConfig, cfg, 0644)
if err != nil {
log.Errorln("ERROR: Can't save config state:", err)
}
}
func startAdminHTTPServer(wg *sync.WaitGroup) {
defer wg.Done()
addHandlers()
log.Println("INFO: Admin server listening on", adminListenAddr)
netw := "tcp"
if strings.HasPrefix(adminListenAddr, "/") {
netw = "unix"
// remove our old socket if we left it lying around
if stats, err := os.Stat(adminListenAddr); err == nil {
if stats.Mode()&os.ModeSocket != 0 {
os.Remove(adminListenAddr)
}
}
defer os.Remove(adminListenAddr)
}
listener, err := net.Listen(netw, adminListenAddr)
if err != nil {
log.Fatalf("FATAL: Admin server failed and exited with %s", err)
}
http.Serve(listener, nil)
}
// update the VirtualHost entries for this service
// only to be called from UpdateService.
func (s *ServiceRegistry) updateVHosts(service *Service, newHosts []string) {
// We could just clear the vhosts and the new list since we're doing
// this all while the registry is locked, but because we want sane log
// messages about adding remove endpoints, we have to diff the slices
// anyway.
oldHosts := service.VirtualHosts
sort.Strings(oldHosts)
sort.Strings(newHosts)
// find the relative compliments of each set of hostnames
var remove, add []string
i, j := 0, 0
for i < len(oldHosts) && j < len(newHosts) {
if oldHosts[i] != newHosts[j] {
if oldHosts[i] < newHosts[j] {
// oldHosts[i] can't be in newHosts
remove = append(remove, oldHosts[i])
i++
continue
} else {
// newHosts[j] can't be in oldHosts
add = append(add, newHosts[j])
j++
continue
}
}
i++
j++
}
if i < len(oldHosts) {
// there's more!
remove = append(remove, oldHosts[i:]...)
}
if j < len(newHosts) {
add = append(add, newHosts[j:]...)
}
// remove existing vhost entries for this service, and add new ones
for _, name := range remove {
vhost := s.vhosts[name]
if vhost != nil {
vhost.Remove(service)
}
if vhost.Len() == 0 {
log.Println("INFO: Removing empty VirtualHost", name)
delete(s.vhosts, name)
}
}
for _, name := range add {
vhost := s.vhosts[name]
if vhost == nil {
vhost = &VirtualHost{Name: name}
s.vhosts[name] = vhost
}
vhost.Add(service)
}
// and replace the list
service.VirtualHosts = newHosts
}