使用VC++ ATL将结构体数组写入文件并读取的完整实现方案

源码资源网

使用VC++ ATL将结构体数组写入文件并读取的完整实现方案
该实现包含完整的错误处理，先写入数组长度再写入数据，确保读取时可以正确分配内存。使用时注意内存管理，读取后需要手动释放数组内存。
StructArrayIO.h

1. 
   
2. #pragma once
   
3. #include <atlfile.h>
   
4. #include <atlstr.h>
   
5. 
   
6. // 示例结构体
   
7. struct AccountInfo {
   
8.     int id;
   
9.     char username[32];
   
10.     double balance;
    
11.     bool isActive;
    
12. };
    

复制代码

StructArrayIO.cpp

1. #include "StructArrayIO.h"
   
2. 
   
3. // 写入结构体数组到文件
   
4. bool WriteStructArray(const CString& filePath, const AccountInfo* data, int count) {
   
5.     CAtlFile file;
   
6.     if (FAILED(file.Create(filePath, GENERIC_WRITE, FILE_SHARE_READ, CREATE_ALWAYS))) {
   
7.         return false;
   
8.     }
   
9. 
   
10.     // 先写入数组长度
    
11.     DWORD bytesWritten = 0;
    
12.     if (FAILED(file.Write(&count, sizeof(int), bytesWritten))) {
    
13.         file.Close();
    
14.         return false;
    
15.     }
    
16. 
    
17.     // 写入数组数据
    
18.     HRESULT hr = file.Write(data, sizeof(AccountInfo) * count, bytesWritten);
    
19.     file.Close();
    
20.    
    
21.     return SUCCEEDED(hr);
    
22. }
    
23. 
    
24. // 从文件读取结构体数组
    
25. bool ReadStructArray(const CString& filePath, AccountInfo*& outData, int& outCount) {
    
26.     CAtlFile file;
    
27.     if (FAILED(file.Create(filePath, GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING))) {
    
28.         return false;
    
29.     }
    
30. 
    
31.     // 读取数组长度
    
32.     DWORD bytesRead = 0;
    
33.     if (FAILED(file.Read(&outCount, sizeof(int), bytesRead))) {
    
34.         file.Close();
    
35.         return false;
    
36.     }
    
37. 
    
38.     // 分配内存
    
39.     outData = new AccountInfo[outCount];
    
40.    
    
41.     // 读取数组数据
    
42.     HRESULT hr = file.Read(outData, sizeof(AccountInfo) * outCount, bytesRead);
    
43.     file.Close();
    
44.    
    
45.     if (FAILED(hr)) {
    
46.         delete[] outData;
    
47.         return false;
    
48.     }
    
49.     return true;
    
50. }
    
51. 
    
52. // 使用示例
    
53. void ExampleUsage() {
    
54.     AccountInfo accounts[3] = {
    
55.         {1001, "user1", 1000.0, true},
    
56.         {1002, "user2", 2000.0, false},
    
57.         {1003, "user3", 3000.0, true}
    
58.     };
    
59.     CString filePath = L"C:\\temp\\accounts.dat";
    
60.    
    
61.     // 写入示例
    
62.     if (WriteStructArray(filePath, accounts, 3)) {
    
63.         // 读取示例
    
64.         AccountInfo* loadedAccounts = nullptr;
    
65.         int count = 0;
    
66.         if (ReadStructArray(filePath, loadedAccounts, count)) {
    
67.             // 使用读取的数据...
    
68.             delete[] loadedAccounts; // 记得释放内存
    
69.         }
    
70.     }
    
71. }
    

复制代码