From 8780dda26a819b4b93caba75a893fb082849406e Mon Sep 17 00:00:00 2001 From: Nick White Date: Tue, 27 Jul 2021 14:22:03 +0100 Subject: internal/pipeline: add tests for DetectQueueType --- internal/pipeline/put.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'internal/pipeline/put.go') diff --git a/internal/pipeline/put.go b/internal/pipeline/put.go index 647b7a9..7b292b9 100644 --- a/internal/pipeline/put.go +++ b/internal/pipeline/put.go @@ -23,6 +23,8 @@ func (w NullWriter) Write(p []byte) (n int, err error) { type fileWalk chan string +// Walk sends the path of all files to the channel, with the exception of +// any file or directory which starts with "." func (f fileWalk) Walk(path string, info os.FileInfo, err error) error { if err != nil { return err @@ -38,6 +40,8 @@ func (f fileWalk) Walk(path string, info os.FileInfo, err error) error { return nil } +// CheckImages checks that all files in a directory are images +// that can be decoded (skipping dotfiles) func CheckImages(dir string) error { checker := make(fileWalk) go func() { @@ -59,8 +63,9 @@ func CheckImages(dir string) error { return nil } +// DetectQueueType detects which queue to use based on the preponderance +// of files of a particular extension in a directory func DetectQueueType(dir string, conn Queuer) string { - // Auto detect type of queue to send to based on file extension pngdirs, _ := filepath.Glob(dir + "/*.png") jpgdirs, _ := filepath.Glob(dir + "/*.jpg") pngcount := len(pngdirs) -- cgit v1.2.1-24-ge1ad