Golang channel 用法与实现原理

2023-02-23 09:28:09 浏览数 (1)

文章目录

  • 1.简介
  • 2.用法
  • 3.三种状态
  • 4.实现原理
    • 数据结构
    • 原理概述
  • 5.小结
  • 参考文献

1.简介

Golang channel 是一种并发原语,用于在不同 goroutine 之间进行通信和同步。本质上,channel 是一种类型安全的 FIFO 队列,它可以实现多个 goroutine 之间的同步和通信。

channel 是一种引用类型,即使是在不同的 goroutine 之间传递channel时,它们仍然指向相同的底层数据结构。

2.用法

在 Golang 中,可以使用 make 函数创建 channel,语法如下:

代码语言:javascript复制
ch := make(chan T)

其中,T 表示 channel 中元素的类型。

channel 可以使用以下操作来实现同步和通信:

  • 发送操作: 用于将数据发送到channel中,语法为 ch <- data。
  • 接收操作: 用于从channel中接收数据,语法为 data := <- ch。
  • 关闭操作: 用于关闭 channel,语法为 close(ch)。

需要注意的是,发送和接收操作都是阻塞的,即如果没有 goroutine 同时进行对应的操作,它们将一直阻塞,直到其他 goroutine 进行操作为止。而关闭操作是非阻塞的。即使 channel 已经被关闭,仍然可以从中读取数据。

3.三种状态

channel 有三种状态:未关闭,已关闭和 nil。

对三种不同的 channel 进行操作,会有不同的结果。

操作

未关闭

已关闭

nil

发送

阻塞或成功发送

panic

永久阻塞

读取

阻塞或成功读取

成功读取或返回零值

永久阻塞

关闭

成功关闭

panic

panic

4.实现原理

数据结构

channel 是一个结构体,运行时使用 runtime.hchan(go 1.19 runtime/chan.go)结构体表示。

代码语言:javascript复制
type hchan struct {
	qcount   uint           // total data in the queue
	dataqsiz uint           // size of the circular queue
	buf      unsafe.Pointer // points to an array of dataqsiz elements
	elemsize uint16
	closed   uint32
	elemtype *_type // element type
	sendx    uint   // send index
	recvx    uint   // receive index
	recvq    waitq  // list of recv waiters
	sendq    waitq  // list of send waiters

	// lock protects all fields in hchan, as well as several
	// fields in sudogs blocked on this channel.
	//
	// Do not change another G's status while holding this lock
	// (in particular, do not ready a G), as this can deadlock
	// with stack shrinking.
	lock mutex
}

字段的具体含义如下:

qcount:表示当前队列中元素的个数。 dataqsiz:表示队列中元素的个数上限,即缓冲区大小。 buf:指向缓存的环形数组,存储实际元素的值。由于 channel 内的元素在堆上分配,因此 buf 是一个 unsafe.Pointer 类型的指针。 elemsize:表示单个元素的大小,以字节为单位。 closed:表示 channel 是否已关闭,0 表示未关闭,1 表示已关闭。 recvx:表示下一个被接收的元素在 buf 中的位置。 sendx:表示下一个被发送的元素在 buf 中的位置。 recvq:接收者的等待队列,用于存储等待从 channel 中读取数据的 goroutine。 sendq:发送者的等待队列,用于存储等待向 channel 中发送数据的 goroutine。 lock:保护 channel 的锁,防止多个 goroutine 同时访问 channel 时发生竞争条件。这里的 lock 是一个 mutex 类型的变量。

需要注意的是,这里的 waitq 和 mutex 分别表示等待队列和互斥锁,是 Golang 内部实现 channel 同步和通信机制所需要的结构体,由 Golang 运行时库提供支持。

原理概述

在 Golang 中,channel 是用于协程之间通信的重要机制,其内部实现涉及到以下几个方面:

数据结构:channel 本质上是一个带有同步功能的队列,Go 语言中的 channel 通过数据结构来实现同步和通信。具体而言,channel 内部包含了一个指向队列数据的指针和两个指向队列头和尾的索引。

内存管理:channel 内部存储的元素是在堆上分配的,这意味着在使用 channel 时不用考虑内存分配和回收的问题,由 Go 运行时自动管理。

同步机制:channel 的本质是一种同步机制,因此其内部实现必须包括同步相关的机制,以确保通信的正确性。Go 语言采用了类似于信号量的方法实现 channel 的同步,即在发送和接收操作时使用锁和条件变量来实现同步。

调度器:Golang 中的调度器负责协程的调度和管理,其在 channel 的实现中起到了重要的作用。调度器通过在不同协程之间切换来实现 channel 的通信和同步。

具体来说,当一个协程试图向 channel 发送数据时,调度器会检查 channel 的缓冲区状态。如果 channel 的缓冲区未满,则将数据写入缓冲区并唤醒等待接收的协程。如果 channel 的缓冲区已满,则当前协程会被阻塞,等待其他协程取走缓冲区中的数据。

当一个协程试图从 channel 中接收数据时,调度器会检查 channel 的缓冲区状态。如果 channel 的缓冲区非空,则将缓冲区中的数据读出并唤醒等待发送的协程。如果 channel 的缓冲区为空,则当前协程会被阻塞,等待其他协程向缓冲区中写入数据。

在 channel 的实现中,Go 语言使用了类似于操作系统中的管道机制,以及用于进程间通信的信号量机制,通过同步、调度等多种机制实现了协程之间的通信和同步。

5.小结

总的来说,channel是Golang中非常重要的并发原语,它为多个 goroutine 之间的通信和同步提供了方便而高效的方式。使用channel需要注意它的阻塞特性以及引用类型的特点。channel的底层实现基于管道,使用两个goroutine之间的无缓冲管道来进行通信,确保操作的顺序满足FIFO的规则。


0 人点赞