How to list all files with txt and csv extension?

Question

go

Here is an example of how to list all files with csv and txt extension?
Here is the code to bypass all files:

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

I'm just learning Go myself, I wanted to do this

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

The option proposed by Andrey Burov is more correct.