SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-03 03:47:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add must register func for service

Browse source
This commit is contained in:
世界 2023-09-03 19:41:21 +08:00
parent 30bf19f283
commit 0eec7bbe19
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
service/context.go
View file

@ -68,3 +68,19 @@ func ContextWithPtr[T any](ctx context.Context, servicePtr *T) context.Context {
registry.Register(common.DefaultValue[*T](), servicePtr)
return ctx
}
func MustRegister[T any](ctx context.Context, service T) {
registry := RegistryFromContext(ctx)
if registry == nil {
panic("missing service registry in context")
}
registry.Register(common.DefaultValue[*T](), service)
}
func MustRegisterPtr[T any](ctx context.Context, servicePtr *T) {
registry := RegistryFromContext(ctx)
if registry == nil {
panic("missing service registry in context")
}
registry.Register(common.DefaultValue[*T](), servicePtr)
}