ลองใช้ Google Wire เพื่อทำ Dependency Injection ใน Go

Wire is a code generation tool that automates connecting components using dependency injection

ลองใช้ Wire กับ Project API Todo แบบง่ายๆ

เครื่องมือที่ใช้

• golang
• go fiber
• postgres

ขั้นตอน

1. เริ่มจากสร้างโปรเจคขึ้นมา 1 โปรเจคโดยมีไฟล์ในโฟรเดอร์ดังนี้

• go.mod
• cmd
  • webservice
  • main.go
  • wire.go
• controllers
  • todo_controller.go
• services
  • config
  • env.go
  • db
  • gorm.go
  • server
  • fiber.go

1. โฟกัสที่ file cmd/webservice/main.go สร้าง struct ขึ้นมาหนึ่งตัวและ function ที่ทำหน้าที่สร้าง struct ตัวนั้นเสมือนเป็น constructor

// vvvvvvvv สนใจ struct ตัวนี้ vvvvvvvv type WebService struct { ENVConfig *config.ENVConfig DB *gorm.DB TodoController *controllers.TodoController } // vvvvvvvv สนใจ function ตัวนี้ vvvvvvvv func NewWebService(envConfig *config.ENVConfig, db *gorm.DB, todoController *controllers.TodoController) *WebService { return &WebService{ DB: db, ENVConfig: envConfig, TodoController: todoController, } } func main() { // ยังไม่ต้องสนใจตรงนี้ } 

1. ไปที่ file cmd/webservice/wire.go จะเห็นว่าที่ wire.Build(...) ใน ... จะต้องมี function ที่ return struct ที่เราสนใจจะใช้งานใน struct Webservice

// vvvvvvvv ต้องใส่นะ vvvvvvvv //go:build wireinject // +build wireinject func InitWebservice() *Webservice { wire.Build(NewWebService, config.NewENVConfig, db.NewDB, controllers.NewTodoController) return &Webservice{} } 

1. กลับมาที่ file cmd/webservice/main.go แล้ว run คำสั่งใน terminal run คำสั่ง wire แล้วมันจะสร้าง file มาให้อันนึงชื่อ wire_gen.go ซึ่งในนั้นจะมี function InitWebservice มาให้ในแบบที่ inject ตัว dependencies ทั้งหมดที่จำเป็นมาให้แล้ว

func main() { webservice := InitWebservice() } 

1. ซึ่งถ้าต้องการเพิ่ม dependencies ต้องเพิ่ม 2 ที่
2. function New....(..., dependency ที่เราอยากเพิ่ม *DependencyStruct)
3. wire.Build(...., NewDependencyFn ที่ return dependency นั้นๆ) แล้ว wire จะไปหา dependency เหล่านั้นมาใส่ให้เราเอง ความ

ความรู้สึกหลังใช้

รู้สึกว่าไม่ค่อยจำเป็นต้องใช้มันเท่าไหร่ 555