Comment gérer le cryptage et le déchiffrement des fichiers simultanés en langage Go ?

package main

import (
    "crypto/aes"
    "crypto/cipher"
    "fmt"
    "io"
    "os"
    "path/filepath"
    "sync"
)

// 加密文件
func encryptFile(inPath, outPath string, key []byte) error {
    inFile, err := os.Open(inPath)
    if err != nil {
        return err
    }
    defer inFile.Close()

    outFile, err := os.OpenFile(outPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
    if err != nil {
        return err
    }
    defer outFile.Close()

    block, err := aes.NewCipher(key)
    if err != nil {
        return err
    }

    iv := make([]byte, aes.BlockSize)
    stream := cipher.NewCTR(block, iv)

    writer := &cipher.StreamWriter{S: stream, W: outFile}
    if _, err := io.Copy(writer, inFile); err != nil {
        return err
    }

    return nil
}

// 解密文件
func decryptFile(inPath, outPath string, key []byte) error {
    inFile, err := os.Open(inPath)
    if err != nil {
        return err
    }
    defer inFile.Close()

    outFile, err := os.OpenFile(outPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
    if err != nil {
        return err
    }
    defer outFile.Close()

    block, err := aes.NewCipher(key)
    if err != nil {
        return err
    }

    iv := make([]byte, aes.BlockSize)
    stream := cipher.NewCTR(block, iv)

    reader := &cipher.StreamReader{S: stream, R: inFile}
    if _, err := io.Copy(outFile, reader); err != nil {
        return err
    }

    return nil
}

func main() {
    var wg sync.WaitGroup
    key := []byte("YOUR_KEY")

    // 要加密的文件列表
    files := []string{"file1.txt", "file2.txt", "file3.txt"}

    wg.Add(len(files))
    for _, file := range files {
        go func(f string) {
            defer wg.Done()
            inPath := filepath.Join("input", f)
            outPath := filepath.Join("output", "encrypted_"+f)

            if err := encryptFile(inPath, outPath, key); err != nil {
                fmt.Printf("Error encrypting file %s: %s
", inPath, err.Error())
            }
        }(file)
    }

    wg.Wait()

    // 要解密的文件列表
    files = []string{"encrypted_file1.txt", "encrypted_file2.txt", "encrypted_file3.txt"}

    wg.Add(len(files))
    for _, file := range files {
        go func(f string) {
            defer wg.Done()
            inPath := filepath.Join("output", f)
            outPath := filepath.Join("output", "decrypted_"+f[len("encrypted_"):])

            if err := decryptFile(inPath, outPath, key); err != nil {
                fmt.Printf("Error decrypting file %s: %s
", inPath, err.Error())
            }
        }(file)
    }

    wg.Wait()

    fmt.Println("All files encrypted and decrypted successfully.")
}