一个非侵入的Go事务管理库——如何使用

Posted 2020-12-11 倚天码农

在文章"清晰架构（Clean Architecture）的Go微服务: 事物管理"中，我谈到了如何在清晰架构中实现非侵入的事务管理。

它允许你把事务代码与业务逻辑代码分开，并且让你在编写业务逻辑时不必考虑事务。但它也有一些缺点。首先，它是整个清晰框架（Clean Architecture）的一部分，所以你不能抛开框架单独使用它。其次，尽管它对业务逻辑没有侵入，但它对框架有侵入。你需要修改框架的各个层，使其工作，这使他看起来比较复杂。 第三，正如我在文章中提到的，它存在一个依赖泄漏的漏洞。我虽然在文章中指出了解决方案，但它是一个比较大的改动，因此我当时就把它先放下了。现在，我终于有时间重新拾起它，并做了重要改进，结果令人非常满意。

项目需求

以下是新的项目需求:

1. 把事务管理代码写成一个单独的第三方库，这样人们就可以在任何框架中使用它。
2. 使其对框架无侵入，这意味着除了在用例层之外，在清晰架构的任何层中都没有事务代码。几乎所有的事务代码都在第三方的事务库中。
3. 修复以前设计中的依赖泄漏。

最终，我完成了所有的目标，结果出乎意料的好。我将写两篇文章来描述它，这篇文章讨论如何使用这个第三方库，下一篇文章讨论事务管理库的工作原理。当你要在应用程序里使用事务管理库时，你的程序分成了两部分。一部分是第三方库的程序，另一部分是应用程序的代码（它需要调用事务管理库中的函数）。本文只讲应用程序代码。

如何在项目中使用事务管理库

要想让业务函数支持事务，需要做两件事。首先，创建数据库链接;其次，使用创建的数据库链接运行SQL语句。我假设你在项目中使用了清晰架构。在这种情况下，“创建数据库链接”会在应用程序容器（（详情参见"清晰架构（Clean Architecture）的Go微服务: 程序容器（Application Container）" ）中完成，“运行SQL语句”会在业务逻辑（数据持久层）中完成。如果没有使用清晰架构，你可能会使用某种非常类似的分层架构，结构还是一样。如果你没有使用任何框架或分层架构，那么这两种代码可能会在一个地方。

你可能想知道它与没有事务支持的代码有什么不同?几乎没有。不管有没有事务支持，应用程序都要编写相同的代码，事务管理库会在后端处理所有事情。

本文中的所有代码都在"jfeng45/servicetmpl1"中，这是一个能自我进化的微服务框架，它提供了如何使用事务库的例子。

创建数据库链接

创建数据库链接有两种不同的方法，使用哪种方法取决于是否需要缓存数据库链接。

获取数据库链接

下面是创建数据库链接的代码。它在"sqlFactory.go"文件中。因为清晰架构使用了工厂方法模式（factory method pattern），这里的代码是它的一部分。如果你不想使用工厂方法模式，也是一点问题都没有的。因为数据库链接在架构中是要被缓存的，所以框架代码需要调用事务库中的函数“BuildSqlDB()”，它首先检查数据库链接是否已经存在。如果没有，则调用“factory.BuildSqlDB(&tdbc)”来创建一个。在此之前，它获取需要的参数并将它们保存在“DatabaseConfig”中，“DatabaseConfig”也是在事务库中定义的。之后它调用内部函数“buildGdbc()”生成合适的“gdbc.SqlGdbc"接口（要根据你是否需要事务）。最后，检查如果数据库链接不在缓存中，则把它放入缓存。

// implement Build method for SQL database
func (sf *sqlFactory) Build(c container.Container, dsc *config.DataStoreConfig) (DataStoreInterface, error) {
	logger.Log.Debug("sqlFactory")
	key := dsc.Code
	//if it is already in container, return
	if value, found := c.Get(key); found {
		logger.Log.Debug("found db in container for key:", key)
		sdb := value.(*sql.DB)
		return buildGdbc(sdb, dsc.Tx)
	}
	tdbc :=databaseConfig.DatabaseConfig{dsc.DriverName, dsc.UrlAddress, dsc.Tx}
	db, err := factory.BuildSqlDB(&tdbc)
	if err != nil {
		return nil, err
	}
	gdbc, err := buildGdbc(db, dsc.Tx)
	if err != nil {
		return nil, err
	}
	c.Put(key, gdbc)
	return gdbc, nil

}

下面是创建“SqlGdbc”接口的内部函数。"SqlGdbc"接口有两种实现，一种是"SqlConnTx"，它支持事务。另一个是“SqlDBTx”，它不支持事务。

func buildGdbc(sdb *sql.DB,tx bool) (gdbc.SqlGdbc, error){
	var sdt gdbc.SqlGdbc
	if tx {
		tx, err := sdb.Begin()
		if err != nil {
			return nil, err
		}
		sdt = &gdbc.SqlConnTx{DB: tx}
		logger.Log.Debug("buildGdbc(), create TX:")
	} else {
		sdt = &gdbc.SqlDBTx{sdb}
		logger.Log.Debug("buildGdbc(), create DB:")
	}
	return sdt, nil
}

有一种更简单的方法可以直接从事务库中获得"SqlGdbc"，函数是"factory.Build()"。但是当使用它时，你不能缓存数据库链接，所以我没有在框架中使用它。但是如果你不需要缓存数据库链接，调用“factory.Build()”是一个更好的方法。

数据库配置参数

数据库配置参数是在第三方事务库中定义的，但数据本身是保存在业务项目中。应用程序首先需要组装参数并将它们传递给事务库，以便得到合适的数据库链接。在我们的框架中，包括数据库参数在内的所有应用程序配置数据都保存在一个文件中。框架代码将负责从文件中获取数据。你如果不想将参数保存在文件中，直接将参数写成程序中的硬编码传递给事务库更容易。

下面是配置文件“appConfigDev.yaml”中的部分代码。对于数据库来说，关键是如何让事务库知道需要的是事务链接还是非事务链接。它有多种办法可以完成。例如，你可以为每个函数设置一个事务标志，但这需要改动大量的代码。我发现最简单的方法是将所有支持事务的函数放在一个特殊的用例（Use Case）中。在下面的示例中，有三个用例:“registration”、“listUser”和“registrationTx”，其中只有“registrationTx”是支持事务的，因为它使用“*sqlConfigTx”作为“dataStoreConfig”。

useCaseConfig:
    registration:
        code: registration
        userDataConfig: &userDataConfig
            code: userData
            dataStoreConfig: *sqlConfig
    listUser:
        code: listUser
        userDataConfig: *userDataConfig
        cacheDataConfig: &cacheDataConfig
            code: cacheData
            dataStoreConfig: *cacheGrpcConfig
    registrationTx:
        code: registrationTx
        userDataConfig: &userDataConfigTx
            code: userData
            dataStoreConfig: *sqlConfigTx

下面是来自同一配置文件的部分代码。可以看到，在“sqlConfigTx”中，有一个参数“tx:ture”，它表明它是支持事务的。

sqlConfig: &sqlConfig
    code: sqldb
    driverName: mysql
    urlAddress: "root:@tcp(localhost:4333)/service_config?charset=utf8"
    dbName:
    tx:  false
sqlConfigTx: &sqlConfigTx
    code: sqldb
    driverName: mysql
    urlAddress: "root:@tcp(localhost:4333)/service_config?charset=utf8"
    dbName:
    tx: true

在业务逻辑中访问数据库

我们用一个业务函数做例子，来展示支持事务和不支持事务的两种不同实现方式，这样你就能看到他们的区别。

不支持事务的代码

下面是“ModifyAndUnregister(user *model.User)”的非事务函数, 它在“registration.go”文件中。它是对业务函数“ModifyAndUnregister(ruc.UserDataInterface, user)”的一个简单封装，这个业务函数是被事务和非事务代码共享的。

// The use case of ModifyAndUnregister without transaction
func (ruc *RegistrationUseCase) ModifyAndUnregister(user *model.User) error {
	return ModifyAndUnregister(ruc.UserDataInterface, user)
}

下面是共享的业务函数"ModifyAndUnregister(ruc.UserDataInterface, user)"的代码，所有的业务逻辑都在这个函数里。

func ModifyAndUnregister(udi dataservice.UserDataInterface, user *model.User) error {
	//loggera.Log.Debug("ModifyAndUnregister")
	err := modifyUser(udi, user)
	if err != nil {
		return errors.Wrap(err, "")
	}
	err = unregisterUser(udi, user.Name)
	if err != nil {
		return errors.Wrap(err, "")
	}
	return nil
}

支持事务的代码

下面是相同的业务函数，但支持事务的代码。它在“registrationTx.go”文件中。你要做全部工作就是在“EnableTx()”中调用业务函数。

// The use case of ModifyAndUnregister with transaction
func (rtuc *RegistrationTxUseCase) ModifyAndUnregisterWithTx(user *model.User) error {

	udi := rtuc.UserDataInterface
	return udi.EnableTx(func() error {
		// wrap the business function inside the TxEnd function
		return ModifyAndUnregister(udi, user)
	})
}

下面是函数“EnableTx()”的实现代码(它在文件“userDataSql.go”中)。 这个代码是在持久层中。它的实现非常简单，只需调用事务库中的函数“TxEnd()”。

func (uds *UserDataSql) EnableTx(txFunc func() error) error {
	return uds.DB.TxEnd(txFunc)
}

以上就是为业务函数添加事务支持所需要做的全部工作，其余代码均在事务库中。

如果你想了解更多关于事务库本身的信息，请阅读“一个非侵入的Go事务管理库——工作原理”,

结论:

我对去年写的事务管理代码进行了升级，使其成为一个非侵入式的轻量级事务管理库。当你使用它时，只需要在应用程序中额外增加两三行代码就能搞定，所有其他代码都放在了事务管理库。它很好地将业务代码与数据库事务代码隔离开来，这样你的业务代码里就只有纯粹的业务逻辑。它是一个库而不是框架，所以不论你使用任何框架都可以使用它。

源代码:

完整的源码： "jfeng45/servicetmpl1"

索引:

1 "清晰架构（Clean Architecture）的Go微服务: 事物管理"

2 "清晰架构（Clean Architecture）的Go微服务: 程序容器（Application Container）"

3 "一个非侵入的Go事务管理库——工作原理"