안정적인 API 키 관리 솔루션 구축: Vault를 사용한 Golang

package main

import (
    "crypto/rand"
    "encoding/base64"
    "fmt"
)

func generateAPIKey() string {
    key := make([]byte, 32)
    _, err := rand.Read(key)
    if err != nil {
        panic(err)
    }
    return base64.URLEncoding.EncodeToString(key)
}

func main() {
    apiKey := generateAPIKey()
    fmt.Println("API Key:", apiKey)
}

package main

import (
    "fmt"
    "log"

    "github.com/hashicorp/vault/api"
)

func storeAPIKey(apiKey string) {
    vaultConfig := &api.Config{
        Address: "<vault_address>", // Vault服务器的地址
    }

    client, err := api.NewClient(vaultConfig)
    if err != nil {
        log.Fatal(err)
    }

    vaultToken := "<vault_token>" // 需要替换为你在Vault中创建的Token
    client.SetToken(vaultToken)

    secretValue := map[string]interface{}{
        "value": apiKey,
    }

    secretPath := "secret/api_key" // 存储API密钥的路径
    _, err = client.Logical().Write(secretPath, secretValue)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("API Key stored successfully")
}

func main() {
    apiKey := generateAPIKey()
    storeAPIKey(apiKey)
}

package main

import (
    "fmt"
    "log"

    "github.com/hashicorp/vault/api"
)

func getAPIKey() (string, error) {
    vaultConfig := &api.Config{
        Address: "<vault_address>", // Vault服务器的地址
    }

    client, err := api.NewClient(vaultConfig)
    if err != nil {
        log.Fatal(err)
    }

    vaultToken := "<vault_token>" // 需要替换为你在Vault中创建的Token
    client.SetToken(vaultToken)

    secretPath := "secret/api_key" // 存储API密钥的路径
    secret, err := client.Logical().Read(secretPath)
    if err != nil {
        return "", err
    }

    apiKey, ok := secret.Data["value"].(string)
    if !ok {
        return "", fmt.Errorf("Invalid API Key")
    }

    return apiKey, nil
}

func main() {
    apiKey, err := getAPIKey()
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("API Key:", apiKey)
}