istio
1// Copyright Istio Authors
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package sleep
16
17import (
18"context"
19"time"
20)
21
22// UntilContext sleeps for the given duration, or until the context is complete.
23// Returns true if the sleep completes the full duration
24func UntilContext(ctx context.Context, d time.Duration) bool {
25return Until(ctx.Done(), d)
26}
27
28// Until sleeps for the given duration, or until the channel is closed.
29// Returns true if the sleep completes the full duration
30func Until(ch <-chan struct{}, d time.Duration) bool {
31timer := time.NewTimer(d)
32select {
33case <-ch:
34if !timer.Stop() {
35<-timer.C
36}
37return false
38case <-timer.C:
39return true
40}
41}
42