From aa7309cdea0cf51951db583cb166b757fd44ac3b Mon Sep 17 00:00:00 2001
From: krocard <krocard@google.com>
Date: Thu, 24 Sep 2020 15:43:53 +0100
Subject: [PATCH] Release wakelock when sleeping for offload

#exo-offload

PiperOrigin-RevId: 333512383
---
 .../com/google/android/exoplayer2/SimpleExoPlayer.java | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/library/core/src/main/java/com/google/android/exoplayer2/SimpleExoPlayer.java b/library/core/src/main/java/com/google/android/exoplayer2/SimpleExoPlayer.java
index dd41d8e2bc..baa1400143 100644
--- a/library/core/src/main/java/com/google/android/exoplayer2/SimpleExoPlayer.java
+++ b/library/core/src/main/java/com/google/android/exoplayer2/SimpleExoPlayer.java
@@ -2413,5 +2413,15 @@ public class SimpleExoPlayer extends BasePlayer
         boolean playWhenReady, @PlayWhenReadyChangeReason int reason) {
       updateWakeAndWifiLock();
     }
+
+    @Override
+    public void onExperimentalSleepingForOffloadChanged(boolean sleepingForOffload) {
+      if (sleepingForOffload) {
+        // The wifi lock is not released to avoid interrupting downloads.
+        wakeLockManager.setStayAwake(false);
+      } else {
+        updateWakeAndWifiLock();
+      }
+    }
   }
 }