Как вывести список всех файлов с расширением txt и csv?

Question

[beduin01]

Вот пример того, как вывести список всех файлов с расширением csv и txt?
Вот код обхода всех файлов:

package main

import (
    "fmt"
    "os"
    "path/filepath"
)

func main() {
    var files []string

    root := "/some/folder/to/scan"
    err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
        files = append(files, path)
        return nil
    })
    if err != nil {
        panic(err)
    }
    for _, file := range files {
        fmt.Println(file)
    }
}

Answer 1

[Андрей Буров]

err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if !info.IsDir() {
			ext := filepath.Ext(path)
			if ext == ".txt" || ext == ".csv" {
				files = append(files, path)
			}
		}
		return nil
	})

Answer 2

[SergeySlukin]

Сам только изучаю го, хотел сделать так

if !info.IsDir() {
			fileLen := len(info.Name())
			if info.Name()[fileLen - 4:] == ".csv" || info.Name()[fileLen - 4:] == ".txt" {
				files = append(files, path)
			}
		}

Вариант предложенный Андрей Буров правильней.

Comments

[Андрей Буров]

а если файл lalacsv?

[SergeySlukin]

Да, действительно при таком варианте будет добавлять лишние файлы, код уже изменил.