mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-26 03:25:35 +00:00
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
|
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
// Package dirwalk contains code to walk a directory.
|
||
|
|
package dirwalk
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"io/fs"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"go4.org/mem"
|
||
|
|
)
|
||
|
|
|
||
|
|
var osWalkShallow func(name mem.RO, fn WalkFunc) error
|
||
|
|
|
||
|
|
// WalkFunc is the callback type used with WalkShallow.
|
||
|
|
//
|
||
|
|
// The name and de are only valid for the duration of func's call
|
||
|
|
// and should not be retained.
|
||
|
|
type WalkFunc func(name mem.RO, de fs.DirEntry) error
|
||
|
|
|
||
|
|
// WalkShallow reads the entries in the named directory and calls fn for each.
|
||
|
|
// It does not recurse into subdirectories.
|
||
|
|
//
|
||
|
|
// If fn returns an error, iteration stops and WalkShallow returns that value.
|
||
|
|
//
|
||
|
|
// On Linux, WalkShallow does not allocate, so long as certain methods on the
|
||
|
|
// WalkFunc's DirEntry are not called which necessarily allocate.
|
||
|
|
func WalkShallow(dirName mem.RO, fn WalkFunc) error {
|
||
|
|
if f := osWalkShallow; f != nil {
|
||
|
|
return f(dirName, fn)
|
||
|
|
}
|
||
|
|
of, err := os.Open(dirName.StringCopy())
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer of.Close()
|
||
|
|
for {
|
||
|
|
fis, err := of.ReadDir(100)
|
||
|
|
for _, de := range fis {
|
||
|
|
if err := fn(mem.S(de.Name()), de); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
if err == io.EOF {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|