Go语言中字符串与配置解析的常见陷阱与最佳实践

buffer := bytes.NewBuffer(make([]byte, 2048)) // 创建一个长度和容量都为2048的切片作为初始内容
buffer.Write(part) // 写入 part 会覆盖掉 buffer 的前缀，而不是追加

buffer := bytes.NewBuffer(make([]byte, 0, 2048)) // 创建一个长度为0，容量为2048的切片
buffer.Write(part) // 写入 part 会追加到缓冲区末尾

var buffer bytes.Buffer // 推荐：创建一个空的 Buffer
// 或
// buffer := new(bytes.Buffer) // 同样创建一个空的 Buffer
buffer.Write(part) // 追加写入

package main

import (
    "bufio"
    "fmt"
    "io" // 导入 io 包以使用 io.EOF
    "os"
    "strings"
)

// Config 类型用于存储解析后的配置
type Config map[string]string

// ReadConfig 从指定文件读取配置，如果文件名为空或文件不存在，则返回默认配置
func ReadConfig(filename string) (Config, error) {
    // 设置默认配置
    config := Config{
        "browsercommand": "%u",
        "port":           "7896",
        "password":       "hallo",
        "ip":             "127.0.0.1",
    }

    // 如果未指定配置文件，直接返回默认配置
    if len(filename) == 0 {
        return config, nil
    }

    // 打开文件
    file, err := os.Open(filename)
    if err != nil {
        // 文件不存在或无法打开时，返回错误，但可以根据需求决定是否返回默认配置
        // 这里选择返回错误，让调用者决定如何处理
        return nil, fmt.Errorf("无法打开配置文件 %s: %w", filename, err)
    }
    defer file.Close() // 确保文件在函数返回前关闭

    rdr := bufio.NewReader(file)
    for {
        line, err := rdr.ReadString('\n') // 逐行读取，直到遇到换行符
        line = strings.TrimSpace(line)    // 移除行首尾空格

        // 忽略空行和注释行（以;或#开头）
        if len(line) == 0 || strings.HasPrefix(line, ";") || strings.HasPrefix(line, "#") {
            if err == io.EOF {
                break // 文件末尾，退出循环
            }
            if err != nil {
                return nil, fmt.Errorf("读取配置文件时发生错误: %w", err)
            }
            continue
        }

        // 查找等号
        if eq := strings.Index(line, "="); eq >= 0 {
            key := strings.TrimSpace(line[:eq]) // 提取键，并移除空格
            value := ""
            if len(line) > eq {
                value = strings.TrimSpace(line[eq+1:]) // 提取值，并移除空格
            }

            if len(key) > 0 { // 确保键不为空
                config[key] = value
            }
        }

        if err == io.EOF {
            break // 文件末尾，退出循环
        }
        if err != nil {
            return nil, fmt.Errorf("读取配置文件时发生错误: %w", err)
        }
    }
    return config, nil
}

func main() {
    // 示例使用
    config, err := ReadConfig(`netconfig.txt`) // 假设配置文件名为 netconfig.txt
    if err != nil {
        fmt.Println("Error reading config:", err)
        // 可以在这里选择退出或使用默认配置
        // os.Exit(1)
        // 如果 ReadConfig 返回 nil, err，这里需要检查 config 是否为 nil
        // 否则，如果 ReadConfig 内部已返回默认配置，则可继续
    }

    fmt.Println("Parsed config:", config)

    // 从配置中获取特定值
    ip := config["ip"]
    pass := config["password"]
    port := config["port"]
    fmt.Println("Extracted values: IP =", ip, ", Port =", port, ", Password =", pass)

    // 示例：使用一个不存在的配置文件名
    fmt.Println("\n--- Testing with non-existent file ---")
    _, err = ReadConfig("non_existent_config.txt")
    if err != nil {
        fmt.Println("Expected error for non-existent file:", err)
    }

    // 示例：使用空文件名，应返回默认配置
    fmt.Println("\n--- Testing with empty filename ---")
    defaultConfig, err := ReadConfig("")
    if err != nil {
        fmt.Println("Unexpected error for empty filename:", err)
    } else {
        fmt.Println("Default config (empty filename):", defaultConfig)
    }
}

[network_settings]
ip = 217.110.104.156
port = 80
password = hello
; This is a comment line
# Another comment style
url = test.de
file =

// 假设 Sendtext 函数定义如下
// func Sendtext(ip string, port string, text string) (err int) { ... }

func main() {
    // ... 获取配置
    config, err := ReadConfig(`netconfig.txt`)
    if err != nil {
        fmt.Println("Error reading config:", err)
        os.Exit(1) // 错误时退出
    }

    ip := config["ip"]
    port := config["port"]
    pass := config["password"] // 密码也可以从这里获取

    // 假设 GetURL() 和 browserbridge_config.ReadPropertiesFile() 已经适应新的配置读取方式
    // 或者直接使用 config 中的值
    url := GetURL() // 假设 GetURL() 仍然存在并获取URL
    message := url + "\n" + pass + "\n"

    fmt.Printf("sending this url to %s:%s\n", ip, port)
    fmt.Println("sending...")

    // 调用 Sendtext 函数
    e := Sendtext(ip, port, message)
    if e != 0 {
        fmt.Println("ERROR")
        os.Exit(e)
    }
    fmt.Println("DONE")
}