缘起
最近阅读<<Go微服务实战>> (刘金亮, 2021.1)
本系列笔记拟采用golang练习之
Saga模式
- saga模式将分布式长事务切分为一系列独立短事务
- 每个短事务是可通过补偿动作进行撤销的
- 事务动作和补动作偿都是幂等的, 允许重复执行而不会有副作用
Saga由一系列的子事务“Ti”组成,
每个Ti都有对应的补偿“Ci”,
当Ti出现问题时Ci用于处理Ti执行带来的问题。
可以通过下面的两个公式理解Saga模式。
T = T1 T2 … Tn
T = TCT
Saga模式的核心理念是避免使用长期持有锁(如14.2.2节介绍的两阶段提交)的长事务,
而应该将事务切分为一组按序依次提交的短事务,
Saga模式满足ACD(原子性、一致性、持久性)特征。
摘自 <<Go微服务实战>> 刘金亮, 2021.1
目标
- 为实现saga模式的分布式事务, 先撸一个pub/sub事务消息队列服务
事务消息队列服务的功能性要求
- 消息不会丢失: 消息的持久化
- 消息的唯一性: 要求每个消息有全局ID和子事务ID
- 确保投递成功: 投递队列持久化, 投递状态持久化, 失败重试
子目标(Day 2)
- 消息的设计
- 消息的持久化
- 投递队列的持久化
- 投递状态的持久化
设计
- TxMsg: 事务消息模型
- database: 添加事务消息表, 消息投递表, 成功投递表, 失败投递表
- publish: 消息发布api
- routers: 添加/publish路由
TxMsg.go
事务消息模型
package models
// 事务消息体
type TxMsg struct {
// 全局事务ID
GlobalID string
// 子事务ID
SubID string
// 发送者ID
SenderID string
// 时间戳, 使用time.Now().UnixNano()
CreateTime int64
// 主题, 即消息类型
Topic string
// 消息内容, 一般是json
Content string
}
database.go
添加事务消息表, 消息投递表, 成功投递表, 失败投递表
package database
import "github.com/jmoiron/sqlx"
import _ "github.com/mattn/go-sqlite3"
type DBFunc func(db *sqlx.DB) error
type TXFunc func(db *sqlx.DB, tx *sqlx.Tx) error
func init() {
// must prepare tables
err := DB(initDB)
if err != nil {
panic(err)
}
}
func initDB(db *sqlx.DB) error {
// 订阅者/消费者: subscriber
_, e := db.Exec(`create table if not exists subscriber(
id integer primary key autoincrement,
client_id varchar(50) unique not null,
topic varchar(100) not null,
notify_url varchar(500) not null,
expire_time integer
)`)
if e != nil {
return e
}
// 事务消息: tx_msg
_, e = db.Exec(`create table if not exists tx_msg (
id integer primary key autoincrement,
global_id string varchar(50) not null,
sub_id string varchar(50) unique not null,
sender_id varchar(50) not null,
create_time integer not null,
topic varchar(100) not null,
content nvarchar(2048) not null
)`)
if e != nil {
return e
}
// 投递队列: delivery_queue
_, e = db.Exec(`create table if not exists delivery_queue (
id integer primary key autoincrement,
msg_id integer not null,
sub_id integer not null,
status_flag integer not null,
failed_count integer not null
)`)
if e != nil {
return e
}
// 成功投递队列: success_queue
_, e = db.Exec(`create table if not exists success_queue (
id integer primary key autoincrement,
msg_id integer not null,
sub_id integer not null
)`)
if e != nil {
return e
}
// 投递失败队列: failed_queue
_, e = db.Exec(`create table if not exists failed_queue (
id integer primary key autoincrement,
msg_id integer not null,
sub_id integer not null
)`)
if e != nil {
return e
}
return nil
}
func open() (*sqlx.DB, error) {
return sqlx.Open("sqlite3", "./mqs.db")
}
func DB(action DBFunc) error {
db,err := open()
if err != nil {
return err
}
defer func() { _ = db.Close() }()
return action(db)
}
func TX(action TXFunc) error {
return DB(func(db *sqlx.DB) error {
tx, err := db.Beginx()
if err != nil {
return err
}
err = action(db, tx)
if err == nil {
return tx.Commit()
} else {
return tx.Rollback()
}
})
}
publish.go
消息发布api
package handlers
import (
"github.com/gin-gonic/gin"
"github.com/gin-gonic/gin/binding"
"github.com/jmoiron/sqlx"
"learning/gooop/saga/mqs/database"
"learning/gooop/saga/mqs/logger"
"learning/gooop/saga/mqs/models"
"net/http"
"time"
)
func Publish(c *gin.Context) {
// parse request
msg := &models.TxMsg{}
if err := c.ShouldBindBodyWith(&msg, binding.JSON); err != nil {
c.AbortWithStatusJSON(
http.StatusInternalServerError,
gin.H{ "ok": false, "error": err.Error()})
return
}
// fixme: validate msg
// save to db
ids := []int64{0}
err := database.TX(func(db *sqlx.DB, tx *sqlx.Tx) error {
id, e := saveTxMsg(db, tx, msg)
if e != nil {
return e
}
ids[0] = id
return nil
})
if ids[0] > 0 {
newMsgId := ids[0]
logger.Logf("publish new msg: %d", newMsgId)
// todo: 新增消息, 开始投送
}
// reply
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{ "ok": false, "error": err.Error()})
} else {
c.JSON(http.StatusOK, gin.H { "ok": true })
}
}
func saveTxMsg(db *sqlx.DB, tx *sqlx.Tx, msg *models.TxMsg) (int64,error) {
// insert tx_msg
r, e := db.Exec(
`replace into tx_msg(global_id, sub_id, sender_id, create_time, topic, content) values(?,?,?,?,?,?)`,
msg.GlobalID, msg.SubID, msg.SenderID, msg.CreateTime, msg.Topic, msg.Content,
)
if e != nil {
return 0,e
}
// get last insert id
id,e := r.LastInsertId()
if e != nil {
return 0,e
}
if id > 0 {
// copy to delivery queue
now := time.Now().UnixNano()
r, e = db.Exec(`
insert into delivery_queue(msg_id, sub_id, status_flag, failed_count)
select
?, s.id, 0, 0
from
sub_info s
where
s.expire_time>?
`, id, now)
if e != nil {
return 0,e
}
}
return id, nil
}
routers.go
添加/publish路由
package routers
import (
"github.com/gin-gonic/gin"
"learning/gooop/saga/mqs/handlers"
)
func RegisterRouters() *gin.Engine {
r := gin.Default()
r.Use(gin.Logger())
r.GET("/ping", handlers.Ping)
r.POST("/subscribe", handlers.Subscribe)
r.POST("/publish", handlers.Publish)
return r
}
(未完待续)
有疑问加站长微信联系(非本文作者)