golang创建线程

发布时间:2024-11-21 21:34:35

在Go语言中，创建并发程序是非常简单的。通过使用goroutine（协程），我们可以轻松地创建并发执行的线程。Goroutine是Go语言中的一种轻量级线程，相比于传统线程，它的启动时间和内存开销都要小很多。下面将介绍goroutine的基本用法以及如何控制它们的执行。

1. 创建goroutine

在Go语言中，我们可以使用关键字go来创建一个goroutine。一个goroutine就是一个独立的执行单元，它是由Go运行时负责调度的。启动一个goroutine非常简单，只需要在函数调用前面加上go关键字即可。下面是一个简单的例子：

func main() {
    go printHello()
    // 其他逻辑代码
}
 
func printHello() {
    fmt.Println("Hello, goroutine!")
}

在上面的例子中，我们使用go关键字启动了一个printHello()函数的goroutine。printHello()函数会在一个新的goroutine中运行，而main()函数会在主goroutine中继续执行其他逻辑代码。

2. 等待goroutine完成

在许多情况下，我们希望等待一个goroutine运行完成后再继续执行后续代码。Go语言提供了一种简单的方式来实现这个目标，那就是使用sync包中的WaitGroup。WaitGroup用于等待一组goroutine的完成。

import (
    "fmt"
    "sync"
)
 
func main() {
    var wg sync.WaitGroup
    wg.Add(1)
    go printHello(&wg)
    wg.Wait()
    // 其他逻辑代码
}
 
func printHello(wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Println("Hello, goroutine!")
}

在上面的例子中，我们首先创建了一个WaitGroup对象wg。然后，在printHello()函数的goroutine中，使用defer关键字注册了一个函数wg.Done()，表示goroutine执行完成时调用。最后，在main()函数中，使用wg.Add(1)增加了等待的goroutine数量，并使用wg.Wait()等待所有goroutine都执行完成。

3. 控制goroutine的执行

在有些情况下，我们希望能够更精确地控制goroutine的执行。Go语言提供了一些机制来实现这个目标。

3.1 通道（Channel）

通道是用来传递数据的管道，它可以将数据从一个goroutine发送到另一个goroutine。通道提供了阻塞机制，当通道为空或者通道已满时，发送和接收操作将会阻塞。

import "fmt"
 
func main() {
    ch := make(chan int)
    go sendData(ch)
    // 接收通道数据
    fmt.Println(<-ch)
}
 
func sendData(ch chan int) {
    ch <- 1
}

在上面的例子中，我们首先使用make函数创建了一个通道ch。然后，在sendData()函数的goroutine中，使用ch <- 1将数据1发送到通道ch中。最后，在main()函数中，使用<-ch接收通道数据，并将结果打印出来。

3.2 互斥锁（Mutex）

在并发程序中，多个goroutine同时访问共享的资源可能会导致数据竞争，从而产生不确定的结果。为了解决这个问题，Go语言提供了互斥锁来保护临界区的代码。

import (
    "fmt"
    "sync"
)
 
var counter int
var mutex sync.Mutex
 
func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go increment(&wg)
    }
    wg.Wait()
    fmt.Println(counter)
}
 
func increment(wg *sync.WaitGroup) {
    defer wg.Done()
    mutex.Lock()
    counter++
    mutex.Unlock()
}

在上面的例子中，我们首先定义了一个互斥锁mutex，并在increment()函数中使用mutex.Lock()和mutex.Unlock()来保护counter变量的修改。通过使用互斥锁，我们确保了同一时间只有一个goroutine可以访问counter变量，从而避免了数据竞争的问题。