CommunityToolkit.Mvvm 源码分析 (11) — IMessenger 接口体系篇

public interface IMessenger
{
    bool IsRegistered<TMessage, TToken>(object recipient, TToken token)
        where TMessage : class where TToken : IEquatable<TToken>;

    void Register<TRecipient, TMessage, TToken>(TRecipient recipient, TToken token,
        MessageHandler<TRecipient, TMessage> handler)
        where TRecipient : class where TMessage : class where TToken : IEquatable<TToken>;

    void UnregisterAll(object recipient);
    void UnregisterAll<TToken>(object recipient, TToken token) where TToken : IEquatable<TToken>;
    void Unregister<TMessage, TToken>(object recipient, TToken token)
        where TMessage : class where TToken : IEquatable<TToken>;

    TMessage Send<TMessage, TToken>(TMessage message, TToken token)
        where TMessage : class where TToken : IEquatable<TToken>;

    void Cleanup();
    void Reset();
}

// 方式 1：Lambda 委托（最常用）
WeakReferenceMessenger.Default.Register<MyMessage>(this, (r, m) =>
{
    // r = this（接收者），m = 消息
    Console.WriteLine($"收到消息: {m.Text}");
});

// 方式 2：静态方法（零分配，推荐）
WeakReferenceMessenger.Default.Register<MyMessage>(this,
    static (r, m) => ((MyViewModel)r).HandleMessage(m));

// 方式 3：方法组（简洁）
WeakReferenceMessenger.Default.Register<MyMessage>(this, HandleMessage);
private void HandleMessage(object recipient, MyMessage message) { }

// 方式 4：IRecipient<T> 接口（最清晰，零分配）
public class MyViewModel : IRecipient<MyMessage>, IRecipient<OtherMessage>
{
    public MyViewModel()
    {
        WeakReferenceMessenger.Default.RegisterAll(this);
    }

    public void Receive(MyMessage message) { }
    public void Receive(OtherMessage message) { }
}

// 方式 5：带 Token 的频道
WeakReferenceMessenger.Default.Register<MyMessage, string>(
    this, "ChannelA", (r, m) => { });

// 委托签名：接收者 + 消息
public delegate void MessageHandler<TRecipient, in TMessage>(
    TRecipient recipient, TMessage message);

// 为什么把接收者作为参数传入？
// → 避免闭包捕获，支持 static 方法

// ❌ 传统事件方式（闭包捕获 = 分配）
button.Click += (s, e) => this.HandleClick(e);
// ↑ 捕获了 this，每次 new 一个闭包

// ✅ 框架方式（静态委托 = 零分配）
Messenger.Register<MyMessage>(this,
    static (r, m) => ((MyViewModel)r).Handle(m));
// ↑ 静态 lambda，编译器缓存委托实例

// 所有默认通道的 API，内部使用 Unit 类型
public static void Register<TMessage>(this IMessenger messenger,
    IRecipient<TMessage> recipient)
{
    // 调用完整接口，但 Unit 作为 TToken
    messenger.Register<IRecipient<TMessage>, TMessage, Unit>(
        recipient, default, static (r, m) => r.Receive(m));
}

// Unit 是一个空的内部结构体
internal readonly struct Unit : IEquatable<Unit>
{
    public bool Equals(Unit other) => true;
    public override int GetHashCode() => 0;
}

public static void RegisterAll(this IMessenger messenger, object recipient)
{
    // Fast Path：从程序集中找生成器生成的注册方法
    var registrationAction = DiscoveredRecipients.RegistrationMethods.GetValue(
        recipient.GetType(), type =>
    {
        if (type.Assembly.GetType(
            "CommunityToolkit.Mvvm.Messaging.__Internals.__IMessengerExtensions")
            is Type t &&
            t.GetMethod("CreateAllMessagesRegistrator", new[] { type })
            is MethodInfo mi)
        {
            return (Action<IMessenger, object>)mi.Invoke(null, null)!;
        }
        return null;
    });

    if (registrationAction is not null)
        registrationAction(messenger, recipient);  // Fast Path
    else
        messenger.RegisterAll(recipient, default(Unit));  // Slow Path
}