304 lines
8.0 KiB
C#
304 lines
8.0 KiB
C#
using System;
|
||
using System.IO;
|
||
using System.Collections;
|
||
using System.Collections.Generic;
|
||
using UnityEngine;
|
||
|
||
namespace YooAsset
|
||
{
|
||
internal sealed class AssetBundleFileLoader : AssetBundleLoaderBase
|
||
{
|
||
private enum ESteps
|
||
{
|
||
None = 0,
|
||
Download,
|
||
CheckDownload,
|
||
Unpack,
|
||
CheckUnpack,
|
||
LoadFile,
|
||
CheckLoadFile,
|
||
Done,
|
||
}
|
||
|
||
private ESteps _steps = ESteps.None;
|
||
private string _fileLoadPath;
|
||
private bool _isWaitForAsyncComplete = false;
|
||
private bool _isShowWaitForAsyncError = false;
|
||
private DownloaderBase _unpacker;
|
||
private DownloaderBase _downloader;
|
||
private AssetBundleCreateRequest _createRequest;
|
||
private FileStream _fileStream;
|
||
|
||
|
||
public AssetBundleFileLoader(AssetSystemImpl impl, BundleInfo bundleInfo) : base(impl, bundleInfo)
|
||
{
|
||
}
|
||
|
||
/// <summary>
|
||
/// 轮询更新
|
||
/// </summary>
|
||
public override void Update()
|
||
{
|
||
if (_steps == ESteps.Done)
|
||
return;
|
||
|
||
if (_steps == ESteps.None)
|
||
{
|
||
if (MainBundleInfo.LoadMode == BundleInfo.ELoadMode.LoadFromRemote)
|
||
{
|
||
_steps = ESteps.Download;
|
||
_fileLoadPath = MainBundleInfo.Bundle.CachedFilePath;
|
||
}
|
||
else if (MainBundleInfo.LoadMode == BundleInfo.ELoadMode.LoadFromStreaming)
|
||
{
|
||
#if UNITY_ANDROID
|
||
EBundleLoadMethod loadMethod = (EBundleLoadMethod)MainBundleInfo.Bundle.LoadMethod;
|
||
if (loadMethod == EBundleLoadMethod.LoadFromMemory || loadMethod == EBundleLoadMethod.LoadFromStream)
|
||
{
|
||
_steps = ESteps.Unpack;
|
||
_fileLoadPath = MainBundleInfo.Bundle.CachedFilePath;
|
||
}
|
||
else
|
||
{
|
||
_steps = ESteps.LoadFile;
|
||
_fileLoadPath = MainBundleInfo.Bundle.StreamingFilePath;
|
||
}
|
||
#else
|
||
_steps = ESteps.LoadFile;
|
||
_fileLoadPath = MainBundleInfo.Bundle.StreamingFilePath;
|
||
#endif
|
||
}
|
||
else if (MainBundleInfo.LoadMode == BundleInfo.ELoadMode.LoadFromCache)
|
||
{
|
||
_steps = ESteps.LoadFile;
|
||
_fileLoadPath = MainBundleInfo.Bundle.CachedFilePath;
|
||
}
|
||
else
|
||
{
|
||
throw new System.NotImplementedException(MainBundleInfo.LoadMode.ToString());
|
||
}
|
||
}
|
||
|
||
// 1. 从服务器下载
|
||
if (_steps == ESteps.Download)
|
||
{
|
||
int failedTryAgain = int.MaxValue;
|
||
_downloader = DownloadSystem.BeginDownload(MainBundleInfo, failedTryAgain);
|
||
_steps = ESteps.CheckDownload;
|
||
}
|
||
|
||
// 2. 检测服务器下载结果
|
||
if (_steps == ESteps.CheckDownload)
|
||
{
|
||
if (_downloader.IsDone() == false)
|
||
return;
|
||
|
||
if (_downloader.HasError())
|
||
{
|
||
_steps = ESteps.Done;
|
||
Status = EStatus.Failed;
|
||
LastError = _downloader.GetLastError();
|
||
}
|
||
else
|
||
{
|
||
_steps = ESteps.LoadFile;
|
||
}
|
||
}
|
||
|
||
// 3. 内置文件解压
|
||
if (_steps == ESteps.Unpack)
|
||
{
|
||
int failedTryAgain = 1;
|
||
var bundleInfo = HostPlayModeImpl.ConvertToUnpackInfo(MainBundleInfo.Bundle);
|
||
_unpacker = DownloadSystem.BeginDownload(bundleInfo, failedTryAgain);
|
||
_steps = ESteps.CheckUnpack;
|
||
}
|
||
|
||
// 4.检测内置文件解压结果
|
||
if (_steps == ESteps.CheckUnpack)
|
||
{
|
||
if (_unpacker.IsDone() == false)
|
||
return;
|
||
|
||
if (_unpacker.HasError())
|
||
{
|
||
_steps = ESteps.Done;
|
||
Status = EStatus.Failed;
|
||
LastError = _unpacker.GetLastError();
|
||
}
|
||
else
|
||
{
|
||
_steps = ESteps.LoadFile;
|
||
}
|
||
}
|
||
|
||
// 5. 加载AssetBundle
|
||
if (_steps == ESteps.LoadFile)
|
||
{
|
||
#if UNITY_EDITOR
|
||
// 注意：Unity2017.4编辑器模式下，如果AssetBundle文件不存在会导致编辑器崩溃，这里做了预判。
|
||
if (System.IO.File.Exists(_fileLoadPath) == false)
|
||
{
|
||
_steps = ESteps.Done;
|
||
Status = EStatus.Failed;
|
||
LastError = $"Not found assetBundle file : {_fileLoadPath}";
|
||
YooLogger.Error(LastError);
|
||
return;
|
||
}
|
||
#endif
|
||
|
||
// Load assetBundle file
|
||
var loadMethod = (EBundleLoadMethod)MainBundleInfo.Bundle.LoadMethod;
|
||
if (loadMethod == EBundleLoadMethod.Normal)
|
||
{
|
||
if (_isWaitForAsyncComplete)
|
||
CacheBundle = AssetBundle.LoadFromFile(_fileLoadPath);
|
||
else
|
||
_createRequest = AssetBundle.LoadFromFileAsync(_fileLoadPath);
|
||
}
|
||
else
|
||
{
|
||
if (Impl.DecryptionServices == null)
|
||
{
|
||
_steps = ESteps.Done;
|
||
Status = EStatus.Failed;
|
||
LastError = $"{nameof(IDecryptionServices)} is null : {MainBundleInfo.Bundle.BundleName}";
|
||
YooLogger.Error(LastError);
|
||
return;
|
||
}
|
||
|
||
DecryptFileInfo fileInfo = new DecryptFileInfo();
|
||
fileInfo.BundleName = MainBundleInfo.Bundle.BundleName;
|
||
fileInfo.FilePath = _fileLoadPath;
|
||
|
||
if (loadMethod == EBundleLoadMethod.LoadFromFileOffset)
|
||
{
|
||
ulong offset = Impl.DecryptionServices.LoadFromFileOffset(fileInfo);
|
||
if (_isWaitForAsyncComplete)
|
||
CacheBundle = AssetBundle.LoadFromFile(_fileLoadPath, 0, offset);
|
||
else
|
||
_createRequest = AssetBundle.LoadFromFileAsync(_fileLoadPath, 0, offset);
|
||
}
|
||
else if (loadMethod == EBundleLoadMethod.LoadFromMemory)
|
||
{
|
||
byte[] fileData = Impl.DecryptionServices.LoadFromMemory(fileInfo);
|
||
if (_isWaitForAsyncComplete)
|
||
CacheBundle = AssetBundle.LoadFromMemory(fileData);
|
||
else
|
||
_createRequest = AssetBundle.LoadFromMemoryAsync(fileData);
|
||
}
|
||
else if (loadMethod == EBundleLoadMethod.LoadFromStream)
|
||
{
|
||
_fileStream = Impl.DecryptionServices.LoadFromStream(fileInfo);
|
||
uint managedReadBufferSize = Impl.DecryptionServices.GetManagedReadBufferSize();
|
||
if (_isWaitForAsyncComplete)
|
||
CacheBundle = AssetBundle.LoadFromStream(_fileStream, 0, managedReadBufferSize);
|
||
else
|
||
_createRequest = AssetBundle.LoadFromStreamAsync(_fileStream, 0, managedReadBufferSize);
|
||
}
|
||
else
|
||
{
|
||
throw new System.NotImplementedException();
|
||
}
|
||
}
|
||
_steps = ESteps.CheckLoadFile;
|
||
}
|
||
|
||
// 6. 检测AssetBundle加载结果
|
||
if (_steps == ESteps.CheckLoadFile)
|
||
{
|
||
if (_createRequest != null)
|
||
{
|
||
if (_isWaitForAsyncComplete)
|
||
{
|
||
// 强制挂起主线程（注意：该操作会很耗时）
|
||
YooLogger.Warning("Suspend the main thread to load unity bundle.");
|
||
CacheBundle = _createRequest.assetBundle;
|
||
}
|
||
else
|
||
{
|
||
if (_createRequest.isDone == false)
|
||
return;
|
||
CacheBundle = _createRequest.assetBundle;
|
||
}
|
||
}
|
||
|
||
// Check error
|
||
if (CacheBundle == null)
|
||
{
|
||
_steps = ESteps.Done;
|
||
Status = EStatus.Failed;
|
||
LastError = $"Failed to load assetBundle : {MainBundleInfo.Bundle.BundleName}";
|
||
YooLogger.Error(LastError);
|
||
|
||
// 注意：当缓存文件的校验等级为Low的时候，并不能保证缓存文件的完整性。
|
||
// 在AssetBundle文件加载失败的情况下，我们需要重新验证文件的完整性！
|
||
if (MainBundleInfo.LoadMode == BundleInfo.ELoadMode.LoadFromCache)
|
||
{
|
||
string cacheLoadPath = MainBundleInfo.Bundle.CachedFilePath;
|
||
if (CacheSystem.VerifyBundle(MainBundleInfo.Bundle, EVerifyLevel.High) != EVerifyResult.Succeed)
|
||
{
|
||
if (File.Exists(cacheLoadPath))
|
||
{
|
||
YooLogger.Error($"Delete the invalid cache file : {cacheLoadPath}");
|
||
File.Delete(cacheLoadPath);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
else
|
||
{
|
||
_steps = ESteps.Done;
|
||
Status = EStatus.Succeed;
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 销毁
|
||
/// </summary>
|
||
public override void Destroy(bool forceDestroy)
|
||
{
|
||
base.Destroy(forceDestroy);
|
||
|
||
if (_fileStream != null)
|
||
{
|
||
_fileStream.Close();
|
||
_fileStream.Dispose();
|
||
_fileStream = null;
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 主线程等待异步操作完毕
|
||
/// </summary>
|
||
public override void WaitForAsyncComplete()
|
||
{
|
||
_isWaitForAsyncComplete = true;
|
||
|
||
int frame = 1000;
|
||
while (true)
|
||
{
|
||
// 保险机制
|
||
// 注意：如果需要从WEB端下载资源，可能会触发保险机制！
|
||
frame--;
|
||
if (frame == 0)
|
||
{
|
||
if (_isShowWaitForAsyncError == false)
|
||
{
|
||
_isShowWaitForAsyncError = true;
|
||
YooLogger.Error($"WaitForAsyncComplete failed ! Try load bundle : {MainBundleInfo.Bundle.BundleName} from remote with sync load method !");
|
||
}
|
||
break;
|
||
}
|
||
|
||
// 驱动流程
|
||
Update();
|
||
|
||
// 完成后退出
|
||
if (IsDone())
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
} |