如何优雅的关闭channel？

一、channel使用存在的不方便地方

1、在不改变channel自身状态的情况下，无法获知一个channnel是否关闭。

2、关闭一个已经关闭的channel，会导致panic。因此，如果关闭channel的一方在不知道channel是否关闭状态时就去贸然关闭channel是件很危险的事。

3、向一个已经关闭的channel发送数据会导致panic。因此，如果向channel发送数据的一方不知道channel是否处于关闭状态就贸然向channel发送数据是很危险的事情。

一个比较粗糙的检查channel是否关闭的函数：

func IsClosed(ch <-chan interface{}) bool {
    select {
    case <-ch:
        return true
    default:
    }
    return false
}

func main() {
    c := make(chan interface{})
    fmt.Println(IsClosed(c))  // false
    close(c)
    fmt.Println(IsClosed(c)) // true
}

上面的代码其实存在很多问题。

首先，IsClosed函数是一个有副作用的函数。每调用一次，都会读出channel里面的一个元素，改变了channel的状态，这不是一个好函数，干活就干活，还顺手牵羊？

其次，IsClosed函数返回的结果仅代表调用时候的那个瞬间，并不能保证调用之后不会有其他goroutine对这个channel进行了一些操作，改变了这个channel的状态。比如：IsClosed函数返回true，但这时有另外一个goroutine关闭了这个channel，这时候我们就会拿着这个过时的"channel未关闭"信息，向其发送数据，就会导致panic的发生。

当然，一个channel不会被重复关闭两次，如果IsClosed函数返回的结果是true,说明channel是真的关闭了。

有一句广泛流传的关闭channel的原则：

不要从一个 receiver 侧关闭 channel，也不要在有多个 sender 时，关闭 channel。

向channel发送元素的就是sender，因此sender可以决定何时不发送数据，并且关闭channel。但是如果有多个sender，某个sender同样没法确定其他sender的情况。这时候也不能贸然的关闭channel。

有两个不优雅关闭channel的方法：

1、使用defer-recover机制，放心大胆地关闭channel或者向channel发送数据，即使发生了panic，有defer-recover在兜底。

2、使用sync.Once来保证只关闭一次。

二、如何优雅关闭channel

根据sender和receiver的个数，可以分为以下几种情况：

1、一个sender，一个receiver

2、一个sender，M个receiver

3、N个sender，一个receiver

4、N个sender，M个receiver

对于1，2这两种情况，只有一个sender，直接从sender段关闭就好。

第3种情形下，优雅关闭channel的方法是：唯一的receiver发出一个关闭channel的信号，senders监听到关闭信号后，停止发送数据。

func main() {
    rand.Seed(time.Now().UnixNano())

    const Max = 10000
    const NumSenders = 1000

    dataChan := make(chan int, 100)
    stopChan := make(chan struct{})

    // senders
    for i := 0; i < NumSenders; i++ {
        go func() {
            for {
                select {
                // 监听关闭channel的信号，退出
                case <-stopChan:
                    return
                // 给dataChan中发送数据
                case dataChan <- rand.Intn(Max):
                }
            }
        }()
    }

    // receiver
    go func() {
        // 从dataChan中遍历数据
        for value := range dataChan {
            // 如果value的值为9999，通知sender停止发送数据
            if value == Max-1 {
                fmt.Println("send stop signal to senders.")
                close(stopChan)
                return
            }
            fmt.Println(value)
        }
    }()

    select {
    // 阻塞一小时退出
    case <-time.After(time.Hour):
    }
}

这里的stopCh就是信号channel，它本身只有一个sender，因此可以直接关闭它。senders收到了关闭信号后，select分支case<-stopCh被选中，退出函数，不再发送数据。

需要说明的是，上面的代码并没有明确关闭dataCh。在Go语言中，对于一个channel，如果最终没有任何goroutine引用它，不管channel有没有被关闭，最终都会被GC回收。所以在这种情形下，所谓的优雅地关闭channel就是不关闭channel，让GC代劳。

最后一种情况，优雅关闭channel的方法是：any one of them says “let’s end the game” by notifying a moderator to close an additional signal channel。

和第三种情况不同，这里有M个receiver，如果，采用第3种解决方案，由receiver直接关闭stopCh的话，就会重复关闭一个channel，导致panic。因此需要增加一个中间人，M个receiver都向它发送关闭dataCh的"请求"，中间人收到第一个请求后，就会直接下达关闭dataCh的指令（通过关闭stopCh,这时就不会发生重复关闭的情况，因为stopCh的发送方只有中间人一个）。另外这里N个sender也可以向中间人发送关闭dataCh的请求。

func main() {
    rand.Seed(time.Now().UnixNano())

    const Max = 10000
    const NumReceivers = 10
    const NumSenders = 1000

    dataCh := make(chan int, 100)
    stopCh := make(chan struct{})

    // 这里必须是一个有缓冲的channel
    toStop := make(chan string, 1)

    var stoppedBy string

    // 中间人
    go func() {
        // 接收到了关闭channel的请求
        stoppedBy = <-toStop
        // 发送关闭dataCh的信号
        close(stopCh)
    }()

    // senders
    for i := 0; i < NumSenders; i++ {
        go func(id string) {
            for {
                value := rand.Intn(Max)
                // 如果value=0，则给中间人发送关闭channel的信号
                if value == 0 {
                    select {
                    case toStop <- "sender#" + id:
                    // 此处是为了防止toStop这个channel阻塞
                    default:
                    }
                    return
                }

                select {
                // 监听关闭channel的信号
                case <-stopCh:
                    return
                case dataCh <- value:
                }
            }
        }(strconv.Itoa(i))
    }

    // receivers
    for i := 0; i < NumReceivers; i++ {
        go func(id string) {
            for {
                select {
                // 监听关闭信号，退出，停止接收数据
                case <-stopCh:
                    return
                // 接收数据
                case value := <-dataCh:
                    // 如果接收到了9999，则给中间人发送关闭channel的信号
                    if value == Max-1 {
                        select {
                        case toStop <- "receiver#" + id:
                        default:
                        }
                        return
                    }
                    fmt.Println(value)
                }
            }
        }(strconv.Itoa(i))
    }

    select {
    case <-time.After(time.Hour):
    }

}

代码里toStop就是中间人的角色，使用它来接收senders和receivers发送过来的关闭dataCh请求。

这里将toStop声明成了一个缓冲型的channel。假设toStop声明的是一个非缓冲型的channel，那么第一个发送的关闭dataCh请求可能会丢失。因为无论是sender还是receiver都是通过select语句来发送请求，如果中间人所在的goroutine没有准备好，那么select语句就不会被选中，直接走default选项了，什么都不做。这样，第一个关闭dataCh的请求就会丢失。

如果把toStop的容量声明成Num(senders) + Num(receivers)，那么发送dataCh请求的部分可以改写成更简洁的形式：

func main() {
	rand.Seed(time.Now().UnixNano())

	const Max = 10000
	const NumReceivers = 10
	const NumSenders = 1000

	dataCh := make(chan int, 100)
	stopCh := make(chan struct{})

	// 这里必须是一个有缓冲的channel
	toStop := make(chan string, NumSenders+NumReceivers)

	var stoppedBy string

	// 中间人
	go func() {
		// 接收到了关闭channel的请求
		stoppedBy = <-toStop
		// 发送关闭dataCh的信号
		close(stopCh)
	}()

	// senders
	for i := 0; i < NumSenders; i++ {
		go func(id string) {
			for {
				value := rand.Intn(Max)
				// 如果value=0，则给中间人发送关闭channel的信号
				if value == 0 {
					toStop <- "sender#" + id
					return
				}

				select {
				// 监听关闭channel的信号
				case <-stopCh:
					return
				case dataCh <- value:
				}
			}
		}(strconv.Itoa(i))
	}

	// receivers
	for i := 0; i < NumReceivers; i++ {
		go func(id string) {
			for {
				select {
				// 监听关闭信号，退出，停止接收数据
				case <-stopCh:
					return
				// 接收数据
				case value := <-dataCh:
					// 如果接收到了9999，则给中间人发送关闭channel的信号
					if value == Max-1 {
						toStop <- "receiver#" + id
						return
					}
					fmt.Println(value)
				}
			}
		}(strconv.Itoa(i))
	}

	select {
	case <-time.After(time.Hour):
	}
}

直接向toStop发送请求，因为toStop容量足够大，所以不用担心阻塞，自然也就不用select语句再加一个default case来避免阻塞。

可以看到，这里同样没有真正关闭dataCh,同第3种情况。

参考链接：

如何优雅地关闭 channel | Go 程序员面试笔试宝典