putting in base query and collector framework

2024-12-18 18:52:38 +00:00
parent b0040f0cbe
commit 2209b877f5
1739 changed files with 288187 additions and 288200 deletions
@@ -1,82 +1,82 @@
-package wait
-
-import (
-	"context"
-	"fmt"
-	"time"
-)
-
-// Implement interface
-var (
-	_ Strategy        = (*MultiStrategy)(nil)
-	_ StrategyTimeout = (*MultiStrategy)(nil)
-)
-
-type MultiStrategy struct {
-	// all Strategies should have a startupTimeout to avoid waiting infinitely
-	timeout  *time.Duration
-	deadline *time.Duration
-
-	// additional properties
-	Strategies []Strategy
-}
-
-// WithStartupTimeoutDefault sets the default timeout for all inner wait strategies
-func (ms *MultiStrategy) WithStartupTimeoutDefault(timeout time.Duration) *MultiStrategy {
-	ms.timeout = &timeout
-	return ms
-}
-
-// WithStartupTimeout sets a time.Duration which limits all wait strategies
-//
-// Deprecated: use WithDeadline
-func (ms *MultiStrategy) WithStartupTimeout(timeout time.Duration) Strategy {
-	return ms.WithDeadline(timeout)
-}
-
-// WithDeadline sets a time.Duration which limits all wait strategies
-func (ms *MultiStrategy) WithDeadline(deadline time.Duration) *MultiStrategy {
-	ms.deadline = &deadline
-	return ms
-}
-
-func ForAll(strategies ...Strategy) *MultiStrategy {
-	return &MultiStrategy{
-		Strategies: strategies,
-	}
-}
-
-func (ms *MultiStrategy) Timeout() *time.Duration {
-	return ms.timeout
-}
-
-func (ms *MultiStrategy) WaitUntilReady(ctx context.Context, target StrategyTarget) error {
-	var cancel context.CancelFunc
-	if ms.deadline != nil {
-		ctx, cancel = context.WithTimeout(ctx, *ms.deadline)
-		defer cancel()
-	}
-
-	if len(ms.Strategies) == 0 {
-		return fmt.Errorf("no wait strategy supplied")
-	}
-
-	for _, strategy := range ms.Strategies {
-		strategyCtx := ctx
-
-		// Set default Timeout when strategy implements StrategyTimeout
-		if st, ok := strategy.(StrategyTimeout); ok {
-			if ms.Timeout() != nil && st.Timeout() == nil {
-				strategyCtx, cancel = context.WithTimeout(ctx, *ms.Timeout())
-				defer cancel()
-			}
-		}
-
-		err := strategy.WaitUntilReady(strategyCtx, target)
-		if err != nil {
-			return err
-		}
-	}
-
-	return nil
-}
+package wait
+
+import (
+	"context"
+	"fmt"
+	"time"
+)
+
+// Implement interface
+var (
+	_ Strategy        = (*MultiStrategy)(nil)
+	_ StrategyTimeout = (*MultiStrategy)(nil)
+)
+
+type MultiStrategy struct {
+	// all Strategies should have a startupTimeout to avoid waiting infinitely
+	timeout  *time.Duration
+	deadline *time.Duration
+
+	// additional properties
+	Strategies []Strategy
+}
+
+// WithStartupTimeoutDefault sets the default timeout for all inner wait strategies
+func (ms *MultiStrategy) WithStartupTimeoutDefault(timeout time.Duration) *MultiStrategy {
+	ms.timeout = &timeout
+	return ms
+}
+
+// WithStartupTimeout sets a time.Duration which limits all wait strategies
+//
+// Deprecated: use WithDeadline
+func (ms *MultiStrategy) WithStartupTimeout(timeout time.Duration) Strategy {
+	return ms.WithDeadline(timeout)
+}
+
+// WithDeadline sets a time.Duration which limits all wait strategies
+func (ms *MultiStrategy) WithDeadline(deadline time.Duration) *MultiStrategy {
+	ms.deadline = &deadline
+	return ms
+}
+
+func ForAll(strategies ...Strategy) *MultiStrategy {
+	return &MultiStrategy{
+		Strategies: strategies,
+	}
+}
+
+func (ms *MultiStrategy) Timeout() *time.Duration {
+	return ms.timeout
+}
+
+func (ms *MultiStrategy) WaitUntilReady(ctx context.Context, target StrategyTarget) error {
+	var cancel context.CancelFunc
+	if ms.deadline != nil {
+		ctx, cancel = context.WithTimeout(ctx, *ms.deadline)
+		defer cancel()
+	}
+
+	if len(ms.Strategies) == 0 {
+		return fmt.Errorf("no wait strategy supplied")
+	}
+
+	for _, strategy := range ms.Strategies {
+		strategyCtx := ctx
+
+		// Set default Timeout when strategy implements StrategyTimeout
+		if st, ok := strategy.(StrategyTimeout); ok {
+			if ms.Timeout() != nil && st.Timeout() == nil {
+				strategyCtx, cancel = context.WithTimeout(ctx, *ms.Timeout())
+				defer cancel()
+			}
+		}
+
+		err := strategy.WaitUntilReady(strategyCtx, target)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}