From 495334062b88e478f3d8a79adb19860b8990aa07 Mon Sep 17 00:00:00 2001
From: Kweku Adams <kwekua@google.com>
Date: Fri, 18 Nov 2022 11:22:36 +0000
Subject: [PATCH] Recalculate network estimate after work is completed.

Don't recalculate the network byte numbers when a JobWorkItem has been
dequeued since the work still needs to be done. Instead, wait until the
work has been marked completed to recalculate the network byte numbers.

Bug: 255828754
Test: N/A
Change-Id: I0deb5d35293a8780329696d2c467270e59d32e05
---
 .../java/com/android/server/job/controllers/JobStatus.java      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/apex/jobscheduler/service/java/com/android/server/job/controllers/JobStatus.java b/apex/jobscheduler/service/java/com/android/server/job/controllers/JobStatus.java
index f6410ffbd3ddd..da20684aaefb1 100644
--- a/apex/jobscheduler/service/java/com/android/server/job/controllers/JobStatus.java
+++ b/apex/jobscheduler/service/java/com/android/server/job/controllers/JobStatus.java
@@ -762,7 +762,6 @@ public final class JobStatus {
                 executingWork.add(work);
                 work.bumpDeliveryCount();
             }
-            updateNetworkBytesLocked();
             return work;
         }
         return null;
@@ -790,6 +789,7 @@ public final class JobStatus {
                 if (work.getWorkId() == workId) {
                     executingWork.remove(i);
                     ungrantWorkItem(work);
+                    updateNetworkBytesLocked();
                     return true;
                 }
             }