快捷導(dǎo)航

go語言實(shí)現(xiàn)依賴注入的示例代碼

更新時(shí)間：2024年01月17日 09:09:06 作者：ldxxxxll

依賴注入和控制反轉(zhuǎn)恰恰相反,它是一種具體的編碼技巧,我們不通過 new 的方式在類內(nèi)部創(chuàng)建依賴類的對(duì)象,而是將依賴的類對(duì)象在外部創(chuàng)建好之后,通過構(gòu)造函數(shù)、函數(shù)參數(shù)等方式傳遞給類來使用,本文將給大家介紹go語言實(shí)現(xiàn)依賴注入,需要的朋友可以參考下

依賴注入(Dependency injection)

依賴注入和控制反轉(zhuǎn)恰恰相反，它是一種具體的編碼技巧。我們不通過 new 的方式在類內(nèi)部創(chuàng)建依賴類的對(duì)象，而是將依賴的類對(duì)象在外部創(chuàng)建好之后，通過構(gòu)造函數(shù)、函數(shù)參數(shù)等方式傳遞（或注入）給類來使用。

DI（依賴性注入）是一種技術(shù)，當(dāng)你的模塊間接地接受依賴性。他們不知道依賴的實(shí)現(xiàn)，只知道接口。

我們?yōu)槭裁葱枰?/h2>
DI 可以幫助我們編寫低耦合的代碼。這意味著你將能夠隨時(shí)交換你的代碼，并有助于重用它們的一些部分。
在寫單元測(cè)試時(shí)，使用 mock 方式注入是一個(gè)非常好的實(shí)踐。

參考代碼

我們有一個(gè)小小的項(xiàng)目：main.go 和兩個(gè)服務(wù)–日志和緩存

├───cmd
│       └───main.go
└───services
        ├───logger
        │       └───logger.go
        └───cache
                └───cache.go

它們是如何工作的？我們有一個(gè)緩存服務(wù)，如果你想把一些東西存儲(chǔ)在像Redis這樣的快速存儲(chǔ)中，就可以使用它。

// cache.go
import (
	"fmt"
	"sync"
)

// 將被注入的記錄器服務(wù)的接口，用小寫的方式來隱藏出口，是一個(gè)好的方式。
type logger interface {
	Error(error)
	Info(string)
}

type Cache struct {
	logger logger
	m      *sync.Map
}

// NewCache 我們服務(wù)的構(gòu)造函數(shù)，接收將被注入的服務(wù)的接口（我們可以注入幾個(gè)服務(wù)），并返回結(jié)構(gòu)（緩存的實(shí)例）。
func NewCache(logger logger) *Cache {
	return &Cache{
		logger: logger,
		m:      new(sync.Map),
	}
}

func (r *Cache) Get(key string) (string, error) {
	rawValue, ok := r.m.Load(key)
	if !ok {
		err := fmt.Errorf("not found: %s", key)
		r.logger.Error(err)
		return "", err
	}
	return rawValue.(string), nil
}

func (r *Cache) Set(key string, data []byte) {
	r.m.Store(key, data)
	r.logger.Info(fmt.Sprintf("store key: %s", key))
}

日志服務(wù)用來對(duì)日志的持久化存儲(chǔ)以及分析

// logger.go
import "fmt"

type sentryClient interface {
	sendMessage(interface{})
}

type NormalSentryClient struct{}

func (s *NormalSentryClient) sendMessage(i interface{}) {
	fmt.Printf("send message %+v to sentry\n", i)
}

type Logger struct {
	sentry sentryClient
}

func NewLogger(sentryClient sentryClient) *Logger {
	return &Logger{
		sentry: sentryClient,
	}
}

// 我們看到這里有四個(gè)方法，但是 cache 對(duì)象只能看到兩個(gè)：Error 以及 Info
func (l *Logger) Error(e error) {
	fmt.Printf("[ERROR] %+v\n", e)
}

func (l *Logger) Info(msg string) {
	fmt.Printf("[INFO] %s\n", msg)
}

func (l *Logger) Debug(msg string) {
	fmt.Printf("[DEBUG] %s\n", msg)
}

func (l *Logger) Log(msg string) {
	fmt.Printf("[LOG] %s\n", msg)
}

// main.go
func main() {
	// 注冊(cè)服務(wù)
	sentryService := new(logger.NormalSentryClient)
	loggerService := logger.NewLogger(sentryService)
	cacheService := cache.NewCache(loggerService)

	cacheService.Set("test", []byte("test"))
	cacheService.Get("test1")
}

輸出：