fix(Load manager): fixed the downloaded bytes onProgress call

awslabs · Aug 14, 2023 · acfded8 · acfded8
1 parent 21bcf3b
commit acfded8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/packages/scene-composer/src/components/three-fiber/hooks/useProgress.spec.tsx b/packages/scene-composer/src/components/three-fiber/hooks/useProgress.spec.tsx
@@ -74,6 +74,7 @@ describe('useProgress', () => {
     // Assert
     expect(setMock).toBeCalledWith({
       active: true,
+      downloaded: 25,
       item,
       loaded,
       total,
@@ -86,6 +87,7 @@ describe('useProgress', () => {
     const item = 'test-url';
     const loaded = 100;
     const total = 100;
+
     const setMock = jest.fn();
     useProgressImpl(setMock);
 
@@ -98,6 +100,7 @@ describe('useProgress', () => {
       item,
       loaded,
       total,
+      downloaded: 0,
       progress: 100,
     });
   });

diff --git a/packages/scene-composer/src/components/three-fiber/hooks/useProgress.tsx b/packages/scene-composer/src/components/three-fiber/hooks/useProgress.tsx
@@ -16,6 +16,7 @@ type Data = {
   downloadProgress: number;
 };
 let saveLastTotalLoaded = 0;
+const downloaded = 0;
 
 export const useProgressImpl = (set) => {
   [DefaultLoadingManager, ...tmLoadingManagers].forEach((manager) => {
@@ -40,6 +41,7 @@ export const useProgressImpl = (set) => {
         active: true,
         item,
         loaded,
+        downloaded: downloaded + (loaded - saveLastTotalLoaded),
         total,
         progress: ((loaded - saveLastTotalLoaded) / (total - saveLastTotalLoaded)) * 100 || 100,
       });