chore: import upstream snapshot with attribution

2026-07-13 12:40:33 +08:00
commit e071084ebe
982 changed files with 160368 additions and 0 deletions
@@ -0,0 +1,29 @@
+package server
+
+import (
+	"context"
+	"sync"
+)
+
+type serverKey struct{}
+type wgKey struct{}
+
+func wait(ctx context.Context) *sync.WaitGroup {
+	if ctx == nil {
+		return nil
+	}
+	wg, ok := ctx.Value(wgKey{}).(*sync.WaitGroup)
+	if !ok {
+		return nil
+	}
+	return wg
+}
+
+func FromContext(ctx context.Context) (Server, bool) {
+	c, ok := ctx.Value(serverKey{}).(Server)
+	return c, ok
+}
+
+func NewContext(ctx context.Context, s Server) context.Context {
+	return context.WithValue(ctx, serverKey{}, s)
+}