Go 框架中依赖注入的替代方案
在 Go 框架中,依赖注入 (DI) 是提供依赖的一种常见方法。但是,DI 可能会引入复杂性和额外代码。对于小型到中型的项目,DI 可能不必要。
以下是一些 DI 的替代方案:
type Service struct { db *sql.DB } func NewService(db *sql.DB) *Service { return &Service{db} }
func HandleRequest(w http.ResponseWriter, r *http.Request) { db := connectToDatabase() service := NewService(db) service.HandleRequest(w, r) }
var db *sql.DB func init() { db = connectToDatabase() } func GetDB() *sql.DB { return db }
警告:全局变量可能导致难以调试和维护的耦合代码。
var globalDB *sql.DB func init() { globalDB = connectToDatabase() }
实战案例
考虑一个处理 HTTP 请求的 simple Service:
type Service struct { db *sql.DB } func (s *Service) HandleRequest(w http.ResponseWriter, r *http.Request) { // 执行数据库操作 }
使用构造函数注入,我们可以:
package main import ( "net/http" "github.com/go-sql-driver/mysql" ) type Service struct { db *sql.DB } func (s *Service) HandleRequest(w http.ResponseWriter, r *http.Request) { // 执行数据库操作 } func main() { db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/database") if err != nil { panic(err) } service := Service{db} http.HandleFunc("/", service.HandleRequest) http.ListenAndServe(":8080", nil) }
或者,使用局部作用域变量:
package main import ( "net/http" "github.com/go-sql-driver/mysql" ) func main() { db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/database") if err != nil { panic(err) } http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { service := Service{db} service.HandleRequest(w, r) }) http.ListenAndServe(":8080", nil) }