golang fx 依赖注入包学习笔记
package main
import (
"context"
"fmt"
"log"
"net/http"
"os"
"time"
"/fx"
)
// 这里把main函数放到最前面，可以清晰的先了解代码调用方法，具体函数定义，自行跳跃过去查看
func main() {
// 整个fx包执行的顺序是
// 1. 先执行中的函数列表，按顺序一个一个执行
// 2. 中构造函数，在Invoke需要的时候，再去执行
// 执行Invoke中的函数时，当前执行的函数传入参数如果用到的变量，则先调用其构造函数
// 这里构造函数构造出来的变量不需要明显的进行定义，会自动传给invoke函数
// 比如nothingUserInvoke 这里没有任何传入参数，则在它之前不执行任何构造函数
//
// Register执行时，需要mux *, h , logger * 三个传入参数，则执行对应的三个构造函数
// 但是在执行 NewHandler构造函数时，需要logger，则在其之前执行NewLogger
//
// invokeUseMyconstruct 执行时，需要先执行 NewMyConstruct
// 至于在 中注册的Onstart OnStop 函数，是在app start 之后，按构造函数的顺序来执行，stop时，按相反顺序执行
app := fx.New(
// 一系列构造函数
fx.Provide(
NewMyConstruct,
NewHandler,
NewMux,
NewLogger,
),
// 构造函数执行完后，执行初始化函数
fx.Invoke(invokeNothingUse, invokeRegister, invokeAnotherFunc, invokeUseMyconstruct),
)
startCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
if err := app.Start(startCtx); err != nil {
log.Fatal(err)
}
http.Get("http://localhost:8080/")
stopCtx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
if err := app.Stop(stopCtx); err != nil {
log.Fatal(err)
}
}
// * 类型对象的构造函数 （注意：这里指针与非指针类型是严格区分的）
func NewLogger(lc fx.Lifecycle) *log.Logger {
logger := log.New(os.Stdout, "" /* prefix */, 0 /* flags */)
logger.Print("Executing NewLogger.")
lc.Append(fx.Hook{
OnStart: func(i context.Context) error {
logger.Println("logger onstart..")
return nil
},
OnStop: func(i context.Context) error {
logger.Println("logger onstop..")
return nil
},
})
return logger
}
// 类型对象的构造函数，它输入参数中需要*类型，所以在它执行之前，先执行NewLogger
func NewHandler(lc fx.Lifecycle, logger *log.Logger) (http.Handler, error) {
logger.Print("Executing NewHandler.")
lc.Append(fx.Hook{
OnStart: func(i context.Context) error {
logger.Println("handler onstart..")
return nil
},
OnStop: func(i context.Context) error {
logger.Println("handler onstop..")
return nil
},
})
return http.HandlerFunc(func(http.ResponseWriter, *http.Request) {
logger.Print("Got a request.")
}), nil
}
// * 类型对象的构造函数，它输入参数中需要*类型，所以在它执行之前，先执行NewLogger
func NewMux(lc fx.Lifecycle, logger *log.Logger) *http.ServeMux {
logger.Print("Executing NewMux.")
mux := http.NewServeMux()
server := &http.Server{
Addr: ":8080",
Handler: mux,
}
lc.Append(fx.Hook{
OnStart: func(context.Context) error {
logger.Print("Starting HTTP server.")
go server.ListenAndServe()
return nil
},
OnStop: func(ctx context.Context) error {
logger.Print("Stopping HTTP server.")
return server.Shutdown(ctx)
},
})
return mux
}
// 自定义类型
type mystruct struct{}
// mystruct 构造函数
func NewMyConstruct(logger *log.Logger) mystruct {
logger.Println("Executing NewMyConstruct.")
return mystruct{}
}
// invokeUseMyconstruct 是invoke函数，在其执行之前，需要执行 NewMyConstruct
func invokeUseMyconstruct(logger *log.Logger, c mystruct) {
logger.Println("invokeUseMyconstruct..")
}
func invokeRegister(mux *http.ServeMux, h http.Handler, logger *log.Logger) {
logger.Println("invokeRegiste...")
mux.Handle("/", h)
}
func invokeAnotherFunc(logger *log.Logger) {
logger.Println("invokeAnotherFunc...")
}
// 这个invoke函数，不依赖任何输入变量，所以不需要执行任何构造函数
func invokeNothingUse() {
fmt.Println("invokeNothingUse...")
}