ansible tasks to reboot server and wait it come back

reboot.yml

- name: restart server if ulimit changed
  become: yes
  shell: sleep 2 && reboot
  async: 1
  poll: 0
  ignore_errors: yes

- name: wait for server to come back
  local_action:
    module: wait_for
    host: "{{ansible_host}}"
    port: 22  # have to provide and harcode it
    delay: 30
    timeout: 120
    sleep: 2
    state: started