ET/Unity/Assets/Scripts/Loader/CodeLoader.cs

using Cysharp.Threading.Tasks;
using HybridCLR;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Reflection;
using UnityEngine;
using YooAsset;

namespace ET
{
	public class CodeLoader: Singleton<CodeLoader>
	{
		public void Start()
		{
			if (Define.EnableCodes)
			{
				GlobalConfig globalConfig = Resources.Load<GlobalConfig>("GlobalConfig");
				if (globalConfig.CodeMode != CodeMode.ClientServer)
				{
					throw new Exception("ENABLE_CODES mode must use ClientServer code mode!");
				}

                GameObject.Find("HotfixUI").SetActive(false);
                Assembly[] assemblies = AppDomain.CurrentDomain.GetAssemblies();
				Dictionary<string, Type> types = AssemblyHelper.GetAssemblyTypes(assemblies);
				EventSystem.Instance.Add(types);

                bool bStart = false;
				foreach (Assembly ass in assemblies)
				{
					string name = ass.GetName().Name;
					if (name == "Unity.Model.Codes")
					{
                        bStart = true;
                        IStaticMethod start = new StaticMethod(ass, "ET.Entry", "Start");
                        start.Run();
                        break;
                    }
				}

                if(!bStart)
                {
                    throw new Exception("not found Assembly: 'Unity.Model.Codes'");
                }
			}
			else
			{
                LoadHotfix();
            }
		}

        //编译时会检查这个方法，需保留
        public void LoadHotfix()
        {
            AsyncLoadHotfix().Forget();
        }

        private async UniTaskVoid AsyncLoadHotfix()
        {
            GlobalConfig globalConfig = Resources.Load<GlobalConfig>("GlobalConfig");
            Log.Info($"start mode: {globalConfig.PlayMode}");

            // 启动YooAsset引擎，并在初始化完毕后进行热更代码加载
            await YooAssetProxy.StartYooAssetEngine(globalConfig.PlayMode);

            // Shader Warm Up
            ShaderVariantCollection shaderVariantCollection =
                (await YooAssetProxy.LoadAssetAsync<ShaderVariantCollection>(
                    YooAssetProxy.GetYooAssetFormatResPath("ProjectSShaderVariant",
                        YooAssetProxy.YooAssetResType.Shader)))
                .GetAssetObject<ShaderVariantCollection>();
            Stopwatch stopwatch = Stopwatch.StartNew();
            shaderVariantCollection.WarmUp();
            stopwatch.Stop();
            Log.Info($"Shader Warm Up完成, 耗时: {stopwatch.ElapsedMilliseconds}ms，shaderCount: {shaderVariantCollection.shaderCount} variantCount: {shaderVariantCollection.variantCount}");

            //给mscorlib.dll等AOT补充元数据
            await LoadMetadataForAOT();
            await LoadHotfixCode();

            GameObject.Find("HotfixUI").SetActive(false);
        }

		// 加载assembly对应的dll，会自动为它hook。一旦aot泛型函数的native函数不存在，用解释器版本代码
		private async UniTask LoadMetadataForAOT()
		{
            //AOT dll列表的配置也可以通过热更新下下来，这样增加了新的泛型就不怕了
            HotUpdateAssemblyManifest mainfest =
                (await YooAssetProxy.LoadAssetAsync<HotUpdateAssemblyManifest>("HotUpdateAssemblyManifest"))
                .GetAssetObject<HotUpdateAssemblyManifest>();
            var DLLNameList_ForABLoad = mainfest.AOTMetadataDlls;

			List<UniTask<RawFileOperation>> tasks = new List<UniTask<RawFileOperation>>();
            foreach (var aotDll in DLLNameList_ForABLoad)
            {
                tasks.Add(YooAssetProxy.GetRawFileAsync(aotDll));
            }
            RawFileOperation[] rawFileOperations = await UniTask.WhenAll(tasks);

            foreach (var task in rawFileOperations)
            {
                var ret = RuntimeApi.LoadMetadataForAOTAssembly(task.GetRawBytes(), HomologousImageMode.SuperSet);
                if (ret == LoadImageErrorCode.OK)
                {
                    Log.Debug($"LoadMetadataForAOTAssembly({task.GetBundleName()}) ok.");
                }
                else
                {
                    Log.Error($"LoadMetadataForAOTAssembly({task.GetBundleName()}) Error: {ret}");
                }
            }
        }

        //热更加载 model.dll & hotfix.dll
        private async UniTask LoadHotfixCode()
        {
            Log.Info("to load model.dll & hotfix.dll.");

            var rawfile_modeldll = await YooAssetProxy.GetRawFileAsync("Model.dll");
            var rawfile_modelpdb = await YooAssetProxy.GetRawFileAsync("Model.pdb");
            var modelAssembly = Assembly.Load(rawfile_modeldll.GetRawBytes(), rawfile_modelpdb.GetRawBytes());

            var rawfile_hotfixdll = await YooAssetProxy.GetRawFileAsync("Hotfix.dll");
            var rawfile_hotfixpdb = await YooAssetProxy.GetRawFileAsync("Hotfix.pdb");
            var hotfixAssembly = Assembly.Load(rawfile_hotfixdll.GetRawBytes(), rawfile_hotfixpdb.GetRawBytes());

            Dictionary<string, Type> types = AssemblyHelper.GetAssemblyTypes(typeof(Game).Assembly, typeof(Init).Assembly, modelAssembly, hotfixAssembly);
            EventSystem.Instance.Add(types);

            Log.Info("to call Client.Entry.");
            IStaticMethod start = new MonoStaticMethod(modelAssembly, "ET.Entry", "Start");
            start.Run();
        }
    }
}