matrix-docker-ansible-deploy/roles/matrix-synapse/tasks/rust-synapse-compress-state/main.yml

---
# Pre-checks

- name: Fail if Postgres not enabled
  fail:
    msg: "Postgres via the matrix-postgres role is not enabled (`matrix_postgres_enabled`). Cannot use rust-synapse-compress-state."
  when: "not matrix_postgres_enabled|bool"


# Defaults

- name: Set matrix_synapse_rust_synapse_compress_state_find_rooms_command_wait_time, if not provided
  set_fact:
    matrix_synapse_rust_synapse_compress_state_find_rooms_command_wait_time: 300
  when: "matrix_synapse_rust_synapse_compress_state_find_rooms_command_wait_time|default('') == ''"

- name: Set matrix_synapse_rust_synapse_compress_state_compress_room_time, if not provided
  set_fact:
    matrix_synapse_rust_synapse_compress_state_compress_room_time: 1800
  when: "matrix_synapse_rust_synapse_compress_state_compress_room_time|default('') == ''"

- name: Set matrix_synapse_rust_synapse_compress_state_psql_import_time, if not provided
  set_fact:
    matrix_synapse_rust_synapse_compress_state_psql_import_time: 1800
  when: "matrix_synapse_rust_synapse_compress_state_psql_import_time|default('') == ''"

- name: Set matrix_synapse_rust_synapse_compress_state_min_state_groups_required, if not provided
  set_fact:
    # The minimum number of state groups we're looking for before we consider a room eligible for compression.
    # Rooms with a smaller state groups count will not be compressed.
    matrix_synapse_rust_synapse_compress_state_min_state_groups_required: 100000
  when: "matrix_synapse_rust_synapse_compress_state_min_state_groups_required|default('') == ''"


# Actual compression work

- name: Ensure rust-synapse-compress-state paths exist
  file:
    path: "{{ matrix_synapse_rust_synapse_compress_state_base_path }}"
    state: directory
    mode: 0750
    owner: "{{ matrix_user_username }}"
    group: "{{ matrix_user_groupname }}"

- name: Ensure rust-synapse-compress-state image is pulled
  docker_image:
    name: "{{ matrix_synapse_rust_synapse_compress_state_docker_image }}"
    source: "{{ 'pull' if ansible_version.major > 2 or ansible_version.minor > 7 else omit }}"
    force_source: "{{ matrix_synapse_rust_synapse_compress_state_docker_image_force_pull if ansible_version.major > 2 or ansible_version.minor >= 8 else omit }}"
    force: "{{ omit if ansible_version.major > 2 or ansible_version.minor >= 8 else matrix_synapse_rust_synapse_compress_state_docker_image_force_pull }}"
  register: result
  retries: "{{ matrix_container_retries_count }}"
  delay: "{{ matrix_container_retries_delay }}"
  until: result is not failed

- name: Generate rust-synapse-compress-state room find command
  set_fact:
    matrix_synapse_rust_synapse_compress_state_find_rooms_command: >-
      {{ matrix_host_command_docker }} run --rm --name matrix-rust-synapse-compress-state-find-rooms
      --user={{ matrix_user_uid }}:{{ matrix_user_gid }}
      --cap-drop=ALL
      --network={{ matrix_docker_network }}
      --env-file={{ matrix_postgres_base_path }}/env-postgres-psql
      {{ matrix_postgres_docker_image_latest }}
      psql -v ON_ERROR_STOP=1 -h matrix-postgres {{ matrix_synapse_database_database }} -c
      'SELECT array_to_json(array_agg(row_to_json (r))) FROM (SELECT room_id, count(*) AS count FROM state_groups_state GROUP BY room_id HAVING count(*) > {{ matrix_synapse_rust_synapse_compress_state_min_state_groups_required }} ORDER BY count DESC) r;'      

- name: Find rooms eligible for compression with rust-synapse-compress-state
  command: "{{ matrix_synapse_rust_synapse_compress_state_find_rooms_command }}"
  async: "{{ matrix_synapse_rust_synapse_compress_state_find_rooms_command_wait_time }}"
  poll: 10
  register: matrix_synapse_rust_synapse_compress_state_find_rooms_command_result

# We expect the output to be like this:
#
# "stdout_lines": [
#     "                                                       array_to_json                                                        ",
#     "----------------------------------------------------------------------------------------------------------------------------",
#     " [{\"room_id\":\"!some-id\",\"count\":2461329},{\"room_id\":\"!another-id\",\"count\":512017}]",
#     "(1 row)"
# ]
#
# Row 3 (out of 4) contains the actual result.
#
# Row 3 contains a space when there's no result.

- block:
    - debug: var="matrix_synapse_rust_synapse_compress_state_find_rooms_command_result"

    - name: Fail if room find result is not what we expect
      fail:
        msg: >-
          Expecting 4 lines in the "find rooms" result.          
  when: "matrix_synapse_rust_synapse_compress_state_find_rooms_command_result.failed or matrix_synapse_rust_synapse_compress_state_find_rooms_command_result.stdout_lines|length != 4"

- block:
    # matrix_synapse_rust_synapse_compress_state_eligible_rooms is a list
    # of dictionaries like this: {'room_id': '!some-id', 'count': 2461329}
    - set_fact:
        matrix_synapse_rust_synapse_compress_state_eligible_rooms: "{{ matrix_synapse_rust_synapse_compress_state_find_rooms_command_result.stdout_lines[2] | from_json }}"

    - name: Display rooms that will be compressed
      debug:
        msg: >-
          The following rooms contain more than {{ matrix_synapse_rust_synapse_compress_state_min_state_groups_required }} state group rows
          (configurable via `matrix_synapse_rust_synapse_compress_state_min_state_groups_required`)
          and will be compressed:
          {{ matrix_synapse_rust_synapse_compress_state_eligible_rooms }}          

    - name: Compress room state
      include_tasks: "{{ role_path }}/tasks/rust-synapse-compress-state/compress_room.yml"
      with_items: "{{ matrix_synapse_rust_synapse_compress_state_eligible_rooms }}"
      loop_control:
        loop_var: room_details
  when: "matrix_synapse_rust_synapse_compress_state_find_rooms_command_result.stdout_lines[2] != ' '"

- name: Show notice about lack of rooms to compress
  debug:
    msg: >-
      No rooms were found to contain more than {{ matrix_synapse_rust_synapse_compress_state_min_state_groups_required }} state group rows
      (configurable via `matrix_synapse_rust_synapse_compress_state_min_state_groups_required`),
      so there's nothing to compress.      
  when: "matrix_synapse_rust_synapse_compress_state_find_rooms_command_result.stdout_lines[2] == ' '"