-
Notifications
You must be signed in to change notification settings - Fork 0
/
strbytesconv.go
54 lines (51 loc) · 2.56 KB
/
strbytesconv.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package utils
import "unsafe"
/*
Str2Bytes 把 string 零成本转换为 []byte 没有多余的内存开销。
对转换后的[]byte添加字符, 都会生成一个新的[]byte即一定触发扩容。
这个函数可以提升 string 转 []byte 的性能,并极大的降低内存开销,
但是却相当的危险,对于不明确这个函数的人来说不建议使用该函数。
该函数使用不当会导致程序直接崩溃,且无法恢复。
s := `{"k":"v"}`
b := Str2Bytes(s)
// b[3] = 'k' // unexpected fault address 0x1118180
data := map[string]string{}
err := json.Unmarshal(b, &data)
fmt.Println(data, err)
这是一个使用的例子,如果我们需要转换一个字符串很方便,且开销非常的低。
但是一定要注意,b[3] = 'k' 如果尝试修改获得的 []byte 将直接导致程序崩溃,并且不可能通过 recover() 恢复。
实际上我们可以突破这个限制,这就要了解字符串的一些规则,
下面的例子可以完美运行,并修改字符串:
s := strings.Repeat("A", 3)
b := exstrings.UnsafeToBytes(s)
b[1] = 'B'
b[2] = 'C'
fmt.Println(s, string(b))
非常完美,s和b变量的值都是 ABC, 为什么会这样呢?
这个就是 string 的内存分配方法, 字面量使用这种方式是没有办法修改的,
因为这是在编译时就决定的,编译时会设定字符串的内存数据是只读数据。
如果程序运行时生成的数据,这种数据是可以安全使用该函数的,
但是要当心你的字符串可能会被修改,
比如我们调用 json.Unmarshal(Str2Bytes(s), &data),
如果 json 包里面出现修改输入参数,那么原来的字符串就可能不是你想想的那样。
使用该函数要明确两个事情:
- 确定字符串是否是字面量,或者内存被分配在只读空间上。
- 确保访问该函数结果的函数是否会按照你的意愿访问或修改数据。
虽然它很危险,但是有时间却很有用,如果我们需要大批量转换字符串的大小写,
而且不再需要原字符串,我们可以原地安全的修改字符串。
当然还有更多的使用方法,可以极大的提升我们程序的性能。
*/
func Str2Bytes(s string) []byte {
return *(*[]byte)(unsafe.Pointer(&struct {
string
Cap int
}{s, len(s)}))
}
/*
Bytes2Str 将bytes通过复用底层指针的方式转换成string, 转换后的string会
跟随用于转换的bytes里数据改变而改变, 所以转换后, 原先的bytes最好不再使用,
或者只用只读的方式
*/
func Bytes2Str(b []byte) string {
return *(*string)(unsafe.Pointer(&b))
}