Perform conversion from string to []byte according to rule wundergraph#6

at https://golang.org/pkg/unsafe/#Pointer
iv-menshenin · Aug 5, 2020 · ecab8fa · ecab8fa
1 parent 44caf1e
commit ecab8fa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/util.go b/util.go
@@ -9,13 +9,13 @@ func b2s(b []byte) string {
 	return *(*string)(unsafe.Pointer(&b))
 }
 
-func s2b(s string) []byte {
+func s2b(s string) (b []byte) {
 	strh := (*reflect.StringHeader)(unsafe.Pointer(&s))
-	var sh reflect.SliceHeader
+	sh := (*reflect.SliceHeader)(unsafe.Pointer(&b))
 	sh.Data = strh.Data
 	sh.Len = strh.Len
 	sh.Cap = strh.Len
-	return *(*[]byte)(unsafe.Pointer(&sh))
+	return b
 }
 
 const maxStartEndStringLen = 80