mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-11 21:26:44 +00:00
kevinveenbirkenbachandClaude Opus 5.5
afaade9bbd
Each vhost used to run its own sequential reachability probe. The role now queues the vhost and deferred_probe.yml checks the whole queue with one uri_probe_batch call after the shared-service preload and after the server roles. A vhost answering like a stale config restarts OpenResty once; one still unserved afterwards fails the deploy by name. uri_probe_batch takes follow_redirects explicitly so the probe sees the redirect status itself. Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
91 lines
3.0 KiB
Python
91 lines
3.0 KiB
Python
"""Probe many URLs from the managed host in one module call.
|
|
|
|
Args:
|
|
urls: mapping of a name to the URL probed for it.
|
|
method: HTTP method of every request.
|
|
status_codes: statuses that count as an answer; any other is retried.
|
|
attempts: requests per URL before it counts as unanswered.
|
|
delay: seconds between two attempts at one URL.
|
|
timeout: seconds per request.
|
|
validate_certs: whether TLS certificates are verified.
|
|
follow_redirects: how a redirect is handled, as open_url takes it; with
|
|
``none`` the redirect status itself is the answer.
|
|
workers: requests in flight at once.
|
|
|
|
Returns:
|
|
status: mapping of every name to its last HTTP status, -1 without a response.
|
|
"""
|
|
|
|
import time
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
from urllib.error import HTTPError, URLError
|
|
|
|
from ansible.module_utils.basic import AnsibleModule
|
|
from ansible.module_utils.urls import open_url
|
|
|
|
ARGUMENT_SPEC = {
|
|
"urls": {"type": "dict", "required": True},
|
|
"method": {"type": "str", "required": True},
|
|
"status_codes": {"type": "list", "elements": "int", "required": True},
|
|
"attempts": {"type": "int", "required": True},
|
|
"delay": {"type": "int", "required": True},
|
|
"timeout": {"type": "int", "required": True},
|
|
"validate_certs": {"type": "bool", "required": True},
|
|
"follow_redirects": {
|
|
"type": "str",
|
|
"required": True,
|
|
"choices": ["all", "none", "safe", "urllib2"],
|
|
},
|
|
"workers": {"type": "int", "required": True},
|
|
}
|
|
|
|
|
|
def probe(url, params):
|
|
status = -1
|
|
for attempt in range(params["attempts"]):
|
|
if attempt:
|
|
time.sleep(params["delay"])
|
|
try:
|
|
with open_url(
|
|
url,
|
|
method=params["method"],
|
|
timeout=params["timeout"],
|
|
validate_certs=params["validate_certs"],
|
|
follow_redirects=params["follow_redirects"],
|
|
) as response:
|
|
status = response.status
|
|
except HTTPError as error:
|
|
status = error.code
|
|
except (URLError, OSError):
|
|
status = -1
|
|
if status in params["status_codes"]:
|
|
break
|
|
return status
|
|
|
|
|
|
def probe_all(urls, params):
|
|
with ThreadPoolExecutor(max_workers=params["workers"]) as pool:
|
|
futures = {name: pool.submit(probe, url, params) for name, url in urls.items()}
|
|
return {name: future.result() for name, future in futures.items()}
|
|
|
|
|
|
def main():
|
|
module = AnsibleModule(argument_spec=ARGUMENT_SPEC, supports_check_mode=True)
|
|
status = probe_all(module.params["urls"], module.params)
|
|
unanswered = sorted(
|
|
name
|
|
for name, code in status.items()
|
|
if code not in module.params["status_codes"]
|
|
)
|
|
if unanswered:
|
|
module.fail_json(
|
|
msg=f"{len(unanswered)} URL(s) gave no expected status",
|
|
unanswered=unanswered,
|
|
status=status,
|
|
)
|
|
module.exit_json(changed=False, status=status)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|