badblocks/docker-easy-haproxy
1
0
Fork 0
Code Activity

Fix Swarm networking

Browse source
This commit is contained in:
Joao Gilberto Magalhaes 2023-02-27 23:28:09 -06:00
parent aa5f8055f5
commit c0eb766339
3 changed files with 31 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

30
src/processor/__init__.py
View file

@ -40,6 +40,7 @@ class ProcessorInterface:
def __init__(self, filename = None):
self.filename = filename
self.label = ContainerEnv.read()['lookup_label']
self.refresh()
@staticmethod
@ -156,25 +157,40 @@ class Swarm(ProcessorInterface):
def inspect_network(self):
ha_proxy_service_name = self.client.containers.get(socket.gethostname()).name.split('.')[0]
for endpoint in self.client.services.get(ha_proxy_service_name).attrs['Endpoint']["VirtualIPs"]:
ha_proxy_network_id = endpoint["NetworkID"]
if self.client.networks.get(ha_proxy_network_id).name != 'ingress':
ha_proxy_network_id = None
swarm_ingress_id = None
# Get the HAProxy network and the ingress network
for endpoint in self.client.services.get(ha_proxy_service_name).attrs['Endpoint']["VirtualIPs"]:
network_name = self.client.networks.get(endpoint["NetworkID"]).name
if swarm_ingress_id is None and network_name == 'ingress':
swarm_ingress_id = endpoint["NetworkID"]
if ha_proxy_network_id is None and network_name != 'ingress':
ha_proxy_network_id = endpoint["NetworkID"]
if ha_proxy_network_id is not None and swarm_ingress_id is not None:
break
# Check if the service is attached to the HAProxy network
self.parsed_object = {}
for service in self.client.services.list():
if not any(self.label in key for key in service.attrs["Spec"]["Labels"]):
continue
ip_address = None
network_list = []
for endpoint in service.attrs["Endpoint"]["VirtualIPs"]:
if ha_proxy_network_id == endpoint["NetworkID"]:
ip_address = endpoint["Addr"].split("/")[0]
break
elif swarm_ingress_id != endpoint["NetworkID"]:
network_list.append(endpoint["NetworkID"])
# add the network ha_proxy_network_id to the service object
# Attach the service to the HAProxy network
if ip_address is None:
network_attachment = docker.types.NetworkAttachmentConfig(target=ha_proxy_network_id)
service.update(networks = [network_attachment])
network_list.append(ha_proxy_network_id)
service.update(networks = network_list)
continue # skip to the next service to give time to update the network
self.parsed_object[ip_address] = service.attrs["Spec"]["Labels"]