94 lines
4.1 KiB
YAML
94 lines
4.1 KiB
YAML
---
|
|
- name: "STRESS-SETUP: Manage download simulation"
|
|
hosts: workers
|
|
gather_facts: no
|
|
vars:
|
|
tmux_session_download: "stress-download-{{ (profile_prefix | default('default')) | replace(',', '-') }}"
|
|
download_policy: "policies/11_direct_docker_download_simulation.yaml"
|
|
vars_files:
|
|
- "group_vars/all/vault.yml"
|
|
pre_tasks:
|
|
- name: Set inventory_env fact
|
|
ansible.builtin.set_fact:
|
|
inventory_env: "{{ inventory_file | basename | splitext | first | replace('inventory.', '') }}"
|
|
- name: Load environment-specific variables
|
|
ansible.builtin.include_vars: "{{ item }}"
|
|
with_fileglob:
|
|
- "group_vars/all/generated_vars{{ '.' + inventory_env if inventory_env else '' }}.yml"
|
|
tasks:
|
|
- name: Define base directory for node
|
|
ansible.builtin.set_fact:
|
|
base_dir: "{{ airflow_worker_dir }}"
|
|
|
|
- name: Validate profile_prefix is provided when starting or stopping
|
|
ansible.builtin.fail:
|
|
msg: "profile_prefix is required when start_download=true or stop_download=true"
|
|
when:
|
|
- (start_download | default(false) | bool or stop_download | default(false) | bool)
|
|
- profile_prefix is not defined
|
|
|
|
- name: "Display policy being used for download simulation"
|
|
ansible.builtin.debug:
|
|
msg: "Using download simulation policy: {{ download_policy }}"
|
|
when: start_download | default(false) | bool
|
|
|
|
- name: Manage download simulation process
|
|
ansible.builtin.include_tasks:
|
|
file: manage-processes-tasks.yml
|
|
vars:
|
|
tmux_session_name: "{{ tmux_session_download }}"
|
|
working_dir: "{{ base_dir }}"
|
|
command_to_run: >
|
|
./bin/ytops-client stress-policy
|
|
--policy {{ download_policy }}
|
|
{% if dummy_batch | default(true) | bool %}--dummy-batch{% endif %}
|
|
{% if download_min_seconds is defined %}--set 'settings.dummy_simulation_settings.download_min_seconds={{ download_min_seconds }}'{% endif %}
|
|
{% if download_max_seconds is defined %}--set 'settings.dummy_simulation_settings.download_max_seconds={{ download_max_seconds }}'{% endif %}
|
|
{% if profile_prefix is defined %}--set 'execution_control.worker_pools=[{"profile_prefix": "{{ profile_prefix }}", "workers": 1}]'{% endif %}
|
|
{% for setting in (extra_set_args | default('[]')) | from_yaml %}--set '{{ setting }}' {% endfor %}
|
|
--profile-prefix {{ profile_prefix }}
|
|
process_grep_pattern: "ytops-client.*stress-policy.*--policy {{ download_policy }}.*--profile-prefix {{ profile_prefix }}"
|
|
start_process: "{{ start_download | default(false) | bool }}"
|
|
stop_process: "{{ stop_download | default(false) | bool }}"
|
|
check_status: "{{ vars.check_status | default(false) | bool }}"
|
|
|
|
- name: List active tmux sessions
|
|
ansible.builtin.shell:
|
|
cmd: tmux list-sessions 2>/dev/null || true
|
|
register: tmux_sessions
|
|
changed_when: false
|
|
|
|
- name: Display active sessions
|
|
ansible.builtin.debug:
|
|
msg: "Active tmux sessions on {{ inventory_hostname }}: {{ tmux_sessions.stdout_lines }}"
|
|
|
|
- name: Check tmux session output for errors
|
|
block:
|
|
- name: Wait for a moment for the process to start
|
|
ansible.builtin.pause:
|
|
seconds: 2
|
|
|
|
- name: Capture tmux pane content
|
|
ansible.builtin.shell:
|
|
cmd: "tmux capture-pane -p -t {{ tmux_session_download }}"
|
|
register: tmux_output
|
|
changed_when: false
|
|
ignore_errors: true
|
|
|
|
- name: Display tmux pane content if session exists
|
|
ansible.builtin.debug:
|
|
msg: "Initial output from tmux session '{{ tmux_session_download }}':"
|
|
when: tmux_output.rc == 0
|
|
|
|
- name: Show output lines if session exists
|
|
ansible.builtin.debug:
|
|
var: tmux_output.stdout_lines
|
|
when: tmux_output.rc == 0
|
|
|
|
- name: Report if session not found
|
|
ansible.builtin.debug:
|
|
msg: "Tmux session '{{ tmux_session_download }}' was not found. It may have exited immediately upon starting."
|
|
when: tmux_output.rc != 0
|
|
|
|
when: start_download | default(false) | bool
|