string to bytes, bytes to string

reference to golang data structure
go-extend/blob/master/exstrings/convert.go

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 将[]byte零成本转换成string, 转换后的string
// 会因为调用该函数时len(b)长度内的内容的改变而改变
func Bytes2Str(b []byte) string {
	return *(*string)(unsafe.Pointer(&b))
}

the tests

import (
	"math/rand"
	"reflect"
	"testing"
)

func TestStr2Bytes(t *testing.T) {
	strSample := []string{
		"你",
		"好",
		"世",
		"界",
		"语",
		"言",
		"变",
		"化",
		"hello",
		"world",
		"language",
		"change",
	}
	type args struct {
		s string
	}
	tests := []struct {
		name string
		args args
		want []byte
	}{
		{
			name: "abb",
			args: args{
				s: "abb",
			},
			want: []byte{'a', 'b', 'b'},
		},
	}
	for i := 0; i < 100; i++ {
		name := ""
		for j := 0; j < i+10; j++ {
			name += strSample[rand.Intn(len(strSample))]
		}
		tests = append(tests, struct {
			name string
			args args
			want []byte
		}{
			name: name,
			args: args{
				s: name,
			},
			want: ([]byte)(name),
		})
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := Str2Bytes(tt.args.s); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("Str2Bytes() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestBytes2Str(t *testing.T) {
	strSample := []string{
		"你",
		"好",
		"世",
		"界",
		"语",
		"言",
		"变",
		"化",
		"hello",
		"world",
		"language",
		"change",
	}
	type args struct {
		b []byte
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{
			name: "abb",
			args: args{
				b: []byte{'a', 'b', 'b'},
			},
			want: "abb",
		},
	}
	for i := 0; i < 100; i++ {
		name := ""
		for j := 0; j < i+10; j++ {
			name += strSample[rand.Intn(len(strSample))]
		}
		tests = append(tests, struct {
			name string
			args args
			want string
		}{
			name: name,
			args: args{
				b: ([]byte)(name),
			},
			want: (name),
		})
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := Bytes2Str(tt.args.b); got != tt.want {
				t.Errorf("Bytes2Str() = %v, want %v", got, tt.want)
			}
		})
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值