执行摘要

• 一句话：修复 Mooncake Store 键值重复标记 bug
• 推荐动作：建议快速合并。该修复逻辑明确、改动精炼，且已有 reviewer stmatengss 批准。同时建议后续可将 gemini-code-assist[bot] 提出的切片优化（keys[:kv_pages]）作为微优化单独提交，尽管当前场景下影响不大。

功能与动机

修复 Mooncake Store 远程 KV 缓存查询中的 double-tag bug。PR body 未详细说明，但从变更逻辑可推断：原始代码先对 keys 调用 _tag_keys 得到 qkeys，然后将 qkeys 既传给 batch_exists，又传给 _get_hybrid_page_component_keys。后者生成的 component key 本身可能已包含 tag 信息，再经过 _batch_exist 前的隐式标记会导致重复，可能产生错误的键，影响命中判断。

实现拆解

该 PR 仅涉及一个文件 mooncake_store.py，核心改动是调整 _tag_keys 的调用时机：

1. 移除对原始 keys 的提前标记：删除 qkeys = self._tag_keys(keys)，直接传入原始 keys 给 batch_exists 和 _get_hybrid_page_component_keys。
2. 将标记延迟到 component keys 生成后：在 _get_hybrid_page_component_keys 返回 component_keys 后，新增一行 component_keys = self._tag_keys(component_keys)，确保仅对 component keys 做一次必要标记。

这样既避免了 double-tag，又保证了最终发送给存储后端的键仍是带有正确 tag 的。

关键文件：

• python/sglang/srt/mem_cache/storage/mooncake_store/mooncake_store.py（模块 内存缓存；类别 source；类型 core-logic；符号 batch_exists_v2）: 核心 bug 修复文件，调整了 _tag_keys 的调用时机以消除 double-tag。

关键符号：batch_exists_v2

关键源码片段

python/sglang/srt/mem_cache/storage/mooncake_store/mooncake_store.py

核心 bug 修复文件，调整了 _tag_keys 的调用时机以消除 double-tag。

def batch_exists_v2(
    self,
    keys: List[str],
    pool_transfers: Optional[List[PoolTransfer]] = None,
    extra_info: Optional[HiCacheStorageExtraInfo] = None,
) -> PoolTransferResult:
    # 关键修复：不再提前对 keys 调用 _tag_keys，直接使用原始 keys
    kv_pages = self.batch_exists(keys, extra_info)
​
    hit_count: dict = {PoolName.KV: kv_pages} if kv_pages else {}
    final_pages = kv_pages
​
    for transfer in pool_transfers or []:
        if final_pages == 0:
            break
        component_keys, key_multiplier = self._get_hybrid_page_component_keys(
            keys, transfer
        )
        # 修复：将标记延迟到 component_keys 生成后，只标记一次
        component_keys = self._tag_keys(component_keys)
        ex = self._batch_exist(component_keys)
        # 后续逻辑不变 ...

评论区精华

Reviewer gemini-code-assist[bot] 提出了一个性能优化建议：由于 kv_pages 仅代表 KV pool 中实际存在的 page 数量，超出部分的 key 在后续 hybrid pool 查询中必然失败，建议将 keys 切片为 keys[:kv_pages] 再传给 _get_hybrid_page_component_keys，以减少对已确定为不存在的页的多余 metadata 查询。该建议未被采纳（suggestion 状态为 pending，但最终合并版本未包含该优化）。

• 性能优化：切片 keys 避免冗余查询 (performance): 未采纳。当前修复只聚焦于 double-tag 问题，切片优化可作为独立改进。

风险与影响

• 风险：该变更为核心逻辑调整，但改动极小（仅 3 行变化），且修复逻辑直观：核心风险在于是否所有调用 batch_exists_v2 的地方都已依赖正确的 key 格式。原始代码中 batch_exists 调用的是 qkeys（已 tag），现在改为 keys（未 tag），而 batch_exists 内部可能会自标记或假定 key 已标记；如果 batch_exists 不进行自标记且依赖调用方预标记，则可能导致 key 缺失 tag。但从代码看 batch_exists 即为底层存储接口，推测其不处理标记，因此风险较低。另一个潜在风险是，如果 _tag_keys 方法有副作用（如修改输入列表），延后调用可能改变行为；但该方法是纯函数式操作，风险可控。
• 影响：影响范围：仅限于 batch_exists_v2 一个方法，影响所有通过 MooncakeStore 进行远程 KV 缓存存在的查询，包括主 KV pool 和 hybrid pool（INDEXER, MAMBA）。影响程度：中等。此 bug 会导致在 hybrid pool 场景下 key 重复标记，可能产生无效的存储键，使得实际存在的页被误判为不存在，或反之，从而影响缓存命中率和推理准确性。修复后，键的生成与标记逻辑正确，缓存查询结果可靠。
• 风险标记：核心路径变更, 缺少测试覆盖

关联脉络

• PR #26675 [mem_cache][1/N] refactor: split allocator.py into allocator/ subpackage: 同属 mem_cache 模块重构，涉及存储层拆分，可能有重叠的 key 处理逻辑。
• PR #26821 Add periodic KV-canary stats logging and kernel-run-counter health check: 涉及 KV 缓存可观测性，可能受益于正确的 key 标记逻辑。