几个提升Go语言开发效率的小技巧

2年前 (2022) 程序员胖胖胖虎阿

210 0 0

前言

哈喽，大家好，我是asong。

每门语言都有自己的语法糖，像java的语法糖就有方法变长参数、拆箱与装箱、枚举、for-each等等，Go语言也不例外，其也有自己的语法糖，掌握这些语法糖可以助我们提高开发的效率，所以本文就来介绍一些Go语言的语法糖，总结的可能不能全，欢迎评论区补充。

可变长参数

Go语言允许一个函数把任意数量的值作为参数，Go语言内置了...操作符，在函数的最后一个形参才能使用...操作符，使用它必须注意如下事项：

可变长参数必须在函数列表的最后一个；
把可变长参数当切片来解析，可变长参数没有没有值时就是nil切片
可变长参数的类型必须相同

func test(a int, b ...int){
  return
}

既然我们的函数可以接收可变长参数，那么我们在传参的时候也可以传递切片使用...进行解包转换为参数列表，append方法就是最好的例子：

var sl []int
sl = append(sl, 1)
sl = append(sl, sl...)

append方法定义如下：

//    slice = append(slice, elem1, elem2)
//    slice = append(slice, anotherSlice...)
func append(slice []Type, elems ...Type) []Type

声明不定长数组

数组是有固定长度的，我们在声明数组时一定要声明长度，因为数组在编译时就要确认好其长度，但是有些时候对于想偷懒的我，就是不想写数组长度，有没有办法让他自己算呢？当然有，使用...操作符声明数组时，你只管填充元素值，其他的交给编译器自己去搞就好了；

a := [...]int{1, 3, 5} // 数组长度是3，等同于 a := [3]{1, 3, 5}

有时我们想声明一个大数组，但是某些index想设置特别的值也可以使用...操作符搞定：

a := [...]int{1: 20, 999: 10} // 数组长度是100, 下标1的元素值是20，下标999的元素值是10，其他元素值都是0

`init`函数

Go语言提供了先于main函数执行的init函数，初始化每个包后会自动执行init函数，每个包中可以有多个init函数，每个包中的源文件中也可以有多个init函数，加载顺序如下：

从当前包开始，如果当前包包含多个依赖包，则先初始化依赖包，层层递归初始化各个包，在每一个包中，按照源文件的字典序从前往后执行，每一个源文件中，优先初始化常量、变量，最后初始化init函数，当出现多个init函数时，则按照顺序从前往后依次执行，每一个包完成加载后，递归返回，最后在初始化当前包！

init函数实现了sync.Once，无论包被导入多少次，init函数只会被执行一次，所以使用init可以应用在服务注册、中间件初始化、实现单例模式等等，比如我们经常使用的pprof工具，他就使用到了init函数，在init函数里面进行路由注册：

//go/1.15.7/libexec/src/cmd/trace/pprof.go
func init() {
 http.HandleFunc("/io", serveSVGProfile(pprofByGoroutine(computePprofIO)))
 http.HandleFunc("/block", serveSVGProfile(pprofByGoroutine(computePprofBlock)))
 http.HandleFunc("/syscall", serveSVGProfile(pprofByGoroutine(computePprofSyscall)))
 http.HandleFunc("/sched", serveSVGProfile(pprofByGoroutine(computePprofSched)))

 http.HandleFunc("/regionio", serveSVGProfile(pprofByRegion(computePprofIO)))
 http.HandleFunc("/regionblock", serveSVGProfile(pprofByRegion(computePprofBlock)))
 http.HandleFunc("/regionsyscall", serveSVGProfile(pprofByRegion(computePprofSyscall)))
 http.HandleFunc("/regionsched", serveSVGProfile(pprofByRegion(computePprofSched)))
}

忽略导包

Go语言在设计师有代码洁癖，在设计上尽可能避免代码滥用，所以Go语言的导包必须要使用，如果导包了但是没有使用的话就会产生编译错误，但有些场景我们会遇到只想导包，但是不使用的情况，比如上文提到的init函数，我们只想初始化包里的init函数，但是不会使用包内的任何方法，这时就可以使用 _ 操作符号重命名导入一个不使用的包：

import _ "github.com/asong"

忽略字段

在我们日常开发中，一般都是在屎上上堆屎，遇到可以用的方法就直接复用了，但是这个方法的返回值我们并不一定都使用，还要绞尽脑汁的给他想一个命名，有没有办法可以不处理不要的返回值呢？当然有，还是 _ 操作符，将不需要的值赋给空标识符：

_, ok := test(a, b int)

json序列化忽略某个字段

大多数业务场景我们都会对struct做序列化操作，但有些时候我们想要json里面的某些字段不参加序列化，-操作符可以帮我们处理，Go语言的结构体提供标签功能，在结构体标签中使用 - 操作符就可以对不需要序列化的字段做特殊处理，使用如下：

type Person struct{
  name string `json:"-"`
  age string `json: "age"`
}

json序列化忽略空值字段

我们使用json.Marshal进行序列化是不会忽略struct中的空值，默认输出字段的类型零值（string类型零值是""，对象类型的零值是nil...），如果我们想在序列化是忽略掉这些没有值的字段时，可以在结构体标签中中添加omitempty tag：

type User struct {
    Name  string   `json:"name"`
    Email string   `json:"email,omitempty"`
  Age int        `json: "age"`
}

func test() {
    u1 := User{
        Name: "asong",
    }
    b, err := json.Marshal(u1)
    if err != nil {
        fmt.Printf("json.Marshal failed, err:%v\n", err)
        return
    }
    fmt.Printf("str:%s\n", b)
}

运行结果：

str:{"name":"asong","Age":0}

Age字段我们没有添加omitempty tag在json序列化结果就是带空值的，email字段就被忽略掉了；

短变量声明

每次使用变量时都要先进行函数声明，对于我这种懒人来说是真的不想写，因为写python写惯了，那么在Go语言是不是也可以不进行变量声明直接使用呢？我们可以使用 name := expression 的语法形式来声明和初始化局部变量，相比于使用var声明的方式可以减少声明的步骤：

var a int = 10
等用于
a := 10

使用短变量声明时有两个注释事项：

短变量声明只能在函数内使用，不能用于初始化全局变量
短变量声明代表引入一个新的变量，不能在同一作用域重复声明变量
多变量声明中如果其中一个变量是新变量，那么可以使用短变量声明，否则不可重复声明变量；

类型断言

我们通常都会使用interface，一种是带方法的interface，一种是空的interface，Go1.18之前是没有泛型的，所以我们可以用空的interface{}来作为一种伪泛型使用，当我们使用到空的interface{}作为入参或返回值时，就会使用到类型断言，来获取我们所需要的类型，在Go语言中类型断言的语法格式如下：

value, ok := x.(T)
or
value := x.(T)

x是interface类型，T是具体的类型，方式一是安全的断言，方式二断言失败会触发panic；这里类型断言需要区分x的类型，如果x是空接口类型：

空接口类型断言实质是将eface中_type与要匹配的类型进行对比，匹配成功在内存中组装返回值，匹配失败直接清空寄存器，返回默认值。

如果x是非空接口类型：

非空接口类型断言的实质是 iface 中 *itab 的对比。*itab 匹配成功会在内存中组装返回值。匹配失败直接清空寄存器，返回默认值。

具体源码剖析可以看这篇文章：源码剖析类型断言是如何实现的！附性能损耗测试

切片循环

切片/数组是我们经常使用的操作，在Go语言中提供了for range语法来快速迭代对象，数组、切片、字符串、map、channel等等都可以进行遍历，总结起来总共有三种方式：

// 方式一：只遍历不关心数据，适用于切片、数组、字符串、map、channel
for range T {}

// 方式二：遍历获取索引或数组，切片，数组、字符串就是索引，map就是key，channel就是数据
for key := range T{}

// 方式三：遍历获取索引和数据，适用于切片、数组、字符串，第一个参数就是索引，第二个参数就是对应的元素值，map 第一个参数就是key，第二个参数就是对应的值；
for key, value := range T{}

判断map的key是否存在

Go语言提供语法 value, ok := m[key]来判断map中的key是否存在，如果存在就会返回key所对应的值，不存在就会返回空值：

import "fmt"

func main() {
    dict := map[string]int{"asong": 1}
    if value, ok := dict["asong"]; ok {
        fmt.Printf(value)
    } else {
      fmt.Println("key:asong不存在")
    }
}

select控制结构

Go语言提供了select关键字，select配合channel能够让Goroutine同时等待多个channel读或者写，在channel状态未改变之前，select会一直阻塞当前线程或Goroutine。先看一个例子：

func fibonacci(ch chan int, done chan struct{}) {
 x, y := 0, 1
 for {
  select {
  case ch <- x:
   x, y = y, x+y
  case <-done:
   fmt.Println("over")
   return
  }
 }
}
func main() {
 ch := make(chan int)
 done := make(chan struct{})
 go func() {
  for i := 0; i < 10; i++ {
   fmt.Println(<-ch)
  }
  done <- struct{}{}
 }()
 fibonacci(ch, done)
}

select与switch具有相似的控制结构，与switch不同的是，select中的case中的表达式必须是channel的收发操作，当select中的两个case同时被触发时，会随机执行其中的一个。为什么是随机执行的呢？随机的引入就是为了避免饥饿问题的发生，如果我们每次都是按照顺序依次执行的，若两个case一直都是满足条件的，那么后面的case永远都不会执行。

上面例子中的select用法是阻塞式的收发操作，直到有一个channel发生状态改变。我们也可以在select中使用default语句，那么select语句在执行时会遇到这两种情况：