你可以在 github.com/steinfletcher/func-dependency-injection-go 找到完整的代码。这个例子包含了一个完整的 REST 的 HTTP 服务器。

简介

在这篇文章中介绍了一种在 go 的实现依赖注入的方法 —— 使用高阶函数和闭包

下面是一个返回用户 profile 的接口。

func GetUserProfile(id string) UserProfile {
    rows, err := db.Query("SELECT ...")
    ...
    return profileText
}

我们希望将处理用户数据的代码和访问数据库的代码分开。在这个例子中我们希望对业务逻辑代码进行单元测试，同时为访问数据库提供 mock。让我们把这些问题分开，以满足单一原则。

// domain layer function containing any business logic or mapping code
func GetUserProfile(id string) User {
    ...
}

// database access layer function
func SelectUserByID(id string) UserProfile {
    ...
}
SelectUserByIDGetUserProfileSelectUserByIDGetUserProfilego

Type aliases 

GetUserProfilego
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}
}
SelectUserByIDNewGetUserProfileselectUserclosing over

然后我们可以像这样使用这些函数

// wire dependencies somewhere in the application
getUser := NewGetUserProfile(selectUser)

user := getUser("1234")

另一种观点

如果你熟悉像 Java 这样的语言，在创建累的时候将依赖类注入构造函数，然后在调用方法的时候调用依赖的方法。这些方法不存在功能上的差别 —— 你认为这是一个拥有单个抽象方法的函数 type alias，在 Java 中在构造函数中注入依赖。

interface DB {
    User SelectUser(String id)
}

public class UserService {
    private final DB db;

    public UserService(DB db) { // inject dependency into constructor
        this.DB = db;
    }

    public UserProfile getUserProfile(String id) { // access method
        User user = this.DB.SelectUser(id);
        ...
        return userProfile;
    }
}
go
type SelectUser func(id string) User

type GetUserProfile func(id string) UserProfile

func NewGetUserProfile(selectUser SelectUser) { // factory method to inject dependency
    return func(id string) UserProfile { // access method
        user := selectUser(id)
        ...
        return userProfile        
    } 
}

测试

我们现在可以对接口进行单元测试，为数据库访问提供 mock。

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)
}

你可以在 https://github.com/steinfletcher/func-dependency-injection-go 上找到更全面的代码示例。该示例包含一个 REST 的 http 服务器。