Update awx_collection to support ReceptorAddress

- Add receptor_address module which allows
users to create addresses for instances

- Update awx_collection functional and integration
tests to support new peering design

Signed-off-by: Seth Foster <fosterbseth@gmail.com>
This commit is contained in:
Seth Foster
2023-11-15 12:34:19 -05:00
committed by Seth Foster
parent d1cacf64de
commit 04cbbbccfa
10 changed files with 236 additions and 52 deletions

View File

@@ -58,23 +58,15 @@ options:
- installed
required: False
type: str
listener_port:
description:
- Port that Receptor will listen for incoming connections on.
required: False
type: int
peers:
description:
- List of peers to connect outbound to. Only configurable for hop and execution nodes.
- To remove all current peers, set value to an empty list, [].
- Each item is an ID or address of a receptor address.
- If item is address, it must be unique across all receptor addresses.
required: False
type: list
elements: str
peers_from_control_nodes:
description:
- If enabled, control plane nodes will automatically peer to this node.
required: False
type: bool
extends_documentation_fragment: awx.awx.auth
'''
@@ -83,12 +75,24 @@ EXAMPLES = '''
awx.awx.instance:
hostname: my-instance.prod.example.com
capacity_adjustment: 0.4
listener_port: 31337
- name: Deprovision the instance
awx.awx.instance:
hostname: my-instance.prod.example.com
node_state: deprovisioning
- name: Create execution node
awx.awx.instance:
hostname: execution.example.com
node_type: execution
peers:
- 12
- route.to.hop.example.com
- name: Remove peers
awx.awx.instance:
hostname: execution.example.com
peers:
'''
from ..module_utils.controller_api import ControllerAPIModule
@@ -103,9 +107,7 @@ def main():
managed_by_policy=dict(type='bool'),
node_type=dict(type='str', choices=['execution', 'hop']),
node_state=dict(type='str', choices=['deprovisioning', 'installed']),
listener_port=dict(type='int'),
peers=dict(required=False, type='list', elements='str'),
peers_from_control_nodes=dict(required=False, type='bool'),
)
# Create a module for ourselves
@@ -118,12 +120,21 @@ def main():
managed_by_policy = module.params.get('managed_by_policy')
node_type = module.params.get('node_type')
node_state = module.params.get('node_state')
listener_port = module.params.get('listener_port')
peers = module.params.get('peers')
peers_from_control_nodes = module.params.get('peers_from_control_nodes')
# Attempt to look up an existing item based on the provided data
existing_item = module.get_one('instances', name_or_id=hostname)
# peer item can be an id or address
# if address, get the id
peers_ids = []
if peers:
for p in peers:
if not p.isdigit():
p_id = module.get_one('receptor_addresses', allow_none=False, data={'address': p})
peers_ids.append(p_id['id'])
else:
peers_ids.append(p)
# Create the data that gets sent for create and update
new_fields = {'hostname': hostname}
if capacity_adjustment is not None:
@@ -136,12 +147,8 @@ def main():
new_fields['node_type'] = node_type
if node_state is not None:
new_fields['node_state'] = node_state
if listener_port is not None:
new_fields['listener_port'] = listener_port
if peers is not None:
new_fields['peers'] = peers
if peers_from_control_nodes is not None:
new_fields['peers_from_control_nodes'] = peers_from_control_nodes
new_fields['peers'] = peers_ids
module.create_or_update_if_needed(
existing_item,

View File

@@ -0,0 +1,136 @@
#!/usr/bin/python
# coding: utf-8 -*-
# (c) 2023 Red Hat, Inc.
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
from __future__ import absolute_import, division, print_function
__metaclass__ = type
ANSIBLE_METADATA = {'metadata_version': '1.1', 'status': ['preview'], 'supported_by': 'community'}
DOCUMENTATION = '''
---
module: receptor_address
author: "Seth Foster (@fosterseth)"
version_added: "4.5.0"
short_description: create, update, or destroy Automation Platform Controller receptor addresses.
description:
- Create, update, or destroy Automation Platform Controller receptor addresses. See
U(https://www.ansible.com/tower) for an overview.
options:
address:
description:
- Routable address for this instance.
required: True
type: str
instance:
description:
- ID or hostname of instance this address belongs to.
required: True
type: str
peers_from_control_nodes:
description:
- If True, control plane cluster nodes should automatically peer to it.
required: False
type: bool
port:
description:
- Port for the address.
required: False
type: int
protocol:
description:
- Protocol to use when connecting.
required: False
type: str
choices: [ 'tcp', 'ws', 'wss' ]
websocket_path:
description:
- Websocket path.
required: False
type: str
state:
description:
- Desired state of the resource.
choices: ["present", "absent", "exists"]
default: "present"
type: str
extends_documentation_fragment: awx.awx.auth
'''
EXAMPLES = '''
- name: Create receptor address
awx.awx.receptor_address:
address: exec1addr
instance: exec1.example.com
peers_from_control_nodes: false
port: 6791
protocol: ws
websocket_path: service
state: present
register: exec1addr
'''
from ..module_utils.controller_api import ControllerAPIModule
def main():
# Any additional arguments that are not fields of the item can be added here
argument_spec = dict(
address=dict(required=True, type='str'),
instance=dict(type='str'),
peers_from_control_nodes=dict(type='bool'),
port=dict(type='int'),
protocol=dict(type='str'),
websocket_path=dict(type='str'),
state=dict(choices=['present', 'absent', 'exists'], default='present'),
)
# Create a module for ourselves
module = ControllerAPIModule(argument_spec=argument_spec)
# Extract our parameters
address = module.params.get('address')
peers_from_control_nodes = module.params.get('peers_from_control_nodes')
port = module.params.get('port')
protocol = module.params.get('protocol', 'tcp')
websocket_path = module.params.get('websocket_path')
instance_name_or_id = module.params.get('instance')
state = module.params.get('state')
# Attempt to look up an existing instance
receptor_address = module.get_one('receptor_addresses', allow_none=True, data=dict(address=address, protocol=protocol))
if receptor_address:
receptor_address['type'] = 'receptor_address'
if receptor_address and state == 'absent':
module.delete_if_needed(receptor_address)
instance = module.get_one('instances', allow_none=False, name_or_id=instance_name_or_id)
# Create the data that gets sent for create and update
new_fields = {'instance': instance['id'], 'address': address}
if port:
new_fields['port'] = port
if protocol:
new_fields['protocol'] = protocol
if peers_from_control_nodes:
new_fields['peers_from_control_nodes'] = peers_from_control_nodes
if websocket_path:
new_fields['websocket_path'] = websocket_path
module.create_or_update_if_needed(
receptor_address,
new_fields,
endpoint='receptor_addresses',
item_type='receptor_address',
)
if __name__ == '__main__':
main()