mirror of
https://github.com/Alex-Rachel/TEngine.git
synced 2025-08-14 16:51:28 +00:00
557 lines
21 KiB
C#
557 lines
21 KiB
C#
using System;
|
||
using System.IO;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
|
||
namespace YooAsset
|
||
{
|
||
/// <summary>
|
||
/// 缓存文件系统
|
||
/// 说明：正在进行的下载器会在ResourcePackage销毁的时候执行Abort操作！
|
||
/// </summary>
|
||
internal class DefaultCacheFileSystem : IFileSystem
|
||
{
|
||
protected readonly Dictionary<string, RecordFileElement> _wrappers = new Dictionary<string, RecordFileElement>(10000);
|
||
protected readonly Dictionary<string, string> _bundleDataFilePaths = new Dictionary<string, string>(10000);
|
||
protected readonly Dictionary<string, string> _bundleInfoFilePaths = new Dictionary<string, string>(10000);
|
||
protected readonly Dictionary<string, string> _tempFilePaths = new Dictionary<string, string>(10000);
|
||
protected DefaultCacheDownloadCenter _downloadCenter;
|
||
|
||
protected string _packageRoot;
|
||
protected string _tempFilesRoot;
|
||
protected string _cacheBundleFilesRoot;
|
||
protected string _cacheManifestFilesRoot;
|
||
|
||
/// <summary>
|
||
/// 包裹名称
|
||
/// </summary>
|
||
public string PackageName { private set; get; }
|
||
|
||
/// <summary>
|
||
/// 文件根目录
|
||
/// </summary>
|
||
public string FileRoot
|
||
{
|
||
get
|
||
{
|
||
return _packageRoot;
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 文件数量
|
||
/// </summary>
|
||
public int FileCount
|
||
{
|
||
get
|
||
{
|
||
return _wrappers.Count;
|
||
}
|
||
}
|
||
|
||
#region 自定义参数
|
||
/// <summary>
|
||
/// 自定义参数：远程服务接口
|
||
/// </summary>
|
||
public IRemoteServices RemoteServices { private set; get; } = null;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：初始化的时候缓存文件校验级别
|
||
/// </summary>
|
||
public EFileVerifyLevel FileVerifyLevel { private set; get; } = EFileVerifyLevel.Middle;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：数据文件追加文件格式
|
||
/// </summary>
|
||
public bool AppendFileExtension { private set; get; } = false;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：最大并发连接数
|
||
/// </summary>
|
||
public int DownloadMaxConcurrency { private set; get; } = int.MaxValue;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：每帧发起的最大请求数
|
||
/// </summary>
|
||
public int DownloadMaxRequestPerFrame { private set; get; } = int.MaxValue;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：启用断点续传的最小尺寸
|
||
/// </summary>
|
||
public long ResumeDownloadMinimumSize { private set; get; } = long.MaxValue;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：断点续传下载器关注的错误码
|
||
/// </summary>
|
||
public List<long> ResumeDownloadResponseCodes { private set; get; } = null;
|
||
|
||
/// <summary>
|
||
/// 自定义参数：解密方法类
|
||
/// </summary>
|
||
public IDecryptionServices DecryptionServices { private set; get; }
|
||
#endregion
|
||
|
||
|
||
public DefaultCacheFileSystem()
|
||
{
|
||
}
|
||
public virtual FSInitializeFileSystemOperation InitializeFileSystemAsync()
|
||
{
|
||
var operation = new DCFSInitializeOperation(this);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
public virtual FSLoadPackageManifestOperation LoadPackageManifestAsync(string packageVersion, int timeout)
|
||
{
|
||
var operation = new DCFSLoadPackageManifestOperation(this, packageVersion, timeout);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
public virtual FSRequestPackageVersionOperation RequestPackageVersionAsync(bool appendTimeTicks, int timeout)
|
||
{
|
||
var operation = new DCFSRequestPackageVersionOperation(this, appendTimeTicks, timeout);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
public virtual FSClearCacheFilesOperation ClearCacheFilesAsync(PackageManifest manifest, string clearMode, object clearParam)
|
||
{
|
||
if (clearMode == EFileClearMode.ClearAllBundleFiles.ToString())
|
||
{
|
||
var operation = new ClearAllCacheBundleFilesOperation(this);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else if (clearMode == EFileClearMode.ClearUnusedBundleFiles.ToString())
|
||
{
|
||
var operation = new ClearUnusedCacheBundleFilesOperation(this, manifest);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else if (clearMode == EFileClearMode.ClearBundleFilesByTags.ToString())
|
||
{
|
||
var operation = new ClearCacheBundleFilesByTagsOperaiton(this, manifest, clearParam);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else if (clearMode == EFileClearMode.ClearAllManifestFiles.ToString())
|
||
{
|
||
var operation = new ClearAllCacheManifestFilesOperation(this);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else if (clearMode == EFileClearMode.ClearUnusedManifestFiles.ToString())
|
||
{
|
||
var operation = new ClearUnusedCacheManifestFilesOperation(this, manifest);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else
|
||
{
|
||
string error = $"Invalid clear mode : {clearMode}";
|
||
var operation = new FSClearCacheFilesCompleteOperation(error);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
}
|
||
public virtual FSDownloadFileOperation DownloadFileAsync(PackageBundle bundle, DownloadParam param)
|
||
{
|
||
return _downloadCenter.DownloadFileAsync(bundle, param);
|
||
}
|
||
public virtual FSLoadBundleOperation LoadBundleFile(PackageBundle bundle)
|
||
{
|
||
if (bundle.BundleType == (int)EBuildBundleType.AssetBundle)
|
||
{
|
||
var operation = new DCFSLoadAssetBundleOperation(this, bundle);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else if (bundle.BundleType == (int)EBuildBundleType.RawBundle)
|
||
{
|
||
var operation = new DCFSLoadRawBundleOperation(this, bundle);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
else
|
||
{
|
||
string error = $"{nameof(DefaultCacheFileSystem)} not support load bundle type : {bundle.BundleType}";
|
||
var operation = new FSLoadBundleCompleteOperation(error);
|
||
OperationSystem.StartOperation(PackageName, operation);
|
||
return operation;
|
||
}
|
||
}
|
||
|
||
public virtual void SetParameter(string name, object value)
|
||
{
|
||
if (name == FileSystemParametersDefine.REMOTE_SERVICES)
|
||
{
|
||
RemoteServices = (IRemoteServices)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.FILE_VERIFY_LEVEL)
|
||
{
|
||
FileVerifyLevel = (EFileVerifyLevel)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.APPEND_FILE_EXTENSION)
|
||
{
|
||
AppendFileExtension = (bool)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.DOWNLOAD_MAX_CONCURRENCY)
|
||
{
|
||
DownloadMaxConcurrency = (int)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.DOWNLOAD_MAX_REQUEST_PER_FRAME)
|
||
{
|
||
DownloadMaxRequestPerFrame = (int)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.RESUME_DOWNLOAD_MINMUM_SIZE)
|
||
{
|
||
ResumeDownloadMinimumSize = (long)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.RESUME_DOWNLOAD_RESPONSE_CODES)
|
||
{
|
||
ResumeDownloadResponseCodes = (List<long>)value;
|
||
}
|
||
else if (name == FileSystemParametersDefine.DECRYPTION_SERVICES)
|
||
{
|
||
DecryptionServices = (IDecryptionServices)value;
|
||
}
|
||
else
|
||
{
|
||
YooLogger.Warning($"Invalid parameter : {name}");
|
||
}
|
||
}
|
||
public virtual void OnCreate(string packageName, string packageRoot)
|
||
{
|
||
PackageName = packageName;
|
||
|
||
if (string.IsNullOrEmpty(packageRoot))
|
||
_packageRoot = GetDefaultCachePackageRoot(packageName);
|
||
else
|
||
_packageRoot = packageRoot;
|
||
|
||
_cacheBundleFilesRoot = PathUtility.Combine(_packageRoot, DefaultCacheFileSystemDefine.BundleFilesFolderName);
|
||
_tempFilesRoot = PathUtility.Combine(_packageRoot, DefaultCacheFileSystemDefine.TempFilesFolderName);
|
||
_cacheManifestFilesRoot = PathUtility.Combine(_packageRoot, DefaultCacheFileSystemDefine.ManifestFilesFolderName);
|
||
_downloadCenter = new DefaultCacheDownloadCenter(this);
|
||
}
|
||
public virtual void OnUpdate()
|
||
{
|
||
_downloadCenter.Update();
|
||
}
|
||
|
||
public virtual bool Belong(PackageBundle bundle)
|
||
{
|
||
// 注意：缓存文件系统保底加载！
|
||
return true;
|
||
}
|
||
public virtual bool Exists(PackageBundle bundle)
|
||
{
|
||
return _wrappers.ContainsKey(bundle.BundleGUID);
|
||
}
|
||
public virtual bool NeedDownload(PackageBundle bundle)
|
||
{
|
||
if (Belong(bundle) == false)
|
||
return false;
|
||
|
||
return Exists(bundle) == false;
|
||
}
|
||
public virtual bool NeedUnpack(PackageBundle bundle)
|
||
{
|
||
return false;
|
||
}
|
||
public virtual bool NeedImport(PackageBundle bundle)
|
||
{
|
||
if (Belong(bundle) == false)
|
||
return false;
|
||
|
||
return Exists(bundle) == false;
|
||
}
|
||
|
||
public virtual string GetBundleFilePath(PackageBundle bundle)
|
||
{
|
||
return GetCacheBundleFileLoadPath(bundle);
|
||
}
|
||
public virtual byte[] ReadBundleFileData(PackageBundle bundle)
|
||
{
|
||
if (Exists(bundle) == false)
|
||
return null;
|
||
|
||
if (bundle.Encrypted)
|
||
{
|
||
if (DecryptionServices == null)
|
||
{
|
||
YooLogger.Error($"The {nameof(IDecryptionServices)} is null !");
|
||
return null;
|
||
}
|
||
|
||
string filePath = GetCacheBundleFileLoadPath(bundle);
|
||
var fileInfo = new DecryptFileInfo()
|
||
{
|
||
BundleName = bundle.BundleName,
|
||
FileLoadCRC = bundle.UnityCRC,
|
||
FileLoadPath = filePath,
|
||
};
|
||
return DecryptionServices.ReadFileData(fileInfo);
|
||
}
|
||
else
|
||
{
|
||
string filePath = GetCacheBundleFileLoadPath(bundle);
|
||
return FileUtility.ReadAllBytes(filePath);
|
||
}
|
||
}
|
||
public virtual string ReadBundleFileText(PackageBundle bundle)
|
||
{
|
||
if (Exists(bundle) == false)
|
||
return null;
|
||
|
||
if (bundle.Encrypted)
|
||
{
|
||
if (DecryptionServices == null)
|
||
{
|
||
YooLogger.Error($"The {nameof(IDecryptionServices)} is null !");
|
||
return null;
|
||
}
|
||
|
||
string filePath = GetCacheBundleFileLoadPath(bundle);
|
||
var fileInfo = new DecryptFileInfo()
|
||
{
|
||
BundleName = bundle.BundleName,
|
||
FileLoadCRC = bundle.UnityCRC,
|
||
FileLoadPath = filePath,
|
||
};
|
||
return DecryptionServices.ReadFileText(fileInfo);
|
||
}
|
||
else
|
||
{
|
||
string filePath = GetCacheBundleFileLoadPath(bundle);
|
||
return FileUtility.ReadAllText(filePath);
|
||
}
|
||
}
|
||
|
||
#region 缓存相关
|
||
public List<string> GetAllCachedBundleGUIDs()
|
||
{
|
||
return _wrappers.Keys.ToList();
|
||
}
|
||
|
||
public string GetTempFilePath(PackageBundle bundle)
|
||
{
|
||
if (_tempFilePaths.TryGetValue(bundle.BundleGUID, out string filePath) == false)
|
||
{
|
||
filePath = PathUtility.Combine(_tempFilesRoot, bundle.BundleGUID);
|
||
_tempFilePaths.Add(bundle.BundleGUID, filePath);
|
||
}
|
||
return filePath;
|
||
}
|
||
public string GetBundleDataFilePath(PackageBundle bundle)
|
||
{
|
||
if (_bundleDataFilePaths.TryGetValue(bundle.BundleGUID, out string filePath) == false)
|
||
{
|
||
string folderName = bundle.FileHash.Substring(0, 2);
|
||
filePath = PathUtility.Combine(_cacheBundleFilesRoot, folderName, bundle.BundleGUID, DefaultCacheFileSystemDefine.BundleDataFileName);
|
||
if (AppendFileExtension)
|
||
filePath += bundle.FileExtension;
|
||
_bundleDataFilePaths.Add(bundle.BundleGUID, filePath);
|
||
}
|
||
return filePath;
|
||
}
|
||
public string GetBundleInfoFilePath(PackageBundle bundle)
|
||
{
|
||
if (_bundleInfoFilePaths.TryGetValue(bundle.BundleGUID, out string filePath) == false)
|
||
{
|
||
string folderName = bundle.FileHash.Substring(0, 2);
|
||
filePath = PathUtility.Combine(_cacheBundleFilesRoot, folderName, bundle.BundleGUID, DefaultCacheFileSystemDefine.BundleInfoFileName);
|
||
_bundleInfoFilePaths.Add(bundle.BundleGUID, filePath);
|
||
}
|
||
return filePath;
|
||
}
|
||
|
||
public bool IsRecordBundleFile(string bundleGUID)
|
||
{
|
||
return _wrappers.ContainsKey(bundleGUID);
|
||
}
|
||
public bool RecordBundleFile(string bundleGUID, RecordFileElement element)
|
||
{
|
||
if (_wrappers.ContainsKey(bundleGUID))
|
||
{
|
||
YooLogger.Error($"{nameof(DefaultCacheFileSystem)} has record file element : {bundleGUID}");
|
||
return false;
|
||
}
|
||
|
||
_wrappers.Add(bundleGUID, element);
|
||
return true;
|
||
}
|
||
|
||
public EFileVerifyResult VerifyCacheFile(PackageBundle bundle)
|
||
{
|
||
if (_wrappers.TryGetValue(bundle.BundleGUID, out RecordFileElement wrapper) == false)
|
||
return EFileVerifyResult.CacheNotFound;
|
||
|
||
EFileVerifyResult result = FileVerifyHelper.FileVerify(wrapper.DataFilePath, wrapper.DataFileSize, wrapper.DataFileCRC, EFileVerifyLevel.High);
|
||
return result;
|
||
}
|
||
public bool WriteCacheBundleFile(PackageBundle bundle, string copyPath)
|
||
{
|
||
if (_wrappers.ContainsKey(bundle.BundleGUID))
|
||
{
|
||
throw new Exception("Should never get here !");
|
||
}
|
||
|
||
string infoFilePath = GetBundleInfoFilePath(bundle);
|
||
string dataFilePath = GetBundleDataFilePath(bundle);
|
||
|
||
try
|
||
{
|
||
if (File.Exists(infoFilePath))
|
||
File.Delete(infoFilePath);
|
||
if (File.Exists(dataFilePath))
|
||
File.Delete(dataFilePath);
|
||
|
||
FileUtility.CreateFileDirectory(dataFilePath);
|
||
|
||
// 拷贝数据文件
|
||
FileInfo fileInfo = new FileInfo(copyPath);
|
||
fileInfo.CopyTo(dataFilePath);
|
||
|
||
// 写入文件信息
|
||
WriteBundleInfoFile(infoFilePath, bundle.FileCRC, bundle.FileSize);
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
YooLogger.Error($"Failed to write cache file ! {e.Message}");
|
||
return false;
|
||
}
|
||
|
||
var recordFileElement = new RecordFileElement(infoFilePath, dataFilePath, bundle.FileCRC, bundle.FileSize);
|
||
return RecordBundleFile(bundle.BundleGUID, recordFileElement);
|
||
}
|
||
public bool DeleteCacheBundleFile(string bundleGUID)
|
||
{
|
||
if (_wrappers.TryGetValue(bundleGUID, out RecordFileElement wrapper))
|
||
{
|
||
try
|
||
{
|
||
string dataFilePath = wrapper.DataFilePath;
|
||
FileInfo fileInfo = new FileInfo(dataFilePath);
|
||
if (fileInfo.Exists)
|
||
fileInfo.Directory.Delete(true);
|
||
_wrappers.Remove(bundleGUID);
|
||
return true;
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
YooLogger.Error($"Failed to delete cache file ! {e.Message}");
|
||
return false;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
return false;
|
||
}
|
||
}
|
||
|
||
private readonly BufferWriter _sharedBuffer = new BufferWriter(1024);
|
||
public void WriteBundleInfoFile(string filePath, string dataFileCRC, long dataFileSize)
|
||
{
|
||
using (FileStream fs = new FileStream(filePath, FileMode.Create, FileAccess.Write, FileShare.Read))
|
||
{
|
||
_sharedBuffer.Clear();
|
||
_sharedBuffer.WriteUTF8(dataFileCRC);
|
||
_sharedBuffer.WriteInt64(dataFileSize);
|
||
_sharedBuffer.WriteToStream(fs);
|
||
fs.Flush();
|
||
}
|
||
}
|
||
public void ReadBundleInfoFile(string filePath, out string dataFileCRC, out long dataFileSize)
|
||
{
|
||
byte[] binaryData = FileUtility.ReadAllBytes(filePath);
|
||
BufferReader buffer = new BufferReader(binaryData);
|
||
dataFileCRC = buffer.ReadUTF8();
|
||
dataFileSize = buffer.ReadInt64();
|
||
}
|
||
#endregion
|
||
|
||
#region 内部方法
|
||
public string GetDefaultCachePackageRoot(string packageName)
|
||
{
|
||
string rootDirectory;
|
||
|
||
#if UNITY_EDITOR
|
||
rootDirectory = YooAssetSettingsData.GetYooEditorCacheRoot();
|
||
#elif UNITY_STANDALONE
|
||
rootDirectory = YooAssetSettingsData.GetYooStandaloneCacheRoot();
|
||
#else
|
||
rootDirectory = YooAssetSettingsData.GetYooMobileCacheRoot();
|
||
#endif
|
||
|
||
return PathUtility.Combine(rootDirectory, packageName);
|
||
}
|
||
public string GetCacheBundleFileLoadPath(PackageBundle bundle)
|
||
{
|
||
return GetBundleDataFilePath(bundle);
|
||
}
|
||
public string GetCachePackageHashFilePath(string packageVersion)
|
||
{
|
||
string fileName = YooAssetSettingsData.GetPackageHashFileName(PackageName, packageVersion);
|
||
return PathUtility.Combine(_cacheManifestFilesRoot, fileName);
|
||
}
|
||
public string GetCachePackageManifestFilePath(string packageVersion)
|
||
{
|
||
string fileName = YooAssetSettingsData.GetManifestBinaryFileName(PackageName, packageVersion);
|
||
return PathUtility.Combine(_cacheManifestFilesRoot, fileName);
|
||
}
|
||
public string GetSandboxAppFootPrintFilePath()
|
||
{
|
||
return PathUtility.Combine(_cacheManifestFilesRoot, DefaultCacheFileSystemDefine.AppFootPrintFileName);
|
||
}
|
||
public string GetCacheBundleFilesRoot()
|
||
{
|
||
return _cacheBundleFilesRoot;
|
||
}
|
||
public string GetCacheManifestFilesRoot()
|
||
{
|
||
return _cacheManifestFilesRoot;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 删除所有清单文件
|
||
/// </summary>
|
||
public void DeleteAllManifestFiles()
|
||
{
|
||
if (Directory.Exists(_cacheManifestFilesRoot))
|
||
{
|
||
Directory.Delete(_cacheManifestFilesRoot, true);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 加载加密资源文件
|
||
/// </summary>
|
||
public DecryptResult LoadEncryptedAssetBundle(PackageBundle bundle)
|
||
{
|
||
string filePath = GetCacheBundleFileLoadPath(bundle);
|
||
var fileInfo = new DecryptFileInfo()
|
||
{
|
||
BundleName = bundle.BundleName,
|
||
FileLoadCRC = bundle.UnityCRC,
|
||
FileLoadPath = filePath,
|
||
};
|
||
return DecryptionServices.LoadAssetBundle(fileInfo);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 加载加密资源文件
|
||
/// </summary>
|
||
public DecryptResult LoadEncryptedAssetBundleAsync(PackageBundle bundle)
|
||
{
|
||
string filePath = GetCacheBundleFileLoadPath(bundle);
|
||
var fileInfo = new DecryptFileInfo()
|
||
{
|
||
BundleName = bundle.BundleName,
|
||
FileLoadCRC = bundle.UnityCRC,
|
||
FileLoadPath = filePath,
|
||
};
|
||
return DecryptionServices.LoadAssetBundleAsync(fileInfo);
|
||
}
|
||
#endregion
|
||
}
|
||
} |