如何在 Golang 中使用自訂分隔符號讀取檔案？

在 Golang 中使用自訂分隔符號讀取檔案：使用 bufio.NewReader 建立一個 Reader 物件。設定 reader.SplitFunc 函數來傳回自訂分隔符號的位置和行。循環讀取文件，逐行處理。

如何在Golang 中使用自訂分隔符號讀取檔案

在Golang 中讀取檔案時，預設的分隔符是換行符。但是，有時我們可能需要使用自訂分隔符號來讀取檔案。本文將介紹如何使用自訂分隔符號讀取檔案。

程式碼範例

package main

import (
    "bufio"
    "fmt"
    "log"
    "os"
)

func main() {
    // 打开文件以进行读取
    file, err := os.Open("data.csv")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    // 使用自定义分隔符创建 bufio.Reader 对象
    reader := bufio.NewReader(file)
    reader.SplitFunc = func(data []byte, atEOF bool) (int, []byte, error) {
        // 返回自定义分隔符的位置
        delimIndex := bytes.IndexByte(data, ';') // 假设分隔符是分号（;）

        if delimIndex == -1 {
            // 如果没有找到分隔符，则返回当前行的剩余部分
            return len(data), data, nil
        }

        // 返回分隔符的位置和分隔符之前的行
        return delimIndex + 1, data[:delimIndex], nil
    }

    // 逐行读取文件
    for {
        line, _, err := reader.ReadLine()
        if err == io.EOF {
            break
        } else if err != nil {
            log.Fatal(err)
        }

        // 处理每行
        fmt.Println(string(line))
    }
}

實戰案例

以下是使用分號分隔符號讀取CSV 檔案的程式碼範例：

package main

import (
    "bufio"
    "fmt"
    "log"
    "os"
)

func main() {
    // 打开 CSV 文件
    file, err := os.Open("customers.csv")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    // 使用分号分隔符创建 bufio.Reader 对象
    reader := bufio.NewReader(file)
    reader.SplitFunc = func(data []byte, atEOF bool) (int, []byte, error) {
        delimIndex := bytes.IndexByte(data, ';')
        if delimIndex == -1 {
            return len(data), data, nil
        }

        return delimIndex + 1, data[:delimIndex], nil
    }

    // 逐行读取 CSV 文件
    for {
        line, _, err := reader.ReadLine()
        if err == io.EOF {
            break
        } else if err != nil {
            log.Fatal(err)
        }

        // 根据分隔符拆分每一行
        parts := bytes.Split(line, []byte(";"))

        // 处理每一行
        fmt.Printf("ID: %s, Name: %s, Email: %s\n", parts[0], parts[1], parts[2])
    }
}

以上是如何在 Golang 中使用自訂分隔符號讀取檔案？的詳細內容。更多資訊請關注PHP中文網其他相關文章！