From 0eec7bbe1934bc13adb954e77c882ac0327d63b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Sun, 3 Sep 2023 19:41:21 +0800
Subject: [PATCH] Add must register func for service

---
 service/context.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/service/context.go b/service/context.go
index 327d8ba..6817582 100644
--- 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)
+}