WCF服务IErrorHandler.ProvideFault中CurrentPrincipal为null的问题

这个问题我之前做WCF服务异常处理时也踩过坑，确实是线程上下文切换导致的——WCF默认情况下，ProvideFault方法的执行线程大概率和服务业务方法的线程不是同一个，所以Thread.CurrentPrincipal不会自动传递过来，就出现了你看到的null情况。

下面给你几个经过验证的可行方案，按通用性排序：

方案一：利用OperationContext存储Principal（最推荐） #

OperationContext是和当前请求强绑定的上下文对象，不管线程怎么切换，只要是同一个请求的处理流程，都能通过它拿到关联的数据。你可以在IAuthorizationPolicy.Evaluate方法里，把设置好的Principal同时存入OperationContext的消息属性中：

步骤1：在IAuthorizationPolicy.Evaluate中存储Principal #

public bool Evaluate(EvaluationContext evaluationContext, ref object state)
{
    // 你的逻辑：从IncomingMessageHeader解析并创建Principal
    var customPrincipal = new GenericPrincipal(/* 你的身份标识和角色 */);
    
    // 1. 按原来的逻辑设置到evaluationContext，保证服务方法能拿到
    evaluationContext.Properties["Principal"] = customPrincipal;
    
    // 2. 同时存入OperationContext的消息属性
    if (OperationContext.Current != null)
    {
        OperationContext.Current.IncomingMessageProperties["CustomRequestPrincipal"] = customPrincipal;
    }
    
    return true;
}

步骤2：在IErrorHandler.ProvideFault中读取 #

public void ProvideFault(Exception error, MessageVersion version, ref Message fault)
{
    IPrincipal currentPrincipal = null;
    
    // 从OperationContext中取出之前存储的Principal
    if (OperationContext.Current?.IncomingMessageProperties.TryGetValue("CustomRequestPrincipal", out var principalObj) == true)
    {
        currentPrincipal = principalObj as IPrincipal;
    }
    
    // 现在你可以用currentPrincipal做权限相关的异常处理了
    if (currentPrincipal != null)
    {
        // 比如记录当前用户的异常日志，或者根据角色返回不同的错误信息
        Log.Error($"用户 {currentPrincipal.Identity.Name} 执行操作时出错：{error.Message}");
    }
    
    // 你的其他错误处理逻辑...
}

方案二：通过InstanceContext扩展存储Principal #

如果你的服务是实例上下文绑定（比如PerCall、PerSession），可以把Principal关联到InstanceContext上，通过自定义扩展类来存储：

步骤1：创建自定义InstanceContext扩展和初始化器 #

// 存储Principal的扩展类
public class PrincipalExtension : IExtension<InstanceContext>
{
    public IPrincipal RequestPrincipal { get; }

    public PrincipalExtension(IPrincipal principal)
    {
        RequestPrincipal = principal;
    }

    public void Attach(InstanceContext owner) { }
    public void Detach(InstanceContext owner) { }
}

// 初始化器，把Principal注入到InstanceContext
public class PrincipalInstanceInitializer : IInstanceContextInitializer
{
    public void Initialize(InstanceContext instanceContext, Message message)
    {
        if (OperationContext.Current?.IncomingMessageProperties.TryGetValue("CustomRequestPrincipal", out var principalObj) == true)
        {
            var principal = principalObj as IPrincipal;
            instanceContext.Extensions.Add(new PrincipalExtension(principal));
        }
    }
}

步骤2：在服务配置中注册初始化器 #

你需要在WCF的服务行为里注册这个初始化器，确保每个实例上下文创建时都会执行初始化逻辑。

步骤3：在ProvideFault中读取 #

public void ProvideFault(Exception error, MessageVersion version, ref Message fault)
{
    IPrincipal currentPrincipal = null;
    var instanceContext = OperationContext.Current?.InstanceContext;
    
    if (instanceContext != null)
    {
        var principalExtension = instanceContext.Extensions.Find<PrincipalExtension>();
        currentPrincipal = principalExtension?.RequestPrincipal;
    }
    
    // 后续处理逻辑...
}

方案三：在异常的Data属性中传递Principal（适合局部场景） #

如果你的服务方法可以手动捕获异常，也可以把当前Principal附加到异常的Data集合中，然后在ProvideFault里取出：

在服务方法中处理 #

public string MyServiceMethod()
{
    try
    {
        // 你的服务业务逻辑
        return "Success";
    }
    catch (Exception ex)
    {
        // 把当前Principal存入异常的Data属性
        ex.Data["CurrentPrincipal"] = Thread.CurrentPrincipal;
        throw; // 重新抛出，让IErrorHandler捕获
    }
}

在ProvideFault中读取 #

public void ProvideFault(Exception error, MessageVersion version, ref Message fault)
{
    IPrincipal currentPrincipal = null;
    
    if (error.Data.Contains("CurrentPrincipal"))
    {
        currentPrincipal = error.Data["CurrentPrincipal"] as IPrincipal;
    }
    
    // 处理逻辑...
}

关键说明 #

WCF的Thread.CurrentPrincipal是线程级别的，当线程切换时（比如错误处理线程池线程），这个值不会自动传递。而OperationContext和InstanceContext是和请求/服务实例绑定的，属于请求级别的上下文，所以能跨线程保留数据，这也是前两种方案更可靠的原因。

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