Sidekiq中CurrentAttributes在inline执行后丢失的问题分析

问题背景

在Sidekiq 7.2.4版本中，当使用perform_inline方法执行作业时，发现CurrentAttributes中的当前属性会在执行后被意外重置。这个问题在从Sidekiq 6升级到7时被发现，特别是在父作业调用子作业的场景下表现得尤为明显。

CurrentAttributes机制解析

CurrentAttributes是Rails提供的一种机制，用于在请求或作业执行期间存储和访问全局状态。在Sidekiq中，这个功能被扩展用于在作业执行期间保持特定类的属性状态。

问题重现

通过一个简单的测试用例可以重现这个问题：

class CurrentAttributesJob
  include Sidekiq::Job
  def perform; end
end

Sidekiq::CurrentAttributes.persist(Myapp::Current, config)
Myapp::Current.user_id = 1
CurrentAttributesJob.perform_inline
# 此处Myapp::Current.user_id预期为1，实际为nil

问题根源分析

问题的根源在于Sidekiq::CurrentAttributes::Load#call方法的实现。当前实现会在作业执行后重置所有属性，而不是使用ActiveSupport::CurrentAttributes的set方法来正确地嵌套保存属性状态。

解决方案

有两种可行的解决方案：

1. 方法链式调用方案： 通过将各个CurrentAttributes类及其属性值收集起来，然后使用inject方法构建一个嵌套的set调用链。

2. 递归方案： 使用递归方式依次为每个CurrentAttributes类创建嵌套的set块，这种方式代码更清晰易读。

递归方案的实现代码如下：

def call(_, job, _, &block)
  klass_attrs = {}
  
  @cattrs.each do |(key, strklass)|
    next unless job.has_key?(key)
    klass_attrs[strklass.constantize] = job[key]
  end

  wrap(klass_attrs.to_a, &block)
end

def wrap(klass_attrs, &block)
  klass, attrs = klass_attrs.shift
  return block.call unless klass

  klass.set(attrs) do
    wrap(klass_attrs, &block)
  end
end

技术要点

1. CurrentAttributes的持久化： 正确的做法应该是使用ActiveSupport::CurrentAttributes的set方法，它会创建一个嵌套的执行上下文，确保在块执行完毕后恢复之前的状态。

2. 递归与函数式编程： 递归方案展示了如何优雅地处理不确定深度的嵌套结构，这是函数式编程的典型应用场景。

3. 线程安全考虑： 这种实现方式确保了在多线程环境下CurrentAttributes的状态不会互相干扰，每个作业执行都有自己独立的上下文。

影响范围

这个问题主要影响以下场景：

• 使用perform_inline同步执行作业
• 作业内部再调用其他作业（嵌套作业执行）
• 依赖CurrentAttributes在作业执行后保持状态的场景

最佳实践建议

1. 在升级Sidekiq版本时，应特别注意CurrentAttributes相关的测试用例
2. 对于复杂的作业流程，考虑显式传递必要参数而不是过度依赖CurrentAttributes
3. 在测试环境中增加对CurrentAttributes状态保持的验证

这个问题已经在Sidekiq的最新版本中得到修复，开发者可以通过升级到最新版本来解决这个问题。