如何改写Unity中NullReferenceException行为，生成更具信息量的错误提示？

NullReferenceException确实是Unity开发中最头疼的问题之一，尤其是当调用链很长时，根本没法快速定位到底是哪个环节为空。下面我分享几个实用的方案，从主动预防和运行时增强提示两个维度解决这个问题：

方案1：自定义空检查工具类（最直接可控） #

自己封装一个静态工具类，在访问对象/组件/成员前主动做检查，抛出带上下文的异常。这种方式虽然需要修改现有代码，但能得到最精准的错误信息。

public static class NullGuard
{
    // 通用空检查，支持传入上下文描述和来源GameObject
    public static T EnsureNotNull<T>(this T obj, string context, GameObject sourceObj = null) where T : class
    {
        if (obj == null)
        {
            string extraInfo = sourceObj != null ? $" | 来源GameObject: \"{sourceObj.name}\"" : "";
            throw new NullReferenceException($"空引用错误：{context}{extraInfo}");
        }
        return obj;
    }

    // 针对Unity组件的快捷检查
    public static T EnsureComponentNotNull<T>(this GameObject obj) where T : Component
    {
        var component = obj.GetComponent<T>();
        if (component == null)
        {
            throw new NullReferenceException($"空引用错误：GameObject \"{obj.name}\" 上找不到 {typeof(T).Name} 组件");
        }
        return component;
    }
}

使用示例：

// 原来的写法（出错后无法定位）
player.GetComponent<Movement>().AnotherVariable.CallFunction();

// 修改后的写法（精准提示哪个环节为空）
player.EnsureNotNull("访问Player GameObject")
      .EnsureComponentNotNull<Movement>()
      .AnotherVariable.EnsureNotNull("访问Movement组件的AnotherVariable成员", player)
      .CallFunction();

抛出的异常信息会是：NullReferenceException: 空引用错误：访问Movement组件的AnotherVariable成员 | 来源GameObject: "Player"，一眼就能定位问题。

方案2：利用Unity日志回调拦截并增强异常 #

如果不想大规模修改业务代码，可以通过Unity的日志回调拦截默认的空引用异常，尝试解析栈追踪补充上下文信息。

using UnityEngine;
using System.Diagnostics;
using System.Reflection;

public class ExceptionEnhancer : MonoBehaviour
{
    private void OnEnable()
    {
        Application.logMessageReceived += EnhanceNullReferenceLogs;
    }

    private void OnDisable()
    {
        Application.logMessageReceived -= EnhanceNullReferenceLogs;
    }

    private void EnhanceNullReferenceLogs(string logString, string stackTrace, LogType logType)
    {
        if (logType != LogType.Exception || !logString.StartsWith("NullReferenceException:"))
            return;

        StackTrace trace = new StackTrace(stackTrace, true);
        foreach (StackFrame frame in trace.GetFrames())
        {
            MethodBase method = frame.GetMethod();
            // 尝试定位异常所在的MonoBehaviour
            if (method.DeclaringType?.IsSubclassOf(typeof(MonoBehaviour)) == true)
            {
                // 查找场景中该类型的实例（如果是单例或唯一实例会更准确）
                MonoBehaviour mbInstance = FindObjectOfType(method.DeclaringType) as MonoBehaviour;
                if (mbInstance != null)
                {
                    string enhancedMsg = $"{logString}\n--- 上下文补充 ---\n异常发生在MonoBehaviour: {method.DeclaringType.Name}\n所属GameObject: \"{mbInstance.gameObject.name}\"";
                    Debug.LogError(enhancedMsg);
                }
            }
        }
    }
}

这个方案的优点是不用改业务代码，但缺点是栈追踪解析的精准度有限——比如无法直接定位到具体是空成员还是空组件，只能拿到所在的MonoBehaviour和GameObject。

方案3：编辑器层面提前排查（从根源减少空引用） #

很多空引用是因为设计师忘记给序列化变量赋值，我们可以在编辑阶段就自动检查并给出警告，避免带到运行时。

using UnityEditor;
using UnityEngine;

public class MissingReferenceChecker : EditorWindow
{
    [MenuItem("Tools/检查缺失引用")]
    public static void OpenChecker()
    {
        GetWindow<MissingReferenceChecker>("缺失引用检查");
    }

    private void OnGUI()
    {
        if (GUILayout.Button("扫描当前场景所有对象"))
        {
            ScanSceneForMissingReferences();
        }
    }

    private void ScanSceneForMissingReferences()
    {
        MonoBehaviour[] allBehaviours = FindObjectsOfType<MonoBehaviour>(true);
        foreach (MonoBehaviour mb in allBehaviours)
        {
            SerializedObject serializedObj = new SerializedObject(mb);
            SerializedProperty prop = serializedObj.GetIterator();
            
            while (prop.NextVisible(true))
            {
                if (prop.propertyType == SerializedPropertyType.ObjectReference && prop.objectReferenceValue == null)
                {
                    Debug.LogWarning($"[缺失引用] {mb.GetType().Name} 组件的 \"{prop.displayName}\" 未赋值", mb.gameObject);
                }
            }
        }
    }
}

把这个脚本放在Editor文件夹下，就能通过菜单栏的Tools/检查缺失引用按钮，一键扫描场景中所有未赋值的序列化变量，提前解决问题。

额外提示：升级Unity版本 #

Unity 2021及以上版本已经优化了NullReferenceException的栈追踪信息，会自动显示异常所在的GameObject名称（如果异常发生在MonoBehaviour方法内），可以先升级版本看看是否满足你的需求，再考虑自定义方案。

内容的提问来源于stack exchange，提问作者LouisGameDev