Go 性能优化

string 与 []byte 互转

利用了底层 string 和 byte slice 实现的技巧，如果需要大量互转可以使用这种方式。

  1. /*
  2. type StringHeader struct {
  3.     Data uintptr
  4.     Len  int
  5. }
  6. type SliceHeader struct {
  7.     Data uintptr
  8.     Len  int
  9.     Cap  int
  10. }
  11. */
  13. func str2bytes(s string) []byte {
  14.    x := (*[2]uintptr)(unsafe.Pointer(&s))
  15.    b := [3]uintptr{x[0], x[1], x[1]}
  16.    return *(*[]byte)(unsafe.Pointer(&b))
  17. }
  19. func bytes2str(b []byte) string {
  20.    return *(*string)(unsafe.Pointer(&b))
  21. }

字符串拼接

大量字符串拼接不要用 + ，使用 bytes.Buffer 或者 strings.Builder

  1. // bytes.Buffer
  2. package main
  4. import (
  5.     "fmt"
  6.     "bytes"
  7. )
  9. func main() {
  10.     var b bytes.Buffer
  12.     b.WriteString("abc")
  13.     b.WriteString("def")
  15.     fmt.Println(b.String()) // abcdef
  16. }
  1. // strings.Builder
  2. package main
  4. import (
  5.     "fmt"
  6.     "strings"
  7. )
  9. func main() {
  10.     var sb strings.Builder
  11.     sb.WriteString("First")
  12.     sb.WriteString("Second")
  13.     fmt.Println(sb.String())    // FirstSecond
  14. }