[uv] uv의 휠 태그 호환성 검사 최적화: 불필요한 메모리 할당 제거하기

PR 링크: astral-sh/uv#20171 상태: Merged | 변경: +95 / -10

들어가며

Python 패키지 매니저인 uv는 압도적인 속도를 자랑합니다. 이 속도의 비결 중 하나는 패키지 설치 시 휠(wheel) 파일이 현재 환경과 호환되는지 판단하는 로직이 매우 효율적이라는 점입니다. 하지만 기존 ExpandedTags::compatibility 구현에서는 매번 Vec을 할당하여 태그를 수집하는 과정이 포함되어 있었습니다. 이 PR은 불필요한 메모리 할당을 제거하고, 단일 태그 케이스를 직접 처리함으로써 성능을 획기적으로 개선했습니다.

코드 분석

1. ExpandedTags::compatibility 최적화

기존 코드에서는 태그를 검사할 때마다 collect::<Vec<_>>()를 호출하여 힙 메모리를 할당했습니다. 이를 단일 태그인 경우 즉시 반환하도록 수정했습니다.

// Before
compatible_tags.compatibility(
    self.python_tags().copied().collect::<Vec<_>>().as_slice(),
    self.abi_tags().copied().collect::<Vec<_>>().as_slice(),
    // ...
)

// After
if let [tag] = self.0.as_slice() {
    return tag.compatibility(compatible_tags);
}

2. Tags::compatibility_tag 도입

uv-platform-tags 크레이트에 단일 태그에 대한 호환성을 직접 확인하는 compatibility_tag 메서드를 추가했습니다. 이는 기존의 복잡한 슬라이스 기반 검사 대신, 해시맵 조회를 통해 즉시 결과를 반환합니다.

pub fn compatibility_tag(
    &self, 
    wheel_python_tag: &LanguageTag, 
    wheel_abi_tag: &AbiTag, 
    wheel_platform_tag: &PlatformTag
) -> TagCompatibility {
    // ... (freethreaded 체크)
    let Some(abis) = self.map.get(wheel_python_tag) else { return ... };
    let Some(platforms) = abis.get(wheel_abi_tag) else { return ... };
    let Some(priority) = platforms.get(wheel_platform_tag).copied() else { return ... };
    TagCompatibility::Compatible(priority)
}

왜 이게 좋은가

이번 최적화의 핵심은 '불필요한 할당(Allocation) 제거'입니다. Rust에서 Vec을 생성하는 것은 힙 메모리 할당을 동반하며, 이는 작은 데이터셋을 다룰 때 큰 오버헤드가 됩니다. 특히 휠 태그 검사는 패키지 설치 시 수천 번 호출될 수 있는 핫패스(hot path)이므로, 이 작은 변화가 벤치마크에서 3.8배에서 5.6배에 달하는 성능 향상을 가져왔습니다.

• small_exact: 47.031 ns -> 8.626 ns
• multipart: 55.592 ns -> 14.776 ns

교훈: 핫패스 내에서의 collect() 호출은 항상 주의 깊게 살펴봐야 합니다. 데이터의 구조가 단순하다면, 굳이 중간 컬렉션을 만들지 않고도 직접적인 조회(lookup)를 통해 동일한 로직을 수행할 수 있는지 고민해야 합니다.

참고 자료

• https://doc.rust-lang.org/std/vec/struct.Vec.html
• https://github.com/astral-sh/uv

⚠️ 알림: 이 분석은 AI가 실제 코드 diff를 기반으로 작성했습니다.