집 >백엔드 개발 >Golang >`unsafe`를 사용하여 복사하지 않고 Go 문자열에서 바이트 슬라이스를 얻는 방법은 무엇입니까?

`unsafe`를 사용하여 복사하지 않고 Go 문자열에서 바이트 슬라이스를 얻는 방법은 무엇입니까?

Barbara Streisand원래의: 2024-11-01 01:51:28913검색

How to Obtain a Byte Slice from a Go String without Copying Using `unsafe`?

Unsafe를 사용하여 복사하지 않고 문자열에서 바이트 슬라이스 얻기

Go 문자열은 변경할 수 없습니다. 즉, 바이트 슬라이스로 변환하려면 메모리 복사가 필요합니다. 이는 대규모 데이터 세트로 작업할 때 잠재적으로 성능에 영향을 미칠 수 있습니다. 이 기사에서는 중요한 측면과 제한 사항을 강조하면서 이러한 복사 작업을 피하기 위해 unsafe를 사용하는 방법을 살펴봅니다.

배경

표준 라이브러리 함수 []byte(s)는 문자열 s의 복사본을 만듭니다. . 메모리 소비가 문제라면 이러한 오버헤드를 발생시키지 않고 바이트 슬라이스를 얻는 것이 바람직합니다.

안전하지 않은 변환

안전하지 않은 패키지를 활용하면 이 목표를 달성할 수 있는 방법이 제공됩니다. 문자열 값을 바이트 배열에 대한 포인터로 캐스팅하면 복사본을 만들지 않고도 기본 바이트 슬라이스에 액세스할 수 있습니다.

<code class="go">func unsafeGetBytes(s string) []byte {
    return (*[0x7fff0000]byte)(unsafe.Pointer(
        (*reflect.StringHeader)(unsafe.Pointer(&s)).Data),
    )[:len(s):len(s)]
}</code>

주의

이 접근 방식에 유의하는 것이 중요합니다. 본질적인 위험을 안고 있습니다. Go의 문자열은 변경할 수 없으므로 unsafeGetBytes를 통해 얻은 바이트 조각을 수정하면 예기치 않은 동작이 발생하거나 데이터가 손상될 수도 있습니다. 따라서 이 기술은 메모리 성능이 가장 중요한 통제된 내부 환경에서만 사용해야 합니다.

빈 문자열 처리

빈 문자열("")에는 바이트가 없으므로 해당 데이터 필드 불확실하다. 코드에서 빈 문자열이 발견될 수 있는 경우 이를 명시적으로 확인하는 것이 중요합니다.

<code class="go">func unsafeGetBytes(s string) []byte {
    if s == "" {
        return nil // or []byte{}
    }
    return (*[0x7fff0000]byte)(unsafe.Pointer(
        (*reflect.StringHeader)(unsafe.Pointer(&s)).Data),
    )[:len(s):len(s)]
}</code>

성능 고려 사항

이 변환을 통해 복사 오버헤드를 피할 수 있지만 압축이 필요하다는 점을 명심해야 합니다. gzipWriter를 사용하여 언급한 것과 같은 작업은 계산 집약적입니다. 메모리 복사를 피함으로써 얻을 수 있는 잠재적인 성능 향상은 압축에 필요한 계산에 비해 무시할 수 있습니다.

대체 접근 방식

또는 io.WriteString 함수를 활용하여 io에 문자열을 쓸 수 있습니다. .Writer는 복사 작업을 호출하지 않습니다. 이 함수는 io.Writer에서 WriteString 메서드가 있는지 확인하고 가능한 경우 이를 호출합니다.