Golang 函数无法正确读取文件名

问题描述 投票:0回答:1

因此,我有一个包含多个 .csv 文件的存储库,它们包含数据库的表架构。我编写了一个 Golang 代码,它从存储库获取文件名列表,然后打开这些文件,读取内容并创建 MySQL CREATE 查询。

我面临的问题是,对于某些 .csv 文件,Golang 代码最终会读取错误的标题,这会导致后期出现问题。例如,有一些名为 config_hr.csv、config_oe.csv、contribution_analysis.csv 的文件被读取为 onfig_hr.csv、onfig_oe.csv、ontribution_analysi.csv。如果我将名称大写,这个问题似乎可以得到解决,但是在我们项目的后期阶段还会出现许多其他问题。

这是某种编码问题吗?我已经检查了 Windows、Mac 和 Linux 上的代码,Golang 版本是最新的 v1.21,任何帮助或见解将不胜感激!

读取 CSV 文件名称的 Golang 代码片段

entries, err := FileEntry.Readdir(0)
    if err != nil {
        log.Fatal(err)
    }

    // Now, open all the files one by one, and extract the content of the files.
    // Then modify the resultant string to be of MySQL compatibility.
    for _, e := range entries {
        // Mimicking the SQL Query of Table Creation query.
        Query_String := ("CREATE TABLE IF NOT EXISTS " + strings.ToLower(strings.Trim(strings.Replace(e.Name(), " ", "_", -1), ".csv")) + " (\n")
        fmt.Println("Opening -- " + file_folder + "/" + e.Name())
        file, err := os.Open(file_folder + "/" + e.Name())
        if err != nil {
            log.Fatal(err)
        }
        defer file.Close()
        // Reading the CSV file from path.
        reader := csv.NewReader(file)
        records, err := reader.ReadAll()
        if err != nil {
            log.Fatal(err)
        }
      
csv go file-handling
1个回答
0
投票

这不是编码问题,而是

strings.Trim
问题。
strings.Trim
将使用切割集而不是字符串从开头和结尾进行修剪。所以你的“.csv”本质上是说“如果这个集合中的内容位于字符串的开头或结尾,则将其剪切。如果你的文件被命名为
svconfig_oe.csv
,你也会遇到同样的问题 - 你最终会得到
onfig_oe

您可以使用

strings.TrimRight
(仍然是切割集,但仅限于字符串末尾),或者因为您知道文件以
.csv
结尾,只需再次使用
strings.Replace
和“.csv” - 在本例中,
 strings.Replace
将查找确切的字符串而不是剪切集。

© www.soinside.com 2019 - 2024. All rights reserved.