talos/pkg/conditions/files.go
Andrey Smirnov 9379cf9ee1 refactor: expose provision as public package
This change is only moving packages and updating import paths.

Goal: expose `internal/pkg/provision` as `pkg/provision` to enable other
projects to import Talos provisioning library.

As cluster checks are almost always required as part of provisioning
process, package `internal/pkg/cluster` was also made public as
`pkg/cluster`.

Other changes were direct dependencies discovered by `importvet` which
were updated.

Public packages (useful, general purpose packages with stable API):

* `internal/pkg/conditions` -> `pkg/conditions`
* `internal/pkg/tail` -> `pkg/tail`

Private packages (used only on provisioning library internally):

* `internal/pkg/inmemhttp` -> `pkg/provision/internal/inmemhttp`
* `internal/pkg/kernel/vmlinuz` -> `pkg/provision/internal/vmlinuz`
* `internal/pkg/cniutils` -> `pkg/provision/internal/cniutils`

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2020-08-12 05:12:05 -07:00

54 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("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...)
}