mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-28 10:01:14 +02:00
06bff97a3f
Conditions are now implemented as interface with two methods: `Wait` for condition to be true (cancelable via context) and 'String' which describes what condition is waiting for. Generic 'WaitForAll' was implemented to wait for multiple conditions at once. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
package conditions
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
type file string
|
|
|
|
func (filename file) Wait(ctx context.Context) error {
|
|
for {
|
|
_, err := os.Stat(string(filename))
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
|
|
if !os.IsNotExist(err) {
|
|
return err
|
|
}
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
case <-time.After(1 * time.Second):
|
|
}
|
|
}
|
|
}
|
|
|
|
func (filename file) String() string {
|
|
return fmt.Sprintf("for file %q to exist", string(filename))
|
|
}
|
|
|
|
// WaitForFileToExist is a service condition that will wait for the existence of
|
|
// a file.
|
|
func WaitForFileToExist(filename string) Condition {
|
|
return file(filename)
|
|
}
|
|
|
|
// WaitForFilesToExist is a service condition that will wait for the existence of all the files.
|
|
func WaitForFilesToExist(filenames ...string) Condition {
|
|
conditions := make([]Condition, len(filenames))
|
|
for i := range filenames {
|
|
conditions[i] = WaitForFileToExist(filenames[i])
|
|
}
|
|
return WaitForAll(conditions...)
|
|
}
|