golang 之 wire 库的使用小结

// 定义了一个消息存储接口
type Store interface {
    Save(msg Message) error
    Get() (Message, error)
}
// ProvideStore 将 InMemoryStore 绑定到 Store 接口
func ProvideStore() Store {
    return NewInMemoryStore()
}
// StoreSet 是 Wire 的 provider set
var StoreSet = wire.NewSet(
    ProvideStore,
)

// MessageService 依赖于 Store 接口
type MessageService struct {
    store store.Store
}
// Wire Provider: 告诉 Wire 如何构造 MessageService
func NewMessageService(s store.Store) *MessageService

// MessageApp 依赖于 MessageService
type MessageApp struct {
    msgService *service.MessageService
}
// Wire Provider: 告诉 Wire 如何构造 MessageApp
func NewMessageApp(service *service.MessageService) *MessageApp

// 定义依赖注入的规则
func InitializeMessageApp() (*MessageApp, error) {
    wire.Build(
        store.StoreSet,             // 提供 Store 接口的实现
        service.NewMessageService,  // 提供 MessageService 的构造函数
        NewMessageApp,              // 提供 MessageApp 的构造函数
    )
    return &MessageApp{}, nil       // 这个返回值会被 Wire 生成的代码替换
}

// Code generated by Wire. DO NOT EDIT.
//go:generate go run -mod=mod github.com/google/wire/cmd/wire
//go:build !wireinject
// +build !wireinject
package main
import (
    "example/wire/service"
    "example/wire/store"
)
// Injectors from wire.go:
// InitializeMessageApp 使用 Wire 来注入依赖并初始化应用
func InitializeMessageApp() (*MessageApp, error) {
    storeStore := store.ProvideStore()
    messageService := service.NewMessageService(storeStore)
    messageApp := NewMessageApp(messageService)
    return messageApp, nil
}