「有问必答」秒杀系统 Go并发编程实践！

2. 架构设计

我们的秒杀系统将采用经典的客户端-服务器架构。客户端发送秒杀请求，服务器处理请求并更新库存。为了保证系统的高并发性能，我们将使用以下技术和原语：

• 互斥锁（sync.Mutex）：用于保护共享资源的并发访问。
• 通道（channel）：用于协程间的通讯。
• 计数器（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)
 }
}

func (p \*Product) tryAcquireLock() bool {
//p.TryLock() 方法用于尝试获取锁，如果成功获取到锁，则相当于执行了 Lock() 操作，即加锁成功。 


![img](https://img-blog.csdnimg.cn/img_convert/b08e8dcc07da2728c4bce923f0105f73.png)
![img](https://img-blog.csdnimg.cn/img_convert/122783bdd431871efa0b6a5650843bb7.png)
![img](https://img-blog.csdnimg.cn/img_convert/f5db6e79b02a7ea5c9c0ebd0b4896341.png)

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！**

中...(img-Enwecz8e-1726117633808)]
[外链图片转存中...(img-Ojf9L4v9-1726117633808)]

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！**