大家可以关注我微信公众号【技术闲聊吧】
以发工资为例使用15行代码实现一个分布式事务
开发一个分布式事务不是一件容易的事,经验丰富的老司机也往往会踩各种坑。
今天,我们就用 octopus来开发一个发工资的分布式事务。
首先,我们明确要开发哪些角色
- 发工资的公司:我们称之为AP,也是事务发起方
- 员工的账户银行:称之为RM,事务参与方
- 银联:称为TC,事务协调者。当然事务协调者也可以是放在发工资的公司一方。
如果使用octopus,TC不需要开发,octopus已经集成了,只需要部署就可以了。
那我们来看看如何开发AP和RM代码。
我们使用SAGA类型的分布式事务来发工资,保证事务的最终一致性。
开发AP
我们只看核心代码,具体代码参考demo。
遍历所有员工,将给每个每个员工账户转账作为一个子事务,添加到分布式事务中。
func (app *SagaApplication) Pay(ctx context.Context, users []*saga_rm.BankAccountRecord) (*pb.SagaResponse, error) {
resp, err := saga_cli.SagaTransaction(ctx, app.tcRpcClient, time.Now().Add(1*time.Minute),
func(t *saga_cli.Transaction, gtid string) error {
for i, user := range users {
commit := "/bankservice.SagaBankService/In" // 银行gRPC服务的commit接口名
compensation := "/bankservice.SagaBankService/Out" // 银行gRPC服务的compensation接口名
t.AddGrpcBranch(i + 1, app.getUserBank(user.UserID).Address, commit, compensation, user.ToPb(), saga_cli.WithMaxRetry(1)) // 添加一个子事务
}
return nil
})
return resp, err
}
开发RM
RM需要实现两个接口, commit和compensation,分别代表提交和回滚(SAGA里称为补偿)。
提交接口
开发者不用考虑事务相关的逻辑(提交,回滚,幂等,异常处理等),只需要调用HandleCommit接口来实现业务逻辑即可。
func (rm *SagaRmBankGrpcService) In(ctx context.Context, in *pb.SagaRequest) (*pb.SagaResponse, error) {
gtid, bid := cgrpc.ParseContextMeta(ctx)
err := sagarm.HandleCommit(ctx, rm.db, gtid, bid, func(tx *sql.Tx) error {
rx, err := tx.Exec("UPDATE account SET balance=balance+$1 WHERE id=$2", in.GetAccount(), in.GetUserId())
return rm.handleErr(rx, err)
})
return &pb.SagaResponse{}, err
}
补偿接口
开发者不用考虑事务相关的逻辑(提交,回滚,幂等,异常处理等),只需要调用HandleCompensation接口来实现业务逻辑即可。
func (rm *SagaRmBankGrpcService) Out(ctx context.Context, in *pb.SagaRequest) (*pb.SagaResponse, error) {
gtid, bid := cgrpc.ParseContextMeta(ctx)
err := sagarm.HandleCompensation(ctx, rm.db, gtid, bid, func(tx *sql.Tx) error {
rx, err := tx.Exec("UPDATE account SET balance=balance+$1 WHERE id=$2", -1 * in.GetAccount(), in.GetUserId())
return rm.handleErr(rx, err)
})
return &pb.SagaResponse{}, err
}
可以看到8行代码就实现了提交和补偿接口,还是非常简单的。
我们只需要部署好TC,参考部署文档, 然后运行AP,则可以提交一个发工资的分布式事务。
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。