wip: 登录和认证

pkg/tool/must.go

@@ -0,0 +1,53 @@
+package tool
+
+import (
+	"context"
+	"gitea.loveuer.com/yizhisec/packages/logger"
+	"sync"
+)
+
+func Must(errs ...error) {
+	for _, err := range errs {
+		if err != nil {
+			logger.Panic(err.Error())
+		}
+	}
+}
+
+func MustWithData[T any](data T, err error) T {
+	Must(err)
+	return data
+}
+
+func MustStop(ctx context.Context, stopFns ...func(ctx context.Context) error) {
+	if len(stopFns) == 0 {
+		return
+	}
+
+	ok := make(chan struct{})
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(stopFns))
+
+	for _, fn := range stopFns {
+		go func() {
+			defer wg.Done()
+
+			if err := fn(ctx); err != nil {
+				logger.ErrorCtx(ctx, "stop function failed, err = %s", err.Error())
+			}
+		}()
+	}
+
+	go func() {
+		select {
+		case <-ctx.Done():
+			logger.FatalCtx(ctx, "stop function timeout, force down")
+		case _, _ = <-ok:
+			return
+		}
+	}()
+
+	wg.Wait()
+	close(ok)
+}