From 9827deffd3a0d3ae7c4d725cccfe132092ef745c Mon Sep 17 00:00:00 2001
From: Cody Henthorne <cody@signal.org>
Date: Thu, 4 Nov 2021 15:18:14 -0400
Subject: [PATCH] Make websocket timeouts stay on IO threads.

---
 .../signalservice/internal/websocket/WebSocketConnection.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libsignal/service/src/main/java/org/whispersystems/signalservice/internal/websocket/WebSocketConnection.java b/libsignal/service/src/main/java/org/whispersystems/signalservice/internal/websocket/WebSocketConnection.java
index b43f78637..61df211a0 100644
--- a/libsignal/service/src/main/java/org/whispersystems/signalservice/internal/websocket/WebSocketConnection.java
+++ b/libsignal/service/src/main/java/org/whispersystems/signalservice/internal/websocket/WebSocketConnection.java
@@ -208,7 +208,7 @@ public class WebSocketConnection extends WebSocketListener {
 
     return single.subscribeOn(Schedulers.io())
                  .observeOn(Schedulers.io())
-                 .timeout(10, TimeUnit.SECONDS);
+                 .timeout(10, TimeUnit.SECONDS, Schedulers.io());
   }
 
   public synchronized void sendResponse(WebSocketResponseMessage response) throws IOException {