mirror of https://github.com/siddontang/go.git
add hack for nocopy for string <-> slice
This commit is contained in:
parent
ec6187efd9
commit
e02d4a31aa
|
@ -0,0 +1,27 @@
|
|||
package hack
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
// no copy to change slice to string
|
||||
// use your own risk
|
||||
func String(b []byte) (s string) {
|
||||
pbytes := (*reflect.SliceHeader)(unsafe.Pointer(&b))
|
||||
pstring := (*reflect.StringHeader)(unsafe.Pointer(&s))
|
||||
pstring.Data = pbytes.Data
|
||||
pstring.Len = pbytes.Len
|
||||
return
|
||||
}
|
||||
|
||||
// no copy to change string to slice
|
||||
// use your own risk
|
||||
func Slice(s string) (b []byte) {
|
||||
pbytes := (*reflect.SliceHeader)(unsafe.Pointer(&b))
|
||||
pstring := (*reflect.StringHeader)(unsafe.Pointer(&s))
|
||||
pbytes.Data = pstring.Data
|
||||
pbytes.Len = pstring.Len
|
||||
pbytes.Cap = pstring.Len
|
||||
return
|
||||
}
|
|
@ -0,0 +1,36 @@
|
|||
package hack
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestString(t *testing.T) {
|
||||
b := []byte("hello world")
|
||||
a := String(b)
|
||||
|
||||
if a != "hello world" {
|
||||
t.Fatal(a)
|
||||
}
|
||||
|
||||
b[0] = 'a'
|
||||
|
||||
if a != "aello world" {
|
||||
t.Fatal(a)
|
||||
}
|
||||
|
||||
b = append(b, "abc"...)
|
||||
if a != "aello world" {
|
||||
t.Fatal(a)
|
||||
}
|
||||
}
|
||||
|
||||
func TestByte(t *testing.T) {
|
||||
a := "hello world"
|
||||
|
||||
b := Slice(a)
|
||||
|
||||
if !bytes.Equal(b, []byte("hello world")) {
|
||||
t.Fatal(string(b))
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue