[翻译]在Golang中使用高阶函数实现依赖注入

发表于 分类于 阅读次数： 本文字数： 2.6k 阅读时长 ≈ 3 分钟

原文：Dependency injection in Golang using higher order functions

你可以在github.com/steinfletcher/func-dependency-injection-go上找到完整的代码示例。该示例包含一个公开REST端口的http服务。

介绍

在这篇文章中，我们提出了一种在go中依赖注入的方法——使用高阶函数和闭包。

先看看下列返回用户属性的方法。

1
2
3
4
5
6
func GetUserProfile(id string) UserProfile {
    rows, err := db.Query("SELECT ...")
    ...
    return profileText
}

我们期望将处理用户数据的代码和访问数据库的代码分开。在这个例子中，我们希望通过提供对数据库访问的模拟，对主要业务层和其他业务逻辑进行单元测试。接下来分开这些，以至于每个方法都有一个独立的责任。

1
2
3
4
5
6
7
8
9
// 包含业务逻辑或匹配的核心业务层方法代码
func GetUserProfile(id string) User {
    ...
}

// 数据连接层方法
func SelectUserByID(id string) UserProfile {
    ...
}

我们也可以在其他域方法中重复使用方法SelectUserByID。我们需要一种将SelectUserByID注入到GetUserProfile的方法，以至于我们可以在测试中模拟数据访问，进行单元测试GetUserProfile方法。在Go中一种实现的方法是为函数定义一个类型别名。

类型别名

使GetUserProfile方法依赖一个抽象，意味着我们可以在测试中注入一个模拟的数据访问层。两种常用的方式分别是使用接口，或类型别名。类型别名很简单，不需要生成一个可以在这儿使用的新结构。接下来为这两个方法都声明别名。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
type SelectUserByID func(id string) User

type GetUserProfile func(id string) UserProfile

func NewGetUserProfile(selectUser SelectUserByID) GetUserProfile {
    return func(id string) string {
        user := selectUser(id)
        return user.ProfileText
    }
}

func selectUser(id string) User {
    ...
    return User{ProfileText: userRow.ProfileText}
}

SelectUserByID是一个传入用户ID，返回一个User的方法。我们没有定义它的实现。NewGetUserProfile是一个输入方法selectUser，然后返回一个可以被调用者使用的方法，的工厂方法。这种策略使用一个闭包，来提供一个内部方法可以被外部方法依赖的入口。闭包可以获得上下文定义的变量和常量。这被称为这些变量和常量的封闭。

可以像下面这样调用这些域函数。

1
2
3
4
// 在应用中的直接依赖
getUser := NewGetUserProfile(selectUser)

user := getUser("1234")

另一种方式

如果你熟悉其他类似Java的语言，这就像创建一个类，注入这个类的依赖到构造函数中，然后通过方法访问依赖。这种访问方式没有方法上的差异——可以将函数别名当作一个只有一个简单抽象方法（SAM）的接口。在Java中，可以使用构造函数来注入依赖。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
interface DB {
    User SelectUser(String id)
}

public class UserService {
    private final DB db;
    
    public UserService(DB db) { // 将依赖注入的构造函数
        this.DB = db;
    }

    public UserProfile getUserProfile(String id) { // 访问方法
        User user = this.DB.SelectUser(id);
        ...
        return userProfile;
    }
}

而在Go中使用高阶函数实现等价功能的方式如下：

1
2
3
4
5
6
7
8
9
10
11
12
type SelectUser func(id string) User

type GetUserProfile func(id string) UserProfile

func NewGetUserProfile(selectUser SelectUser) { // 注入依赖的工厂方法
    return func(id string) UserProfile { // 访问方法
        user := selectUser(id)
        ...
        return userProfile        
    } 
}

测试

现在可以通过模拟数据访问层来单元测试业务层方法。

1
2
3
4
5
6
7
8
9
10
func TestGetUserProfile(t *testing.T) {
    selectUserMock := func(id string) User {
        return User{name: "jan"}
    }
    getUser := NewGetUserProfile(selectUserMock)

    user := getUser("12345")

    assert.Equal(t, UserProfile{ID: "12345", Name: "jan"}, user)
}

你可以在github.com/steinfletcher/func-dependency-injection-go找到更完整的代码示例。该示例包含一个公开REST端口的http服务。

译者注：
本文翻译可能不够完美，如有错误请指出，我将改正。