func SaveOS() { var data []model.HistoryMessages gorm.DB.Find(&data) file, err := os.Create("data.csv") if err != nil { panic(err) } defer file.Close() writer := csv.NewWriter(file) defer writer.Flush() writer.Write([]string{"SendId", "ReceiveId", "Object", "Media", "Content", "SendTime"}) for _, message := range data { record := []string{ message.SendId, message.ReceiveId, message.Object, message.Media, message.Content, message.SendTime.Format("2006-01-02 15:04:05"), } writer.Write(record) } }帮我把这代码生成的文件使用aes加密，写出完整代码

好的，请参考下面的代码，使用AES加密文件：

import (
    "crypto/aes"
    "crypto/cipher"
    "crypto/rand"
    "encoding/csv"
    "os"
    "io"
    "log"
)

func SaveOS() {
    var data []model.HistoryMessages
    gorm.DB.Find(&data)

    // 创建文件并打开
    file, err := os.Create("data.csv")
    if err != nil {
        panic(err)
    }
    defer file.Close()

    // 创建CSV writer
    writer := csv.NewWriter(file)
    defer writer.Flush()

    // 写入表头
    writer.Write([]string{"SendId", "ReceiveId", "Object", "Media", "Content", "SendTime"})

    // 写入数据
    for _, message := range data {
        record := []string{
            message.SendId,
            message.ReceiveId,
            message.Object,
            message.Media,
            message.Content,
            message.SendTime.Format("2006-01-02 15:04:05"),
        }
        writer.Write(record)
    }

    // 打开CSV文件并读取数据
    csvfile, err := os.Open("data.csv")
    if err != nil {
        log.Fatalln("Unable to read input file " + "data.csv")
    }
    defer csvfile.Close()
    reader := csv.NewReader(csvfile)
    records, err := reader.ReadAll()
    if err != nil {
        log.Fatalln("Unable to parse file " + "data.csv")
    }

    // 生成随机密钥和向量
    key := make([]byte, 32)
    if _, err := rand.Read(key); err != nil {
        panic(err.Error())
    }
    iv := make([]byte, aes.BlockSize)
    if _, err := rand.Read(iv); err != nil {
        panic(err.Error())
    }

    // 创建加密器
    block, err := aes.NewCipher(key)
    if err != nil {
        panic(err.Error())
    }
    ciphertext := make([]byte, aes.BlockSize+len(records))
    stream := cipher.NewCTR(block, iv)
    stream.XORKeyStream(ciphertext[aes.BlockSize:], []byte(records))

    // 写入加密后的数据和密钥、向量到文件
    outfile, err := os.Create("data.enc")
    if err != nil {
        log.Fatalln("Unable to create output file " + "data.enc")
    }
    defer outfile.Close()

    // 写入密钥和向量
    outfile.Write(iv)
    outfile.Write(key)

    // 写入加密后的数据
    _, err = outfile.Write(ciphertext)
    if err != nil {
        log.Fatalln("Unable to write encrypted data to output file")
    }
}

注意：这个代码中使用了AES-CTR模式进行加密，同时生成了随机的密钥和向量。加密后的数据保存在" data.enc"文件中，同时也保存了密钥和向量。在解密时，需要使用相同的密钥和向量才能成功解密。