TEngine 6

UnityProject/Packages/YooAsset/Runtime/ResourceManager/Operation/UnloadAllAssetsOperation.cs

@@ -0,0 +1,132 @@
+using System;
+using UnityEngine;
+
+namespace YooAsset
+{
+    public sealed class UnloadAllAssetsOptions
+    {
+        /// <summary>
+        /// 释放所有资源句柄，防止卸载过程中触发完成回调！
+        /// </summary>
+        public bool ReleaseAllHandles = false;
+
+        /// <summary>
+        /// 卸载过程中锁定加载操作，防止新的任务请求！
+        /// </summary>
+        public bool LockLoadOperation = false;
+    }
+
+    public sealed class UnloadAllAssetsOperation : AsyncOperationBase
+    {
+        private enum ESteps
+        {
+            None,
+            CheckOptions,
+            ReleaseAll,
+            AbortDownload,
+            CheckLoading,
+            DestroyAll,
+            Done,
+        }
+
+        private readonly ResourceManager _resManager;
+        private readonly UnloadAllAssetsOptions _options;
+        private ESteps _steps = ESteps.None;
+
+        internal UnloadAllAssetsOperation(ResourceManager resourceManager, UnloadAllAssetsOptions options)
+        {
+            _resManager = resourceManager;
+            _options = options;
+        }
+        internal override void InternalStart()
+        {
+            _steps = ESteps.CheckOptions;
+        }
+        internal override void InternalUpdate()
+        {
+            if (_steps == ESteps.None || _steps == ESteps.Done)
+                return;
+
+            if (_steps == ESteps.CheckOptions)
+            {
+                if (_options == null)
+                {
+                    _steps = ESteps.Done;
+                    Status = EOperationStatus.Failed;
+                    Error = $"{nameof(UnloadAllAssetsOptions)} is null.";
+                    return;
+                }
+
+                // 设置锁定状态
+                if (_options.LockLoadOperation)
+                    _resManager.LockLoadOperation = true;
+
+                _steps = ESteps.ReleaseAll;
+            }
+
+            if (_steps == ESteps.ReleaseAll)
+            {
+                // 清空所有场景句柄
+                _resManager.SceneHandles.Clear();
+
+                // 释放所有资源句柄
+                if (_options.ReleaseAllHandles)
+                {
+                    foreach (var provider in _resManager.ProviderDic.Values)
+                    {
+                        provider.ReleaseAllHandles();
+                    }
+                }
+
+                _steps = ESteps.AbortDownload;
+            }
+
+            if (_steps == ESteps.AbortDownload)
+            {
+                // 注意：终止所有下载任务
+                foreach (var loader in _resManager.LoaderDic.Values)
+                {
+                    loader.AbortOperation();
+                }
+                _steps = ESteps.CheckLoading;
+            }
+
+            if (_steps == ESteps.CheckLoading)
+            {
+                // 注意：等待所有任务完成
+                foreach (var provider in _resManager.ProviderDic.Values)
+                {
+                    if (provider.IsDone == false)
+                        return;
+                }
+                _steps = ESteps.DestroyAll;
+            }
+
+            if (_steps == ESteps.DestroyAll)
+            {
+                // 强制销毁资源提供者
+                foreach (var provider in _resManager.ProviderDic.Values)
+                {
+                    provider.DestroyProvider();
+                }
+
+                // 强制销毁文件加载器
+                foreach (var loader in _resManager.LoaderDic.Values)
+                {
+                    loader.DestroyLoader();
+                }
+
+                // 清空数据
+                _resManager.ProviderDic.Clear();
+                _resManager.LoaderDic.Clear();
+                _resManager.LockLoadOperation = false;
+
+                // 注意：调用底层接口释放所有资源
+                Resources.UnloadUnusedAssets();
+
+                _steps = ESteps.Done;
+                Status = EOperationStatus.Succeed;
+            }
+        }
+    }
+}