Go语言利用channel模拟并发售票

最新推荐文章于 2023-05-28 21:22:07 发布

原创最新推荐文章于 2023-05-28 21:22:07 发布 · 423 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#go

本文通过Go语言实现了一个简单的并发售票程序，使用全局chan存储票总数，并通过多个goroutine模拟不同的售票口进行售票。程序中考虑了票售罄的情况，并在票售完后关闭通道。

废话不多说，直接上代码：

// 定义全局chan，存储票总数
var totalTickets chan int
var wg sync.WaitGroup

func main() {
	// 初始化票数量：总票数10张
	totalTickets = make(chan int, 2)
	totalTickets <- 10

	wg.Add(5)

	go sell("售票口1")
	go sell("售票口2")
	go sell("售票口3")
	go sell("售票口4")
	go sell("售票口5")

	wg.Wait()

	fmt.Println("main over.")
}

func sell(name string) {
	defer wg.Done()

	rand.Seed(time.Now().UnixNano())
	for {
		residue, ok := <-totalTickets
		if !ok {
			fmt.Printf("%s: Sold Out 1\n", name)
			break
		}
		if residue > 0 {
			time.Sleep(time.Duration(rand.Intn(1000)) * time.Millisecond)
			totalTickets <- residue - 1
			fmt.Println(name, "售出1张票，余票：", residue)
		} else {
			fmt.Printf("%s: Sold Out 2\n", name)
			close(totalTickets)
			break
		}
	}
}