CalTools/Unity相关/DomainReloadTool.cs

using UnityEditor;
using UnityEditor.Compilation;
using UnityEngine;

namespace DragonSoul.Editor;

public class DomainReloadTool
{
    private const string logColor = "<color=grey>{0}</color>";
    private const string ReloadKey = "DomainReloadTool.Reload";
    private const string ReloadDomainTimerKey = "DomainReloadTool.ReloadDomainTimer";
    private const string CompileDomainTimerKey = "DomainReloadTool.CompileDomainTimer";

    [InitializeOnLoadMethod]
    static void Init()
    {
        
        CompilationPipeline.compilationStarted += OnCompilationStarted;
        CompilationPipeline.compilationFinished += OnCompilationFinished;
        
        AssemblyReloadEvents.beforeAssemblyReload += OnBeforeAssemblyReload;
        AssemblyReloadEvents.afterAssemblyReload += OnAfterAssemblyReload;

        EditorApplication.playModeStateChanged += EditorApplication_playModeStateChanged;
    }

    private static void OnCompilationStarted(object obj)
    {
        
        Debug.LogFormat(logColor, "Begin Compile Domain...");
        //记录时间
        SessionState.SetInt(CompileDomainTimerKey, (int)(EditorApplication.timeSinceStartup * 1000));
    }
    private static void OnCompilationFinished(object obj)
    {
        var timeMS = (int)(EditorApplication.timeSinceStartup * 1000) - SessionState.GetInt(CompileDomainTimerKey, 0);
        Debug.LogFormat(logColor, $"End Compile Domain : {timeMS} ms");
    }


    private static void EditorApplication_playModeStateChanged(PlayModeStateChange state)
    {
        switch (state)
        {
            case PlayModeStateChange.ExitingEditMode:
                if (!SessionState.GetBool(ReloadKey, false))
                    ForceReloadDomain();
                break;
            case PlayModeStateChange.EnteredPlayMode:
                break;
            case PlayModeStateChange.ExitingPlayMode:
                SessionState.SetBool(ReloadKey, false);
                break;
            case PlayModeStateChange.EnteredEditMode:
                break;
        }
    }

    //开始reload domain
    private static void OnBeforeAssemblyReload()
    {
        Debug.LogFormat(logColor, "Begin Reload Domain...");
        //记录时间
        SessionState.SetInt(ReloadDomainTimerKey, (int)(EditorApplication.timeSinceStartup * 1000));
    }

    //结束reload domain
    private static void OnAfterAssemblyReload()
    {
        var timeMS = (int)(EditorApplication.timeSinceStartup * 1000) - SessionState.GetInt(ReloadDomainTimerKey, 0);
        Debug.LogFormat(logColor, $"End Reload Domain : {timeMS} ms");
        SessionState.SetBool(ReloadKey, true);
    }

    private static void ForceReloadDomain()
    {
        EditorUtility.RequestScriptReload();
    }
}