From e54e02fdcf60abebdd9944df58544575acabfce5 Mon Sep 17 00:00:00 2001
From: aquilescanta <aquilescanta@google.com>
Date: Thu, 23 Dec 2021 17:40:14 +0000
Subject: [PATCH] Pre-allocate availableAllocations to prevent a re-size in
 release

PiperOrigin-RevId: 418022431
---
 .../exoplayer2/upstream/DefaultAllocator.java        | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/library/core/src/main/java/com/google/android/exoplayer2/upstream/DefaultAllocator.java b/library/core/src/main/java/com/google/android/exoplayer2/upstream/DefaultAllocator.java
index b79b66c930..47a77beb38 100644
--- a/library/core/src/main/java/com/google/android/exoplayer2/upstream/DefaultAllocator.java
+++ b/library/core/src/main/java/com/google/android/exoplayer2/upstream/DefaultAllocator.java
@@ -102,6 +102,12 @@ public final class DefaultAllocator implements Allocator {
       availableAllocations[availableCount] = null;
     } else {
       allocation = new Allocation(new byte[individualAllocationSize], 0);
+      if (allocatedCount > availableAllocations.length) {
+        // Make availableAllocations be large enough to contain all allocations made by this
+        // allocator so that release() does not need to grow the availableAllocations array. See
+        // [Internal ref: b/209801945].
+        availableAllocations = Arrays.copyOf(availableAllocations, availableAllocations.length * 2);
+      }
     }
     return allocation;
   }
@@ -114,12 +120,6 @@ public final class DefaultAllocator implements Allocator {
 
   @Override
   public synchronized void release(Allocation[] allocations) {
-    if (availableCount + allocations.length >= availableAllocations.length) {
-      availableAllocations =
-          Arrays.copyOf(
-              availableAllocations,
-              max(availableAllocations.length * 2, availableCount + allocations.length));
-    }
     for (Allocation allocation : allocations) {
       availableAllocations[availableCount++] = allocation;
     }