使用Golang配置AWS凭证的最佳实践

AWS SDK Go v2 的配置加载机制主要用于运行时凭证管理，而不是直接修改凭证文件。你的代码成功运行是因为它在内存中使用了静态凭证，但并没有将这些凭证写入到 ~/.aws/credentials 文件中。

要修改凭证文件，你需要直接操作文件系统。以下是一个示例函数，它使用 Go 的标准库来更新 AWS 凭证文件：

package main

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

func updateAWSCredentials(profileName, accessKeyID, secretAccessKey string) error {
    // 获取用户主目录
    homeDir, err := os.UserHomeDir()
    if err != nil {
        return fmt.Errorf("无法获取用户主目录: %w", err)
    }

    // 构建凭证文件路径
    credFilePath := filepath.Join(homeDir, ".aws", "credentials")
    
    // 读取现有文件内容
    content, err := os.ReadFile(credFilePath)
    if err != nil && !os.IsNotExist(err) {
        return fmt.Errorf("读取凭证文件失败: %w", err)
    }

    // 准备新的凭证条目
    newEntry := fmt.Sprintf("[%s]\naws_access_key_id = %s\naws_secret_access_key = %s\n\n",
        profileName, accessKeyID, secretAccessKey)

    var updatedContent string
    lines := strings.Split(string(content), "\n")
    
    inTargetProfile := false
    skipLines := 0
    
    // 遍历文件行，替换目标配置块
    for i := 0; i < len(lines); i++ {
        if skipLines > 0 {
            skipLines--
            continue
        }
        
        line := lines[i]
        if strings.HasPrefix(line, "[") && strings.HasSuffix(line, "]") {
            if inTargetProfile {
                inTargetProfile = false
            }
            if line == "["+profileName+"]" {
                inTargetProfile = true
                skipLines = 2 // 跳过接下来的两行（access key 和 secret key）
                updatedContent += newEntry
                continue
            }
        }
        
        if !inTargetProfile {
            updatedContent += line + "\n"
        }
    }
    
    // 如果配置块不存在，则追加到文件末尾
    if !strings.Contains(updatedContent, "["+profileName+"]") {
        updatedContent += newEntry
    }

    // 确保目录存在
    dir := filepath.Dir(credFilePath)
    if err := os.MkdirAll(dir, 0700); err != nil {
        return fmt.Errorf("创建目录失败: %w", err)
    }

    // 写入更新后的内容
    if err := os.WriteFile(credFilePath, []byte(strings.TrimSpace(updatedContent)), 0600); err != nil {
        return fmt.Errorf("写入凭证文件失败: %w", err)
    }

    return nil
}

func main() {
    // 示例：更新 default 配置的凭证
    err := updateAWSCredentials("default", "AKIAIOSFODNN7EXAMPLE", "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY")
    if err != nil {
        fmt.Printf("更新凭证失败: %v\n", err)
        return
    }
    fmt.Println("凭证更新成功")
}

这个函数会：

1. 定位 ~/.aws/credentials 文件
2. 读取现有内容
3. 更新或添加指定配置块的凭证信息
4. 保持文件格式的一致性

如果你需要更复杂的凭证管理，可以考虑使用 ini 解析库来处理配置文件：

import "gopkg.in/ini.v1"

func updateAWSCredentialsWithINI(profileName, accessKeyID, secretAccessKey string) error {
    homeDir, err := os.UserHomeDir()
    if err != nil {
        return err
    }
    
    credFilePath := filepath.Join(homeDir, ".aws", "credentials")
    
    cfg, err := ini.Load(credFilePath)
    if err != nil && !os.IsNotExist(err) {
        return err
    }
    
    if cfg == nil {
        cfg = ini.Empty()
    }
    
    section := cfg.Section(profileName)
    section.Key("aws_access_key_id").SetValue(accessKeyID)
    section.Key("aws_secret_access_key").SetValue(secretAccessKey)
    
    dir := filepath.Dir(credFilePath)
    if err := os.MkdirAll(dir, 0700); err != nil {
        return err
    }
    
    return cfg.SaveTo(credFilePath)
}

使用 ini 库可以更可靠地处理配置文件的各种格式情况。