needle/replace.go at development · danpasecinic/needle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package needle

import (
	"context"

	"github.com/danpasecinic/needle/internal/container"
	"github.com/danpasecinic/needle/internal/reflect"
)

func Replace[T any](c *Container, provider Provider[T], opts ...ProviderOption) error {
	cfg := &providerConfig{}
	for _, opt := range opts {
		opt(cfg)
	}

	key := reflect.TypeKey[T]()
	if cfg.name != "" {
		key = reflect.TypeKeyNamed[T](cfg.name)
	}

	resolver := c.resolver
	wrappedProvider := func(ctx context.Context, r container.Resolver) (any, error) {
		return provider(ctx, resolver)
	}

	if err := c.internal.Replace(key, wrappedProvider, cfg.dependencies); err != nil {
		return err
	}

	for _, hook := range cfg.onStart {
		c.internal.AddOnStart(key, hook)
	}
	for _, hook := range cfg.onStop {
		c.internal.AddOnStop(key, hook)
	}

	if cfg.scope != 0 {
		c.internal.SetScope(key, cfg.scope)
	}
	if cfg.poolSize > 0 {
		c.internal.SetPoolSize(key, cfg.poolSize)
	}
	if cfg.lazy {
		c.internal.SetLazy(key, true)
	}

	return nil
}

func ReplaceValue[T any](c *Container, value T, opts ...ProviderOption) error {
	cfg := &providerConfig{}
	for _, opt := range opts {
		opt(cfg)
	}

	key := reflect.TypeKey[T]()
	if cfg.name != "" {
		key = reflect.TypeKeyNamed[T](cfg.name)
	}

	if err := c.internal.ReplaceValue(key, value); err != nil {
		return err
	}

	for _, hook := range cfg.onStart {
		c.internal.AddOnStart(key, hook)
	}
	for _, hook := range cfg.onStop {
		c.internal.AddOnStop(key, hook)
	}

	return nil
}

func ReplaceNamed[T any](c *Container, name string, provider Provider[T], opts ...ProviderOption) error {
	opts = append(opts, WithName(name))
	return Replace(c, provider, opts...)
}

func ReplaceNamedValue[T any](c *Container, name string, value T, opts ...ProviderOption) error {
	opts = append(opts, WithName(name))
	return ReplaceValue(c, value, opts...)
}

func MustReplace[T any](c *Container, provider Provider[T], opts ...ProviderOption) {
	if err := Replace(c, provider, opts...); err != nil {
		panic(err)
	}
}

func MustReplaceValue[T any](c *Container, value T, opts ...ProviderOption) {
	if err := ReplaceValue(c, value, opts...); err != nil {
		panic(err)
	}
}

func ReplaceFunc[T any](c *Container, constructor any, opts ...ProviderOption) error {
	provider, depOpts, err := buildFuncProvider[T](c, constructor)
	if err != nil {
		return err
	}

	opts = append(depOpts, opts...)
	return Replace(c, provider, opts...)
}

func MustReplaceFunc[T any](c *Container, constructor any, opts ...ProviderOption) {
	if err := ReplaceFunc[T](c, constructor, opts...); err != nil {
		panic(err)
	}
}

func ReplaceStruct[T any](c *Container, opts ...ProviderOption) error {
	provider, depOpts := buildStructProvider[T](c)
	opts = append(depOpts, opts...)
	return Replace(c, provider, opts...)
}

func MustReplaceStruct[T any](c *Container, opts ...ProviderOption) {
	if err := ReplaceStruct[T](c, opts...); err != nil {
		panic(err)
	}
}