fix: barry quick fix, 2025-06-07 14:40:41

Author: kooksee

CHANGELOG.md

@@ -1,8 +1,11 @@
 package dix
 
 import (
+	"fmt"
+	"github.com/pubgo/funk/errors"
+	"reflect"
+
 	"github.com/pubgo/dix/dixinternal"
-	"github.com/pubgo/funk/assert"
 )
 
 const (
@@ -28,15 +31,12 @@ func New(opts ...Option) Container {
 
 // Inject 统一的依赖注入方法
 //
-// 这是框架的核心方法，支持多种输入类型，提供了统一的依赖注入接口。
-// 一个方法处理所有依赖注入需求，包括获取实例和注入依赖。
+// 这是框架的核心方法，按照最原始的设计，只支持函数和结构体类型。
+// 提供了统一的依赖注入接口，简化API设计。
 //
 // 支持的输入类型：
 //   - 函数：func(deps...) - 解析函数参数并调用函数
 //   - 结构体指针：&struct{} - 注入到结构体字段
-//   - 接口：interface{} - 支持接口类型注入
-//   - 切片：[]T - 注入所有匹配的实例
-//   - 映射：map[string]T - 注入带名称的实例
 //
 // 函数注入规则：
 //   - 函数只能有入参，不能有出参
@@ -62,33 +62,44 @@ func New(opts ...Option) Container {
 //
 // 示例：
 //
-//	// 函数注入（替代传统启动函数）
-//	err := dix.Inject(container, func(logger Logger, db *Database, handlers []Handler) {
+//	// 函数注入
+//	_, err := dix.Inject(container, func(logger Logger, db *Database, handlers []Handler) {
 //	    // 使用注入的依赖启动服务器
 //	    startServer(logger, db, handlers)
 //	})
+//	if err != nil {
+//	    log.Fatal(err)
+//	}
 //
 //	// 结构体注入
 //	type Service struct {
 //	    Logger Logger
 //	    DB     *Database
 //	}
-//	var service Service
-//	err := dix.Inject(container, &service)
+//	service, err := dix.Inject(container, &Service{})
+//	if err != nil {
+//	    log.Fatal(err)
+//	}
 //
 //	// 获取单个依赖实例
 //	var logger Logger
-//	err := dix.Inject(container, func(l Logger) {
+//	_, err := dix.Inject(container, func(l Logger) {
 //	    logger = l
 //	})
+//	if err != nil {
+//	    log.Fatal(err)
+//	}
 //
 //	// 批量获取多个依赖实例
 //	var logger Logger
 //	var db *Database
 //	var handlers []Handler
-//	err := dix.Inject(container, func(l Logger, d *Database, h []Handler) {
+//	_, err := dix.Inject(container, func(l Logger, d *Database, h []Handler) {
 //	    logger, db, handlers = l, d, h
 //	})
+//	if err != nil {
+//	    log.Fatal(err)
+//	}
 //
 //	// 方法注入示例
 //	type UserService struct {
@@ -98,18 +109,33 @@ func New(opts ...Option) Container {
 //	func (s *UserService) DixInjectLogger(logger Logger) { s.logger = logger }
 //	func (s *UserService) DixInjectDatabase(db *Database) { s.db = db }
 //
-//	var service UserService
-//	err := dix.Inject(container, &service)
+//	service, err := dix.Inject(container, &UserService{})
+//	if err != nil {
+//	    log.Fatal(err)
+//	}
 //
 // 参数：
 //   - container: 依赖注入容器
-//   - target: 注入目标（函数、结构体指针、接口等）
+//   - target: 注入目标（函数或结构体指针）
 //   - opts: 注入选项（可选）
 //
 // 返回值：
+//   - T: 如果target是函数，返回零值（nil）；如果是结构体，返回注入后的结构体
 //   - error: 注入失败时的错误信息
-func Inject(container Container, target interface{}, opts ...Option) error {
-	return container.Inject(target, opts...)
+func Inject[T any](container Container, target T, opts ...Option) (T, error) {
+	vp := reflect.ValueOf(target)
+	if vp.Kind() == reflect.Struct {
+		if err := container.Inject(&target, opts...); err != nil {
+			return target, fmt.Errorf("dix: inject failed: %w", err)
+		}
+		return target, nil
+	} else {
+		return target, container.Inject(target, opts...)
+	}
+}
+
+func InjectMust[T any](container Container, target T, opts ...Option) T {
+	return errors.Must1(Inject(container, target, opts...))
 }
 
 // Provide 注册依赖提供者
@@ -165,7 +191,7 @@ func Inject(container Container, target interface{}, opts ...Option) error {
 //   - container: 依赖注入容器
 //   - provider: 提供者函数
 func Provide(container Container, provider any) {
-	assert.Must(container.Provide(provider))
+	errors.Must(container.Provide(provider))
 }
 
 // GetGraph 获取依赖关系图
@@ -174,7 +200,3 @@ func Provide(container Container, provider any) {
 func GetGraph(container Container) *Graph {
 	return container.Graph()
 }
-
-// 为了向后兼容，保留旧的类型别名
-// Deprecated: 使用 Container 替代
-type Dix = Container

dix.go

@@ -1,10 +1,10 @@
 package dixglobal
 
 import (
+	"github.com/pubgo/funk/errors"
 	"reflect"
 
 	"github.com/pubgo/dix/dixinternal"
-	"github.com/pubgo/funk/assert"
 )
 
 var _container = dixinternal.New(dixinternal.WithValuesNull())
@@ -35,7 +35,7 @@ var _container = dixinternal.New(dixinternal.WithValuesNull())
 
 // Provide registers an object constructor
 func Provide(provider any) {
-	assert.Must(_container.Provide(provider))
+	errors.Must(_container.Provide(provider))
 }
 
 // Inject 统一的依赖注入方法
@@ -57,9 +57,9 @@ func Provide(provider any) {
 func Inject[T any](target T, opts ...dixinternal.Option) T {
 	vp := reflect.ValueOf(target)
 	if vp.Kind() == reflect.Struct {
-		assert.Must(_container.Inject(&target, opts...))
+		errors.Must(_container.Inject(&target, opts...))
 	} else {
-		assert.Must(_container.Inject(target, opts...))
+		errors.Must(_container.Inject(target, opts...))
 	}
 	return target
 }

dixglobal/global.go

@@ -89,27 +89,6 @@ func (c *ContainerImpl) Inject(target interface{}, opts ...Option) error {
 	return c.injector.InjectTarget(target, mergedOpts)
 }
 
-// get 获取指定类型的实例（内部方法）
-func (c *ContainerImpl) get(typ reflect.Type, opts ...Option) (interface{}, error) {
-	// 合并选项
-	mergedOpts := c.options
-	for _, opt := range opts {
-		opt(&mergedOpts)
-	}
-
-	// 解析依赖
-	value, err := c.resolver.Resolve(typ, mergedOpts)
-	if err != nil {
-		return nil, err
-	}
-
-	if !value.IsValid() {
-		return nil, NewNotFoundError(typ)
-	}
-
-	return value.Interface(), nil
-}
-
 // Graph 获取依赖关系图
 func (c *ContainerImpl) Graph() *Graph {
 	renderer := NewDotRenderer()

dixinternal/container.go

@@ -2,13 +2,20 @@ package main
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/pubgo/dix/dixglobal"
 	"github.com/pubgo/funk/recovery"
 )
 
 func main() {
-	defer recovery.Exit()
+	defer recovery.Exit(func(err error) error {
+		if strings.Contains(err.Error(), "circular dependency detected") {
+			return nil
+		}
+		return err
+	})
+	
 	defer func() {
 		fmt.Println("\n=== Final Dependency Graph ===")
 		graph := dixglobal.Graph()
@@ -36,13 +43,8 @@ func main() {
 	})
 
 	fmt.Println("Registering C provider (depends on A)")
+
 	dixglobal.Provide(func(a *A) *C {
 		return &C{Name: "C depends on " + a.Name}
 	})
-
-	fmt.Println("\n=== Attempting to Inject (should detect cycle) ===")
-	// 尝试注入，这应该会检测到循环依赖并panic
-	dixglobal.Inject(func(c *C) {
-		fmt.Println("Successfully injected C:", c.Name)
-	})
 }

CONTRIBUTING.md renamed to docs/CONTRIBUTING.md

@@ -139,7 +139,7 @@ func main() {
 	// 示例1：正常获取实例
 	fmt.Println("=== 正常实例获取 ===")
 	var userService *UserService
-	err := dix.Inject(container, func(us *UserService) {
+	_, err := dix.Inject(container, func(us *UserService) {
 		userService = us
 	})
 	if err != nil {
@@ -171,7 +171,7 @@ func main() {
 
 	// 演示配置错误
 	fmt.Println("尝试注入 Config (期望错误):")
-	err = dix.Inject(errorContainer, func(c *Config) {
+	_, err = dix.Inject(errorContainer, func(c *Config) {
 		// 这个函数不会被调用，因为提供者会失败
 	})
 	if err != nil {
@@ -180,7 +180,7 @@ func main() {
 
 	// 演示数据库错误
 	fmt.Println("尝试注入 Logger (期望错误):")
-	err = dix.Inject(errorContainer, func(l Logger) {
+	_, err = dix.Inject(errorContainer, func(l Logger) {
 		// 这个函数不会被调用，因为提供者会失败
 	})
 	if err != nil {
@@ -207,7 +207,7 @@ func main() {
 	// 成功的提供者
 	fmt.Println("获取成功的Logger:")
 	var logger Logger
-	err = dix.Inject(mixedContainer, func(l Logger) {
+	_, err = dix.Inject(mixedContainer, func(l Logger) {
 		logger = l
 	})
 	if err != nil {
@@ -218,7 +218,7 @@ func main() {
 
 	// 失败的提供者
 	fmt.Println("尝试获取失败的Config:")
-	err = dix.Inject(mixedContainer, func(c *Config) {
+	_, err = dix.Inject(mixedContainer, func(c *Config) {
 		// 不会执行到这里
 	})
 	if err != nil {

example/cycle/main.go

@@ -100,7 +100,7 @@ func main() {
 	// 示例1：正常获取实例
 	fmt.Println("=== 正常实例获取 ===")
 	var userService *UserService
-	err := dix.Inject(container, func(us *UserService) {
+	_, err := dix.Inject(container, func(us *UserService) {
 		userService = us
 	})
 	if err != nil {
@@ -140,7 +140,7 @@ func main() {
 
 	// 演示配置错误
 	fmt.Println("尝试注入 Config (期望错误):")
-	err = dix.Inject(errorContainer, func(c *Config) {
+	_, err = dix.Inject(errorContainer, func(c *Config) {
 		// 这个函数不会被调用，因为提供者会失败
 	})
 	if err != nil {
@@ -149,7 +149,7 @@ func main() {
 
 	// 演示数据库错误
 	fmt.Println("尝试注入 Database (期望错误):")
-	err = dix.Inject(errorContainer, func(d Database) {
+	_, err = dix.Inject(errorContainer, func(d Database) {
 		// 这个函数不会被调用，因为提供者会失败
 	})
 	if err != nil {
@@ -177,7 +177,7 @@ func main() {
 	// 成功的提供者
 	fmt.Println("获取成功的字符串:")
 	var str *string
-	err = dix.Inject(mixedContainer, func(s *string) {
+	_, err = dix.Inject(mixedContainer, func(s *string) {
 		str = s
 	})
 	if err != nil {
@@ -188,7 +188,7 @@ func main() {
 
 	// 失败的提供者
 	fmt.Println("尝试获取失败的int:")
-	err = dix.Inject(mixedContainer, func(i *int) {
+	_, err = dix.Inject(mixedContainer, func(i *int) {
 		// 不会执行到这里
 	})
 	if err != nil {

example/provider-error-correct/main.go

@@ -875,6 +875,7 @@ github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw
 github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
 github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
 github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
+github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
 github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
 github.com/golang/snappy v0.0.3/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
 github.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=