go-zero处理本地事务

go-zero处理本地事务,
sqlx.SqlConn 提供了基础的事务机制,官方代码

go 复制代码
    var conn sqlx.SqlConn
    err := conn.TransactCtx(context.Background(), func(ctx context.Context, session sqlx.Session) error {
        r, err := session.ExecCtx(ctx, "insert into user (id, name) values (?, ?)", 1, "test")
        if err != nil {
            return err
        }
        r ,err =session.ExecCtx(ctx, "insert into user (id, name) values (?, ?)", 2, "test")
        if err != nil {
            return err
        }
    })

改造的代码

1.在生成的model文件userauthmodel.go中添加代码

go 复制代码
package user

import (
	"context"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

var _ UserAuthModel = (*customUserAuthModel)(nil)

type (
	UserAuthModel interface {
		UserAuthModel
		withSession(session sqlx.Session) UserAuthModel
		// 添加代码1
		TransactCtx(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error
	}

	customUserAuthModel struct {
		*defaultUserAuthModel
	}
)

func NewUserAuthModel(conn sqlx.SqlConn) UserAuthModel {
	return &customUserAuthModel{
		defaultUserAuthModel: newUserAuthModel(conn),
	}
}

func (m *customUserAuthModel) withSession(session sqlx.Session) UserAuthModel {
	return NewUserAuthModel(sqlx.NewSqlConnFromSession(session))
}

// 添加代码2
func (m *customUserAuthModel) TransactCtx(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error {
	return m.conn.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error {
		return fn(ctx, session)
	})

}

2.在逻辑层处理代码

go 复制代码
err := l.UserAuthModel.TransactCtx(context.Background(), func(ctx context.Context, session sqlx.Session) error {
		t := time.Now()
		// 在这里执行数据库操作
		_, err := session.Exec(`INSERT INTO _user_auth ( user_id,updated_time,created_time) 
                 VALUES (?, ?, ? )`,  11111, t, t)
		if err != nil {
			return err
		}
		_, err = session.Exec(`INSERT INTO _user_auth_log ( user_id,updated_time,created_time) 
                 VALUES (?, ?, ? )`,  11111, t, t)
		if err != nil {
			return err
		}
		if err != nil {
			return err
		}
		return nil
	})

	if err != nil {
		logx.ErrorStack("Transaction failed:", err)
	}

	return err
相关推荐
轻语呢喃11 分钟前
JavaScript :事件循环机制的深度解析
javascript·后端
ezl1fe12 分钟前
RAG 每日一技(四):让AI读懂你的话,初探RAG的“灵魂”——Embedding
后端
经典199214 分钟前
spring boot 详解以及原理
java·spring boot·后端
Aurora_NeAr16 分钟前
Apache Iceberg数据湖高级特性及性能调优
大数据·后端
程序员清风27 分钟前
程序员要在你能挣钱的时候拼命存钱!
后端·面试·程序员
夜阳朔1 小时前
Conda环境激活失效问题
人工智能·后端·python
白仑色2 小时前
Spring Boot 多环境配置详解
java·spring boot·后端·微服务架构·配置管理
懒斌2 小时前
linux驱动程序
后端
mit6.8242 小时前
论容器化 | 分析Go和Rust做医疗的后端服务
docker·golang·rust
超级小忍2 小时前
在 Spring Boot 中优化长轮询(Long Polling)连接频繁建立销毁问题
java·spring boot·后端