「有问必答」秒杀系统 Go并发编程实践！(1)，Golang知识体系大纲

2401_84263208

于 2024-04-13 16:29:53 发布

阅读量767

点赞数 6

分类专栏：程序员文章标签： golang 服务器开发语言

本文链接：https://blog.csdn.net/2401_84263208/article/details/137718470

版权

程序员专栏收录该内容

17 篇文章 0 订阅

订阅专栏

计数器（sync.WaitGroup）：用于等待所有请求完成。

3. 实现步骤

下面是我们实现秒杀系统的关键步骤：

3.1 初始化库存

在系统启动时，我们需要初始化商品的库存。

var stock = 100 // 商品库存
var mu sync.Mutex

3.2 处理秒杀请求

当客户端发送秒杀请求时，服务器需要处理请求并更新库存。

func handleRequest(user int) {
defer wg.Done()
if tryAcquireLock() {
if stock > 0 {
// 执行秒杀逻辑
stock–
fmt.Printf(“用户%d秒杀成功，剩余库存：%d\n”, user, stock)
} else {
fmt.Printf(“用户%d秒杀失败，库存不足\n”, user)
}
releaseLock()
} else {
fmt.Printf(“用户%d未获取到锁，秒杀失败\n”, user)
}
}

3.3 并发控制和等待

为了控制并发请求的数量，我们使用计数器和通道来限制并发度。

var wg sync.WaitGroup

func main() {
for i := 1; i <= 1000; i++ {
wg.Add(1)
go handleRequest(i)
}
wg.Wait()
}

3.4 互斥锁和并发安全

为了保证并发访问的安全性，我们使用互斥锁来保护共享资源的访问。

注意：TryLock()是go1.18才引入的

func tryAcquireLock() bool {
return mu.TryLock()
}

func releaseLock() {
mu.Unlock()
}

4. 完整代码

package main

import (
“fmt”
“sync”
)

//后面开启了1000个goroutine，所以这里channel的缓冲区设置成了1000
var ch = make(chan bool, 1000)

type Product struct {
sync.Mutex
stock int64 // 商品库存
}

func main() {
p := Product{stock: 1000}
for i := 1; i <= 1000; i++ {
go p.handleRequest(i)
}
<-ch
}

func (p *Product) handleRequest(user int) {
if p.tryAcquireLock() {
if p.stock > 0 {
// 执行秒杀逻辑
p.stock–
fmt.Printf(“用户%d秒杀成功，剩余库存：%d\n”, user, p.stock)
} else {
fmt.Printf(“用户%d秒杀失败，库存不足\n”, user)
}
//这里是不可以使用defer的，因为可能会加锁失败，unlock一个不存在的锁
p.releaseLock()
} else {
fmt.Printf(“用户%d未获取到锁，秒杀失败\n”, user)
}
}