快捷導(dǎo)航

淺析golang的依賴(lài)注入

更新時(shí)間：2022年09月28日 09:15:49 作者：bobo_simpler

這篇文章主要介紹了淺析golang的依賴(lài)注入，文章圍繞主題展開(kāi)詳細(xì)的內(nèi)容介紹，具有一定的參考價(jià)值，需要的小伙伴可以參考一下

前言

如果是做web開(kāi)發(fā)，對(duì)依賴(lài)注入肯定不陌生，java程序員早就習(xí)慣了spring提供的依賴(lài)注入，做業(yè)務(wù)開(kāi)發(fā)時(shí)非常方便，只關(guān)注業(yè)務(wù)邏輯即可，對(duì)象之間的依賴(lài)關(guān)系都交給框架。

golang是強(qiáng)類(lèi)型語(yǔ)言，編譯后是機(jī)器碼，所以一般使用反射或 代碼生成 解決依賴(lài)注入的問(wèn)題

基于反射的DI

基于反射解決DI問(wèn)題的框架, 使用比較多的是Uber的 dig 庫(kù)

官方的例子：

type Config struct {
        Prefix string
}

//初始化Config函數(shù)
func NewConfig()(*Config, error) {
        // In a real program, the configuration will probably be read from a
        // file.
        var cfg Config
        err := json.Unmarshal([]byte(`{"prefix": "[foo] "}`), &cfg)
        return &cfg, err
}

//初始化logger函數(shù)
func NewLogger(cfg *Config) *log.Logger {
    return log.New(os.Stdout, cfg.Prefix, 0)
}

func Handle() (l *log.Logger) {
    l.Print("You've been invoked")
}

func main() {
    //初始化dig對(duì)象
    c := dig.New()

    //Provide方法用來(lái)設(shè)置依賴(lài)的對(duì)象
    er := c.Provide(NewConfig)
    if err != nil {
        panic(err)
    }

    //設(shè)置依賴(lài)的對(duì)象
    err = c.Provide(NewLogger)
    if err != nil {
        panic(err)
    }
    
    //執(zhí)行Handle()方法
    //Handle依賴(lài) Config 和 Logger，使用Invoke執(zhí)行方法時(shí)會(huì)自動(dòng)注入依賴(lài)（依賴(lài)的對(duì)象要傳入Provide方法中）
    err = c.Invoke(Handle)
    if err != nil {
        panic(err)
    }
    // Output:
    // [foo] You've been invoked
}

dig提供了一個(gè)容器(container)，所有的依賴(lài)項(xiàng)通過(guò)Provide方法添加，執(zhí)行某個(gè)方法時(shí)使用Invoke方法，該方法會(huì)自動(dòng)注入所需要的依賴(lài)。

dig使用反射機(jī)制解決DI問(wèn)題，所以代碼執(zhí)行性能上會(huì)有損耗

并且因?yàn)槭褂梅瓷渌钥赡艹霈F(xiàn)編譯時(shí)沒(méi)有錯(cuò)誤，執(zhí)行時(shí)報(bào)空指針

詳情使用方法可以參考官方文檔，dig可以繼承到gin框架中，有興趣的可以看看資料。

筆者不太喜歡這種使用方式，為了依賴(lài)注入破壞了代碼原有的調(diào)用方式。

基于代碼生成的DI

wire庫(kù)是google出的解決golang DI問(wèn)題的工具，它可以自動(dòng)生成依賴(lài)注入的代碼，節(jié)省了手動(dòng)去處理依賴(lài)關(guān)系

github地址

wire對(duì)原有代碼的侵入度很低，開(kāi)發(fā)過(guò)程中，在依賴(lài)注入代碼處調(diào)用Build方法（例子中是初始化controller對(duì)象）就可以了

// +build wireinject

package main

import (
    "encoding/json"
    "fmt"
    "github.com/google/wire"
    "net/http"
)

type DataSource struct {
    Operation string
}

func NewDataSource() DataSource {
    return DataSource{Operation: "operation_name"}
}

//==================

type Dao struct {
    DataSource DataSource
}

func NewDao(ds DataSource) *Dao {
    return &Dao{
        DataSource: ds,
    }
}

func (d *Dao) GetItemList() ([]string, error) {
    //TODO 拿到DB對(duì)象做查詢(xún)操作
    fmt.Printf("db object: %s", d.DataSource.Operation)
    return []string{d.DataSource.Operation, "item1", "item2"}, nil
}

//====================

type Service struct {
    Dao *Dao
}

func NewService(dao *Dao) *Service {
    return &Service{Dao: dao}
}

func (s *Service) GetItemList() ([]string, error) {
    return s.Dao.GetItemList()
}

//=====================

type Controller struct {
    Service *Service
}

func NewController(service *Service) *Controller {
    return &Controller{Service: service}
}

func (c *Controller) GetItemList() ([]string, error) {
    return c.Service.GetItemList()
}

var MegaSet = wire.NewSet(NewDataSource, NewDao, NewService, NewController)

func initializeController() *Controller {
    wire.Build(MegaSet)
    return &Controller{}
}

func getItemList(w http.ResponseWriter, r *http.Request) {
    controller := initializeController()
    itemList, _ := controller.GetItemList()
    output, _ := json.Marshal(itemList)
    fmt.Fprintf(w, string(output))
}

func main() {
    http.HandleFunc("/items", getItemList)
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        panic(err)
    }
}

然后再項(xiàng)目根目錄執(zhí)行wire命令，會(huì)生成構(gòu)建好依賴(lài)關(guān)系的代碼(以_gen結(jié)尾的文件)

// Code generated by Wire. DO NOT EDIT.

//go:generate go run github.com/google/wire/cmd/wire
//+build !wireinject

package main

import (
    "encoding/json"
    "fmt"
    "github.com/google/wire"
    "net/http"
)

// Injectors from main.go:
// 此處是生成的代碼
func initializeController() *Controller {
    dataSource := NewDataSource()
    dao := NewDao(dataSource)
    service := NewService(dao)
    controller := NewController(service)
    return controller
}

// main.go:

type DataSource struct {
    Operation string
}

func NewDataSource() DataSource {
    return DataSource{Operation: "operation_name"}
}

type Dao struct {
    DataSource DataSource
}

func NewDao(ds DataSource) *Dao {
    return &Dao{
        DataSource: ds,
    }
}

func (d *Dao) GetItemList() ([]string, error) {
    fmt.Printf("db object: %s", d.DataSource.Operation)
    return []string{d.DataSource.Operation, "item1", "item2"}, nil
}

type Service struct {
    Dao *Dao
}

func NewService(dao *Dao) *Service {
    return &Service{Dao: dao}
}

func (s *Service) GetItemList() ([]string, error) {
    return s.Dao.GetItemList()
}

type Controller struct {
    Service *Service
}

func NewController(service *Service) *Controller {
    return &Controller{Service: service}
}

func (c *Controller) GetItemList() ([]string, error) {
    return c.Service.GetItemList()
}

var MegaSet = wire.NewSet(NewDataSource, NewDao, NewService, NewController)

func getItemList(w http.ResponseWriter, r *http.Request) {
    controller := initializeController()
    itemList, _ := controller.GetItemList()
    output, _ := json.Marshal(itemList)
    fmt.Fprintf(w, string(output))
}

func main() {
    http.HandleFunc("/items", getItemList)
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        panic(err)
    }
}

關(guān)鍵代碼：

//執(zhí)行wire命令前的代碼
func initializeController() *Controller {
    wire.Build(MegaSet)
    return &Controller{}
}

//執(zhí)行后生成的代碼
// Injectors from main.go:
func initializeController() *Controller {
    dataSource := NewDataSource()
    dao := NewDao(dataSource)
    service := NewService(dao)
    controller := NewController(service)
    return controller
}

通過(guò)生成代碼解決依賴(lài)注入的問(wèn)題，既能提升開(kāi)發(fā)效率，又不影響代碼性能，wire更高級(jí)的用法可以去github document查看