RepoMirrors/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus synced 2025-03-11 07:59:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Neater string vs byte-slice conversions (#14425)

Browse Source 
unsafe.Slice and unsafe.StringData were added in Go 1.20

Signed-off-by: Bryan Boreham <bjboreham@gmail.com>
This commit is contained in:
Bryan Boreham 2024-09-21 11:19:21 +01:00 committed by GitHub
parent 6bcb064d93
commit 31c5760551
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model/labels/labels_common.go
View File

@ -230,5 +230,5 @@ func contains(s []Label, n string) bool {
}
func yoloString(b []byte) string {
return *((*string)(unsafe.Pointer(&b)))
return unsafe.String(unsafe.SliceData(b), len(b))
}

11
model/labels/labels_stringlabels.go
View File

@ -16,7 +16,6 @@
package labels
import (
"reflect"
"slices"
"strings"
"unsafe"
@ -299,10 +298,8 @@ func Equal(ls, o Labels) bool {
func EmptyLabels() Labels {
return Labels{}
}
func yoloBytes(s string) (b []byte) {
*(*string)(unsafe.Pointer(&b)) = s
(*reflect.SliceHeader)(unsafe.Pointer(&b)).Cap = len(s)
return
func yoloBytes(s string) []byte {
return unsafe.Slice(unsafe.StringData(s), len(s))
}
// New returns a sorted Labels from the given labels.
@ -338,8 +335,8 @@ func Compare(a, b Labels) int {
}
i := 0
// First, go 8 bytes at a time. Data strings are expected to be 8-byte aligned.
sp := unsafe.Pointer((*reflect.StringHeader)(unsafe.Pointer(&shorter)).Data)
lp := unsafe.Pointer((*reflect.StringHeader)(unsafe.Pointer(&longer)).Data)
sp := unsafe.Pointer(unsafe.StringData(shorter))
lp := unsafe.Pointer(unsafe.StringData(longer))
for ; i < len(shorter)-8; i += 8 {
if *(*uint64)(unsafe.Add(sp, i)) != *(*uint64)(unsafe.Add(lp, i)) {
break

2
model/textparse/promparse.go
View File

@ -502,7 +502,7 @@ func unreplace(s string) string {
}
func yoloString(b []byte) string {
return *((*string)(unsafe.Pointer(&b)))
return unsafe.String(unsafe.SliceData(b), len(b))
}
func parseFloat(s string) (float64, error) {

4
tsdb/encoding/encoding.go
View File

@ -20,7 +20,6 @@ import (
"hash"
"hash/crc32"
"math"
"unsafe"
"github.com/dennwc/varint"
)
@ -75,8 +74,7 @@ func (e *Encbuf) PutVarint64(x int64) {
// PutUvarintStr writes a string to the buffer prefixed by its varint length (in bytes!).
func (e *Encbuf) PutUvarintStr(s string) {
b := *(*[]byte)(unsafe.Pointer(&s))
e.PutUvarint(len(b))
e.PutUvarint(len(s))
e.PutString(s)
}

2
tsdb/index/index.go
View File

@ -2063,5 +2063,5 @@ func (dec *Decoder) Series(b []byte, builder *labels.ScratchBuilder, chks *[]chu
}
func yoloString(b []byte) string {
return *((*string)(unsafe.Pointer(&b)))
return unsafe.String(unsafe.SliceData(b), len(b))
}