githubmirror/trivy: Find vulnerabilities, misconfigurations, secrets, SBOM in containers, Kubernetes, code repositories, clouds and more

pkg/clock/clock.go

38 строк823 B

Teppei Fukuda

feat: add support for plugin index (#6674)

14 май 2024, 11:29

14 май 2024, 11:2926faf8f

package clock

import (
	"context"
	"time"

	"k8s.io/utils/clock"
	clocktesting "k8s.io/utils/clock/testing"
)

type (
	RealClock = clock.RealClock
	FakeClock = clocktesting.FakeClock
)

// clockKey is the context key for clock. It is unexported to prevent collisions with context keys defined in
// other packages.
type clockKey struct{}

// With returns a new context with the given time.
func With(ctx context.Context, t time.Time) context.Context {
	c := clocktesting.NewFakeClock(t)
	return context.WithValue(ctx, clockKey{}, c)
}

// Now returns the current time.
func Now(ctx context.Context) time.Time {
	return Clock(ctx).Now()
}

// Clock returns the clock from the context.
func Clock(ctx context.Context) clock.Clock {
	t, ok := ctx.Value(clockKey{}).(clock.Clock)
	if !ok {
		return RealClock{}
	}
	return t
}