golang 基于 mysql 简单实现分布式读写锁
作者：二牛QAQ 发布时间：2024-01-17 00:07:19
标签：golang,mysql,分布式,读写锁
业务场景
因为项目刚上线，目前暂不打算引入其他中间件，所以打算通过 mysql 来实现分布式读写锁；而该业务场景也满足分布式读写锁的场景，抽象后的业务场景是：特定资源 X，可以执行 2 种操作：读操作和写操作，2种操作需要满足下面条件:
执行操作的机器分布式在不同的节点中，也就是分布式的；
读操作是共享的，也就是说同时可以有多个 goroutine 对资源 X 执行读操作；
写操作是互斥的，也就是说同一时刻只允许有一个 goroutine 对资源 X 执行写操作；
读操作和写操作是互斥的，也就是说写操作和读操作不能同时存在
既然需要如此实现，下面我们看下什么是分布式读写锁。
什么是分布式读写锁
大家对于锁肯定不陌生，在 golang 中 sync.Mutex 锁是常见的，一般用在单节点多 goroutine 中对资源的并发访问；但是分布式场景下，单节点 sync.Mutex 加锁的方式就会失去作用，于是人们为了在分布式环境中实现对共享资源的互斥访问，实现了各种分布式锁。
而分布式读写锁是比分布式锁粒度更小的锁，对业务场景的加锁会更加灵活，其中分布式读写锁也遵循读写锁的原则：
读模式共享，写模式互斥。
它三种模式状态: 读加锁状态、写加锁状态、无锁状态。
分布式读写锁的访问原则与读写锁类似，下面我们具体看下。
分布式读写锁的访问原则
以下列表为读写锁(也就是分布式读写锁)的读写访问原则
| 当前锁状态 | 读锁请求 | 写锁请求 |
|---|---|---|
| 无锁状态 | 可以 | 可以 |
| 读锁状态 | 可以 | 不可以 |
| 写锁状态 | 不可以 | 不可以 |
读锁
只有在无锁和读锁下可以获取读锁。
读锁的模式下，任何请求读锁都可以。
读锁的模式下， 请求写锁不可以，直到所有读锁解锁，写锁才能获取到锁。
写锁
只有在无锁状态下可以获取写锁。
写锁的模式下，任何请求读锁和写锁都阻塞，直到写锁解锁。
具体实现
如果本地没有 mysql 数据库，可以通过这篇文章快速搭建: 如何使用 docker 搭建一个 mysql 服务
通过 gorm 连接 mysql
gorm 是一个 golang 的 orm 框架，可以使用它快速连接数据库，具体代码如下:
package main
import (
"fmt"
"gorm.io/driver/mysql"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)
var (
db *gorm.DB
dbUsername = "kele"
dbPassword = "baishi2020"
dbHost = "127.0.0.1:7306"
dbDatabase = "lingmo"
stateReadLock = "ReadLock"
stateWriteLock = "WriteLock"
stateUnlock = "Unlock"
)
type RWLock struct {
LockMark string `gorm:"default:'Unlock'"`
ReadLockCount uint32 `gorm:"default:0"`
LockReason string
}
type Stock struct {
gorm.Model
RWLock
Count int64
}
func (Stock) TableName() string {
return "stocks"
}
func init() {
dsn := fmt.Sprintf("％s:％s@tcp(％s)/％s?charset=utf8mb4&parseTime=True&loc=Local", dbUsername, dbPassword, dbHost, dbDatabase)
mysqlConfig := mysql.Config{DSN: dsn}
gormConfig := &gorm.Config{Logger: logger.Default.LogMode(logger.Info)}
var err error
if db, err = gorm.Open(mysql.New(mysqlConfig), gormConfig); err != nil {
panic(err)
}
db.Set("db:table_options", "ENGINE = InnoDB DEFAULT CHARSET = utf8")
// register tables
if err = db.AutoMigrate(&Stock{}); err != nil {
panic(err)
}
}
func main() {
if result := db.Model(&Stock{}).Save(&Stock{Model: gorm.Model{}, RWLock: RWLock{}, Count: 10}); result.Error != nil {
panic(result.Error)
}
}
首先我们定义了一个库存表 stocks，并且在其中添加三个和读写锁相关的字段，三个字段的含义如下：
LockMark: 表示某条数据加锁的状态，只能是读锁、写锁、无锁状态中的一种。
ReadLockCount: 首先读模式是共享的，意味着可以有多个 goroutine 并发访问，而 ReadLockCount 字段则记录当前并发访问的 goroutine 数量。
LockReason: 记录当前加锁的原因；读锁是最新的 goroutine 的 lockReason，写锁则是写锁 goroutine 的 lockReason。
其余则是一些 gorm 连接 mysql 逻辑，这里不再多赘述。
实现读锁模式
具体代码如下:
func (s Stock) RLock(db *gorm.DB, lockReason string) error {
condition := "(id = ?) AND (lock_mark != ?)"
fields := map[string]interface{}{
"lock_mark": stateReadLock,
"read_lock_count": gorm.Expr("read_lock_count + ?", 1),
"lock_reason": lockReason,
}
result := db.Model(&Stock{}).Where(condition, s.ID, stateWriteLock).Updates(fields)
if result.Error != nil {
return result.Error
}
if result.RowsAffected == 0 {
return errors.New("failed to rlock Stock, RowsAffected=0")
}
return nil
}
func (s Stock) RUnlock(db *gorm.DB, UnLockReason string) error {
sql := fmt.Sprintf(`UPDATE stocks SET read_lock_count=if(read_lock_count>0,read_lock_count-1,0), lock_mark=if(read_lock_count<1, 'Unlock', 'ReadLock'),lock_reason ='％s' where id= ％d and lock_mark='％s'`, UnLockReason, s.ID, stateReadLock)
result := db.Exec(sql)
if result.Error != nil {
return result.Error
}
if result.RowsAffected == 0 {
return errors.New("failed to RUnlock Stock, RowsAffected=0")
}
return nil
}
func main() {
if result := db.Model(&Stock{}).Save(&Stock{Model: gorm.Model{}, RWLock: RWLock{}, Count: 10}); result.Error != nil {
panic(result.Error)
}
s := &Stock{Model: gorm.Model{ID: 1}}
if result := db.Model(s).First(s); result.Error != nil {
panic(result.Error)
}
if err := s.RLock(db, "readLock_reason_1"); err != nil {
panic(err)
}
if err := s.RLock(db, "readLock_reason_2"); err != nil {
panic(err)
}
if err := s.RUnlock(db, "readLock_unlock_1"); err != nil {
panic(err)
}
if err := s.RUnlock(db, "readLock_unlock_2"); err != nil {
panic(err)
}
}
执行以上代码是可以正常运行的， 下面我们分析下:
读锁的 sql 语句如下，只要在非写锁状态下就能加读锁。
UPDATE `stocks` SET `lock_mark` = 'ReadLock', `lock_reason` = 'readLock_reason_1', `read_lock_count` = read_lock_count + 1, `updated_at` = '2022-09-25 14:58:45.693' WHERE (( id = 1 )
AND ( lock_mark != 'WriteLock' ))
AND `stocks`.`deleted_at` IS NULL
解读锁的 sql 语句如下，只有在读锁状态下才能解读锁，另外还要更新 read_lock_count 和 lock_reason 字段。
UPDATE stocks
SET read_lock_count =
IF
( read_lock_count > 0, read_lock_count - 1, 0 ),
lock_mark =
IF
( read_lock_count < 1, 'Unlock', 'ReadLock' ),
lock_reason = 'readLock_unlock_1'
WHERE
id = 1
AND lock_mark = 'ReadLock'
实现写锁模式
具体代码如下:
func (s Stock) WLock(db *gorm.DB, lockReason string) error {
condition := "(id = ?) AND (lock_mark = ?)"
fields := map[string]interface{}{
"lock_mark": stateWriteLock,
"read_lock_count": 0,
"lock_reason": lockReason,
}
result := db.Model(&Stock{}).Where(condition, s.ID, stateUnlock).Updates(fields)
if result.Error != nil {
return result.Error
}
if result.RowsAffected == 0 {
return errors.New("failed to WLock Stock, RowsAffected=0")
}
return nil
}
func (s Stock) WUnlock(db *gorm.DB, UnLockReason string) error {
condition := "(id = ?) AND (lock_mark = ?)"
fields := map[string]interface{}{
"lock_mark": stateUnlock,
"read_lock_count": 0,
"lock_reason": UnLockReason,
}
result := db.Model(&Stock{}).Where(condition, s.ID, stateWriteLock).Updates(fields)
if result.Error != nil {
return result.Error
}
if result.RowsAffected == 0 {
return errors.New("failed to WUnlock Stock, RowsAffected=0")
}
return nil
}
func main() {
s := &Stock{Model: gorm.Model{ID: 1}}
if result := db.Model(s).First(s); result.Error != nil {
panic(result.Error)
}
if err := s.WLock(db, "writeLock_reason_1"); err != nil {
panic(err)
}
if err := s.WUnlock(db, "unWriteLock_reason_1"); err != nil {
panic(err)
}
}
执行以上代码也是可以运行，下面是分析结果
写锁的 sql 语句如下，只有在无锁状态下才能加锁成功
UPDATE `stocks` SET `lock_mark` = 'WriteLock', `lock_reason` = 'writeLock_reason_1', `read_lock_count` = 0, `updated_at` = '2022-09-25 15:06:10.71' WHERE (( id = 1 )
AND ( lock_mark = 'Unlock' ))
AND `stocks`.`deleted_at` IS NULL
解写锁的 sql 语句如下，只有在写锁状态下才能解写锁
UPDATE `stocks` SET `lock_mark` = 'Unlock', `lock_reason` = 'unWriteLock_reason_1', `read_lock_count` = 0, `updated_at` = '2022-09-25 15:06:10.719' WHERE (( id = 1 )
AND ( lock_mark = 'WriteLock' ))
AND `stocks`.`deleted_at` IS NULL
来源：https://juejin.cn/post/7147214210324234271
0
投稿猜你喜欢
Python有两个用于相等比较的运算符，“is”和“==”（等于）。在这篇文章中，我将教你们两者之间的区别，以及通过几个简单地例子说明什么时- 一、本讲学习目标1.掌握构造方法的使用2.掌握析构方法的使用3.掌握self变量的使用二、构造方法（一）概述构造方法指的是__init__(
- 本文实例讲述了mysql索引对排序的影响。分享给大家供大家参考，具体如下：索引不仅能提高查询速度，还可以添加排序速度，如果order by
- 前言pyinstaller能够在Windows、Linux等操作系统下将Python脚本打包成可直接运行程序。使Python脚本
- (1)安装Jpype 用Python调用jar包需要安装jpype扩展，在Ubuntu上可以直接使用apt-get安装jpype扩展$ su
- 如何获取一个网站的相关信息，获取赶集网的招聘信息，本文为大家介绍利用python获取赶集网招聘信息的关键代码，供大家参考，具体内容如下imp
- django是python语言快速实现web服务的大杀器，其开发效率可以非常的高！但因为秉承了语言的灵活性，django框架又太灵活，以至于
- 这篇文章主要给大家介绍了关于Django跨域请求问题解决的相关资料，文中介绍的实现方法包括：使用django-cors-headers全局控
- 基本思想：归并排序是一种典型的分治思想，把一个无序列表一分为二，对每个子序列再一分为二，继续下去，直到无法再进行划分为止。然后，就开始合并的
- 任务描述本关任务：利用 PyEcharts 绘制一个基本的漏斗图。相关知识为了完成本关任务，你需要掌握：1. Python 的基本语法2.
- 用面向对象的思维解决问题的重点当遇到一个需求的时候不用自己去实现，如果自己一步步实现那就是面向过程；应该找一个专门做这个事的人来做。面向对象
- 1，SELECT 语句 在SQL的世界里，最最基础的操作就是SELECT 语句了。在数据库工具下直接采用SQL
- skimage包的exposure模块图像亮度与对比度的调整，是放在skimage包的exposure模块里面1、gamma调整对原图像的像
- 1、Git Bash默认路径在windows系统上操作Git的客户端是Git Bash。安装完Git Bash之后，双击打开，如下图：使用p
- 层的八条定律当然，这些并非真正的定律，而只是一些有益的忠告，使你免陷于使用层时可能的困顿中。原来有九条定律的，我们精简掉一条，还有下面的八条
- 本文实例讲述了thinkPHP框架实现类似java过滤器的简单方法。分享给大家供大家参考，具体如下：写java web代码的时候，可以定义过
- 译者按：原文写于2011年末，虽然文中关于Python 3的一些说法可以说已经不成立了，但是作为一篇面向从其他语言转型到Python的程序员
- 一个post类型的接口怎么编写脚本实现1、打开网页，在fiddler上获取到接口的URL2、用Python的requests库实现impor
- 上一篇文章讲解了如何实现马丁策略，但没有探索其泛化能力，所以这次来尝试回测3000只股票来查看盈利比例。批量爬取股票数据这里爬取数据继续使用
- 一、将数据写入opengauss前提准备：成功opengauss数据库，并创建用户jack，创建数据库datasets。数据准备：所用数据以
