Add must register func for service

SagerNet · Sep 3, 2023 · 0eec7bb · 0eec7bb
1 parent 30bf19f
commit 0eec7bb
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/service/context.go b/service/context.go
@@ -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)
+}