golang+redis 实现简单秒杀功能
2021/10/12 19:18:16
本文主要是介绍golang+redis 实现简单秒杀功能,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
使用 golang + redis 实现一个简单的秒杀功能
安装 redigogo get github.com/gomodule/redigo/redisredigo 文档:https://pkg.go.dev/github.com/gomodule/redigo/redis
1. 创建 redis 连接池 (我这里是在 global 包下新建的一个 redis.go 文件,只有一行代码如下)
var RedisPool *redis.Pool紧接着,在 main.go 文件里初始化 redis 连接池
func init() { global.RedisPool = &redis.Pool{ MaxIdle: 10, // 最初的连接数量 MaxActive: 0, // 最大连接数量 0表示不限制 IdleTimeout: 300 * time.Second, // 连接关闭时间单位秒 (超时不使用自动关闭) Dial: func() (redis.Conn, error) { return redis.Dial("tcp", "127.0.0.1:6379") }, } }
2. 实现模拟秒杀功能
var lock sync.Mutex var listName = "seckill" func main() { createQueue() }
// 模拟抢购 func createQueue() { var wg sync.WaitGroup rdb := global.RedisPool.Get() defer rdb.Close() // 设定20人抢购(用于随机抢购) userIDs := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20} count := len(userIDs) var ch = make(chan int, count) f := func(id int, done func()) { defer func() { done() }() lock.Lock() defer lock.Unlock() // 判断人数是否少于10 lLen, _ := redis.Int64(rdb.Do("LLEN", listName)) if lLen < 10 { // 写入队列,字符串使用 @ 符合连接用户id和当前时间 rdb.Do("RPUSH", listName, fmt.Sprintf("%d@%v", id, time.Now())) fmt.Println(id, "抢购成功") } else { fmt.Println("抢购活动已结束") } ch <- id } wg.Add(count) for _, v := range userIDs { go f(v, wg.Done) } for i := 0; i < count; i++ { <-ch } close(ch) wg.Wait() }
运行以上代码,结果:
20 抢购成功
10 抢购成功
1 抢购成功
2 抢购成功
3 抢购成功
4 抢购成功
5 抢购成功
6 抢购成功
7 抢购成功
8 抢购成功
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
抢购活动已结束
查看队列数据:
127.0.0.1:6379> lrange seckill 0 100
1) "20@2021-10-12 09:49:08.162533478 +0800 CST m=+0.001920843"
2) "10@2021-10-12 09:49:08.163083661 +0800 CST m=+0.002471032"
3) "1@2021-10-12 09:49:08.163379093 +0800 CST m=+0.002766464"
4) "2@2021-10-12 09:49:08.163807431 +0800 CST m=+0.003194802"
5) "3@2021-10-12 09:49:08.16409304 +0800 CST m=+0.003480411"
6) "4@2021-10-12 09:49:08.164380002 +0800 CST m=+0.003767371"
7) "5@2021-10-12 09:49:08.16480322 +0800 CST m=+0.004190580"
8) "6@2021-10-12 09:49:08.165220687 +0800 CST m=+0.004608047"
9) "7@2021-10-12 09:49:08.165518322 +0800 CST m=+0.004905684"
10) "8@2021-10-12 09:49:08.166351454 +0800 CST m=+0.005738847"
3. 保存到数据库 新建一个 main.go 文件,创建 redis 链接池和初始化 mysql
func init() { global.RedisPool = &redis.Pool{ MaxIdle: 10, // 最初的连接数量 MaxActive: 0, // 最大连接数量 0表示不限制 IdleTimeout: 300 * time.Second, // 连接关闭时间单位秒 (超时不使用自动关闭) Dial: func() (redis.Conn, error) { return redis.Dial("tcp", "127.0.0.1:6379") }, } mysql.Init("root:123456@tcp(127.0.0.1:3306)/test001?charset=utf8&loc=Asia%2FShanghai") }
读取队列数据,保存到数据表 代码实现了,监控队列是否有数据,如果有数据则马上保存到数据表,如果读取插入数据失败,则把移出并获取列表的元素重新写入到队列里
func main() { rdb := global.RedisPool.Get() defer rdb.Close() for { total, err := redis.Int64(rdb.Do("LLEN", listName)) if err != nil { fmt.Println(err) return } if total == 0 { time.Sleep(1 * time.Second) fmt.Println("total:", total) continue } // LPOP:在队列最左侧取出一个值(移除并返回列表的第一个元素) str, err := redis.String(rdb.Do("LPOP", listName)) if err != nil { fmt.Println(err) return } // 分隔字符串 l := strings.Split(str, "@") // 保存到数据库中 query := "insert into seckill_list (user_id, time_stamp) values(?, ?)" _, err = mysql.Insert(query, l[0], l[1]) if err != nil { fmt.Println("mysql.Insert err:", err) // 数据库插入失败时回滚机制 rdb.Do("RPUSH", listName, str) } else { fmt.Println(fmt.Sprintf("id:%v保存成功", l[0])) } } }
运行结果(如果有数据就马上读取并保存到数据表):
id:20保存成功
id:10保存成功
id:1保存成功
id:2保存成功
id:3保存成功
id:4保存成功
id:5保存成功
id:6保存成功
id:7保存成功
id:8保存成功
total: 0
total: 0
total: 0
total: 0
id:1保存成功
id:20保存成功
id:2保存成功
id:3保存成功
id:4保存成功
id:5保存成功
id:6保存成功
id:7保存成功
id:8保存成功
id:9保存成功
total: 0
total: 0
total: 0
total: 0
保存到数据表的结果:
以上是本次简单秒杀功能的实现,如果有其他方式欢迎留言或给出建议,谢谢(#^.^#)
这篇关于golang+redis 实现简单秒杀功能的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-11-08阿里云Redis项目实战入门教程
- 2024-11-08阿里云Redis资料:新手入门与初级使用指南
- 2024-11-08阿里云Redis教程:新手入门及实用指南
- 2024-11-07阿里云Redis学习入门:新手必读指南
- 2024-11-07阿里云Redis学习入门:从零开始的操作指南
- 2024-11-07阿里云Redis学习:初学者指南
- 2024-11-06阿里云Redis入门教程:轻松搭建与使用指南
- 2024-11-02Redis项目实战:新手入门教程
- 2024-10-22Redis入门教程:轻松掌握数据存储与操作
- 2024-10-22Redis缓存入门教程:快速掌握Redis缓存基础知识