Spring Boot настройка таймаута для возвращения ошибки 504

Приложение на Spring Boot У меня контроллер должен принимать запрос с параметром wait. Затем контроллер обращается к другому серверу за данными и, если сервер в течении времени wait не ответил, то контроллер должен вернуть статус "Gateway Timeout" с кодом 504. Подскажите, пожалуйста, как это можно реализовать?