go语言实现依赖注入的示例代码

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

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

// 将被注入的记录器服务的接口，用小写的方式来隐藏出口，是一个好的方式。
type logger interface {
	Error(error)
	Info(string)
}

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

// NewCache 我们服务的构造函数，接收将被注入的服务的接口（我们可以注入几个服务），并返回结构（缓存的实例）。
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))
}

// 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,
	}
}

// 我们看到这里有四个方法，但是 cache 对象只能看到两个：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() {
	// 注册服务
	sentryService := new(logger.NormalSentryClient)
	loggerService := logger.NewLogger(sentryService)
	cacheService := cache.NewCache(loggerService)

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