diff options
author | Chris Hand <dexteradeus@users.noreply.github.com> | 2022-08-12 09:58:57 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-08-12 09:58:57 -0400 |
commit | fc86ab0d8501b10dbe9be203625e9002cf3922ed (patch) | |
tree | 7a8e606cd850af237aae03ef654897238145ffa6 | |
parent | 45bf9f9115e6a3a455d8cc947790b64f8c7101a7 (diff) | |
download | docker-py-fc86ab0d8501b10dbe9be203625e9002cf3922ed.tar.gz |
swarm: add support for DataPathPort on init (#2987)
Adds support for setting the UDP port used for VXLAN traffic between
swarm nodes
Signed-off-by: Chris Hand <dexteradeus@users.noreply.github.com>
-rw-r--r-- | docker/api/swarm.py | 13 | ||||
-rw-r--r-- | docker/models/swarm.py | 7 | ||||
-rw-r--r-- | tests/integration/api_swarm_test.py | 5 |
3 files changed, 23 insertions, 2 deletions
diff --git a/docker/api/swarm.py b/docker/api/swarm.py index db40fdd..d09dd08 100644 --- a/docker/api/swarm.py +++ b/docker/api/swarm.py @@ -85,7 +85,7 @@ class SwarmApiMixin: def init_swarm(self, advertise_addr=None, listen_addr='0.0.0.0:2377', force_new_cluster=False, swarm_spec=None, default_addr_pool=None, subnet_size=None, - data_path_addr=None): + data_path_addr=None, data_path_port=None): """ Initialize a new Swarm using the current connected engine as the first node. @@ -118,6 +118,9 @@ class SwarmApiMixin: networks created from the default subnet pool. Default: None data_path_addr (string): Address or interface to use for data path traffic. For example, 192.168.1.1, or an interface, like eth0. + data_path_port (int): Port number to use for data path traffic. + Acceptable port range is 1024 to 49151. If set to ``None`` or + 0, the default port 4789 will be used. Default: None Returns: (str): The ID of the created node. @@ -166,6 +169,14 @@ class SwarmApiMixin: ) data['DataPathAddr'] = data_path_addr + if data_path_port is not None: + if utils.version_lt(self._version, '1.40'): + raise errors.InvalidVersion( + 'Data path port is only available for ' + 'API version >= 1.40' + ) + data['DataPathPort'] = data_path_port + response = self._post_json(url, data=data) return self._result(response, json=True) diff --git a/docker/models/swarm.py b/docker/models/swarm.py index b0b1a2e..1e39f3f 100644 --- a/docker/models/swarm.py +++ b/docker/models/swarm.py @@ -35,7 +35,8 @@ class Swarm(Model): def init(self, advertise_addr=None, listen_addr='0.0.0.0:2377', force_new_cluster=False, default_addr_pool=None, - subnet_size=None, data_path_addr=None, **kwargs): + subnet_size=None, data_path_addr=None, data_path_port=None, + **kwargs): """ Initialize a new swarm on this Engine. @@ -65,6 +66,9 @@ class Swarm(Model): networks created from the default subnet pool. Default: None data_path_addr (string): Address or interface to use for data path traffic. For example, 192.168.1.1, or an interface, like eth0. + data_path_port (int): Port number to use for data path traffic. + Acceptable port range is 1024 to 49151. If set to ``None`` or + 0, the default port 4789 will be used. Default: None task_history_retention_limit (int): Maximum number of tasks history stored. snapshot_interval (int): Number of logs entries between snapshot. @@ -121,6 +125,7 @@ class Swarm(Model): 'default_addr_pool': default_addr_pool, 'subnet_size': subnet_size, 'data_path_addr': data_path_addr, + 'data_path_port': data_path_port, } init_kwargs['swarm_spec'] = self.client.api.create_swarm_spec(**kwargs) node_id = self.client.api.init_swarm(**init_kwargs) diff --git a/tests/integration/api_swarm_test.py b/tests/integration/api_swarm_test.py index 48c0592..cffe12f 100644 --- a/tests/integration/api_swarm_test.py +++ b/tests/integration/api_swarm_test.py @@ -253,3 +253,8 @@ class SwarmTest(BaseAPIIntegrationTest): @pytest.mark.xfail(reason='Can fail if eth0 has multiple IP addresses') def test_init_swarm_data_path_addr(self): assert self.init_swarm(data_path_addr='eth0') + + @requires_api_version('1.40') + def test_init_swarm_data_path_port(self): + assert self.init_swarm(data_path_port=4242) + assert self.client.inspect_swarm()['DataPathPort'] == 4242 |