1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
|
package endpoints
import (
"fmt"
"net"
"time"
"github.com/lxc/incus/v6/internal/ports"
"github.com/lxc/incus/v6/internal/server/endpoints/listeners"
internalUtil "github.com/lxc/incus/v6/internal/util"
"github.com/lxc/incus/v6/shared/logger"
)
// ClusterAddress returns the cluster address of the cluster endpoint, or an
// empty string if there's no cluster endpoint or the cluster endpoint is provided by the network listener.
func (e *Endpoints) clusterAddress() string {
e.mu.RLock()
defer e.mu.RUnlock()
listener := e.listeners[cluster]
if listener == nil {
return ""
}
return listener.Addr().String()
}
// ClusterUpdateAddress updates the address for the cluster endpoint, shutting
// it down and restarting it.
func (e *Endpoints) ClusterUpdateAddress(address string) error {
networkAddress := e.NetworkAddress()
if address != "" {
address = internalUtil.CanonicalNetworkAddress(address, ports.HTTPSDefaultPort)
}
oldAddress := e.clusterAddress()
if address == oldAddress {
return nil
}
logger.Infof("Update cluster address")
e.mu.Lock()
defer e.mu.Unlock()
// Close the previous socket
_ = e.closeListener(cluster)
// If turning off listening, we're done
if address == "" {
return nil
}
// If networkAddress is set and address is covered, we don't need a new listener.
if networkAddress != "" && internalUtil.IsAddressCovered(address, networkAddress) {
return nil
}
// Attempt to setup the new listening socket
getListener := func(address string) (*net.Listener, error) {
var err error
var listener net.Listener
for range 10 { // Ten retries over a second seems reasonable.
listener, err = net.Listen("tcp", address)
if err == nil {
break
}
time.Sleep(100 * time.Millisecond)
}
if err != nil {
return nil, fmt.Errorf("Cannot listen on cluster HTTPS socket %q: %w", address, err)
}
return &listener, nil
}
// set up the listener
listener, err := getListener(address)
if err != nil {
// Attempt to revert to the previous address
listener, err1 := getListener(oldAddress)
if err1 == nil {
e.listeners[cluster] = listeners.NewFancyTLSListener(*listener, e.cert)
e.serve(cluster)
}
return err
}
e.listeners[cluster] = listeners.NewFancyTLSListener(*listener, e.cert)
e.serve(cluster)
return nil
}
|