/
githubmirror
/
etcd
Обзор
Документация
Войти
/
githubmirror
/
etcd
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
client/pkg/fileutil/read_dir.go
70 строк
2 KB
Piotr Tabor
Move pkg/* used by client to client/pkg.
06 апр 2021, 00:47
06 апр 2021, 00:47
0b81fdf
Код
Авторство
О чём код?
// Copyright 2018 The etcd Authors // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package fileutil import ( "os" "path/filepath" "sort" ) // ReadDirOp represents an read-directory operation. type ReadDirOp struct { ext string } // ReadDirOption configures archiver operations. type ReadDirOption func(*ReadDirOp) // WithExt filters file names by their extensions. // (e.g. WithExt(".wal") to list only WAL files) func WithExt(ext string) ReadDirOption { return func(op *ReadDirOp) { op.ext = ext } } func (op *ReadDirOp) applyOpts(opts []ReadDirOption) { for _, opt := range opts { opt(op) } } // ReadDir returns the filenames in the given directory in sorted order. func ReadDir(d string, opts ...ReadDirOption) ([]string, error) { op := &ReadDirOp{} op.applyOpts(opts) dir, err := os.Open(d) if err != nil { return nil, err } defer dir.Close() names, err := dir.Readdirnames(-1) if err != nil { return nil, err } sort.Strings(names) if op.ext != "" { tss := make([]string, 0) for _, v := range names { if filepath.Ext(v) == op.ext { tss = append(tss, v) } } names = tss } return names, nil }