首页 > 资讯 > 数据库 >golang 基于 mysql 简单实现分布式读写锁

618

分享到

golang 基于 mysql 简单实现分布式读写锁

2024-04-02 19:04:59 618人浏览安东尼

摘要

目录业务场景什么是分布式读写锁分布式读写锁的访问原则读锁写锁具体实现通过 gORM 连接 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

总结

分布式读写锁的实现有多种方式，也可以通过 etcd、Redisson 的方式进行实现，而本文着重说明可通过 mysql 来实现，这种方式的优势在于不必引入额外的组件且实现较为简单，因此也有一定的应用场景，

到此这篇关于golang 基于 mysql 简单实现分布式读写锁的文章就介绍到这了,更多相关golang 读写锁内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！

您可能感兴趣的文档:

点击免费下载>>软考高级考试备考技巧/历年真题/备考精华资料

--结束END--

本文标题: golang 基于 mysql 简单实现分布式读写锁

本文链接: https://www.lsjlt.com/news/121242.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑，方便收藏和打印～

下载Word文档

去做题

猜你喜欢

golang 基于 mysql 简单实现分布式读写锁

目录业务场景什么是分布式读写锁分布式读写锁的访问原则读锁写锁具体实现通过 gorm 连接 mysql实现读锁模式实现写锁模式总结业务场景因为项目刚上线，目前暂不打算引入其他中间件，...

99+

2024-04-02
基于Redis的分布式锁的简单实现方法

Redis官方给出两种思路第一种：SET key value [EX seconds] [PX milliseconds] NX 第二种：SETNX+GETSET 首先，分别看一下这几个命令 SET命令 ...

99+

2024-04-02
详解基于redis实现分布式锁

目录前言原理剖析实现编写注解拦截器拦截上述提及工具问题分析前言为了保证一个在高并发存场景下只能被同一个线程操作，java并发处理提供ReentrantLock或Synchroniz...

99+

2024-04-02
基于Zookeeper实现分布式锁详解

目录1、什么是Zookeeper？2、Zookeeper节点类型3、Zookeeper环境搭建4、Zookeeper基本使用5、Zookeeper应用场景6、Zookeeper分布式...

99+

2024-04-02
基于Zookeeper怎么实现分布式锁

这篇文章主要介绍“基于Zookeeper怎么实现分布式锁”，在日常操作中，相信很多人在基于Zookeeper怎么实现分布式锁问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答”基于Zookeeper怎么实现分布式锁...

99+

2023-06-22
go 分布式锁简单实现实例详解

目录正文案例资源加锁使用redis来实现分布式锁redis lua保证原子性正文其实锁这种东西，都能能不加就不加，锁会导致程序一定程度上退回到串行化，进而降低效率。案例首先，看...

99+

2024-04-02
基于Redis分布式锁的实现代码

概述目前几乎很多大型网站及应用都是分布式部署的，分布式场景中的数据一致性问题一直是一个比较重要的话题。分布式的CAP理论告诉我们“任何一个分布式系统都无法同时满足一致性（Consistency）、可用性（...

99+

2024-04-02
MySQL实现分布式锁

目录基于MySQL分布式锁实现原理及代码MySQL锁InnoDB共享锁排它锁MyISAM表共享读锁表独占写锁分布式锁实现难点：为什么需要for(;总结基于MySQL分布式锁实现原理及...

99+

2022-11-13

MySQL实现分布式锁 MySQL分布式锁
基于Redis缓存怎么实现分布式锁

本篇内容介绍了“基于Redis缓存怎么实现分布式锁”的有关知识，在实际案例的操作过程中，不少人都会遇到这样的困境，接下来就让小编带领大家学习一下如何处理这些情况吧！希望大家仔细阅读，能够学有所成！什么是分布式锁首先我们先来简单了解一下什么是...

99+

2023-06-19
怎么使用RedisTemplat实现简单的分布式锁

这篇文章主要讲解了“怎么使用RedisTemplat实现简单的分布式锁”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“怎么使用RedisTemplat实现简单的分布式锁”吧！不使用rediss...

99+

2023-06-25
如何分析基于redis分布式锁实现秒杀

本篇文章为大家展示了如何分析基于redis分布式锁实现秒杀，内容简明扼要并且容易理解，绝对能使你眼前一亮，通过这篇文章的详细介绍希望你能有所收获。业务场景所谓秒杀，从业务角度看，是短时间内多个用户“争抢”资源，这里的资源在大部分秒杀场景里是...

99+

2023-06-02
Go 语言下基于Redis分布式锁的实现方式

分布式锁一般有三种实现方式：1. 数据库乐观锁；2. 基于Redis的分布式锁；3. 基于ZooKeeper的分布式锁。本篇博客将介绍第二种方式，基于Redis实现分布式锁。虽然网上...

99+

2024-04-02
Go结合Redis用最简单的方式实现分布式锁

目录前言单Redis实例场景加解锁示例小结多Redis实例场景加解锁示例小结总结前言在项目中我们经常有需要使用分布式锁的场景，而Redis是实现分布式锁最常见的一种方式，并且我们也...

99+

2024-04-02
基于redis分布式锁如何实现秒杀功能

这篇文章主要介绍了基于redis分布式锁如何实现秒杀功能，具有一定借鉴价值，感兴趣的朋友可以参考下，希望大家阅读完这篇文章之后大有收获，下面让小编带着大家一起了解一下。业务场景所谓秒杀，从业务角度看，是短时...

99+

2024-04-02
java基于mongodb实现分布式锁的示例代码

目录原理实现使用原理通过线程安全findAndModify 实现锁实现定义锁存储对象： @Data @NoArgsConstruc...

99+

2024-04-02
java基于jedisLock—redis分布式锁实现示例代码

分布式锁是啥？单机锁的概念：我们正常跑的单机项目（也就是在tomcat下跑一个项目不配置集群）想要在高并发的时候加锁很容易就可以搞定，java提供了很多的机制例如：synchronized、volatile、ReentrantLock等锁的...

99+

2023-05-30

jedislock redis 分布式锁
SpringBoot基于Redis的分布式锁实现过程记录

目录一、概述二、环境搭建三、模拟一个库存扣减的场景四、总结一、概述什么是分布式锁在单机环境中，一般在多并发多线程场景下，出现多个线程去抢占一个资源，这个时候会出现线程同步问题，造...

99+

2024-04-02
基于Redis实现分布式锁的方法（lua脚本版）

1、前言在Java中，我们通过锁来避免由于竞争而造成的数据不一致问题。通常我们使用synchronized 、Lock来实现。但是Java中的锁只能保证在同一个JVM进程内中可用...

99+

2024-04-02
分布式系统中使用 Golang 函数实现分布式锁

golang 函数可用于实现分布式锁，协调多个进程对共享资源的访问。这些函数通过利用共享存储（如 redis）来实现锁机制，确保在任何时刻只有一个进程能访问资源。基于 Golang ...

99+

2024-04-20

golang 分布式系统 redis git
如何编写简单的demo实现读写分离

本篇内容主要讲解“如何编写简单的demo实现读写分离”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“如何编写简单的demo实现读写分离”吧! 前言相信有...

99+

2024-04-02