如何處理Go語言中的並發檔案的加密和解密問題？
引言：
隨著網路的發展和資訊傳輸的普及，檔案加密和解密已經成為保護資料安全的重要手段。而且，隨著電腦處理能力和儲存容量的提升，同時處理多個文件的需求也日益增加。在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.")
}上述程式碼實作了一個並發處理文件加密和解密的範例。在範例中，我們首先定義了一個用於等待所有並發任務完成的sync.WaitGroup對象，然後指定了要加密和解密的文件列表，並以並發方式進行加密和解密操作。
在encryptFile和decryptFile函數中，我們先開啟輸入檔案和輸出文件，然後建立一個AES區塊，並使用16位元組IV對其進行初始化。接下來，我們將產生的串流與輸入檔案或輸出檔案進行關聯，然後使用io.Copy函數複製數據，完成檔案加密和解密的操作。
最後，在main函數中，我們使用filepath.Join函數來建立檔案路徑，並啟動並發的加密和解密任務。等待所有任務完成後，列印成功的訊息。
以上是如何處理Go語言中的並發檔案的加密和解密問題？的詳細內容。更多資訊請關注PHP中文網其他相關文章！
