首页 > 编程开发 > Objective-C编程 >
-
使用内存映射文件MMF实现大数据量导出时的内存优化(Linux篇)
前言
今天这篇博客是接我的上一篇博客 https://www.cnblogs.com/y-yp/p/12191258.html,继续介绍一下MMF在Linux上的用法
ps:本来本地调试完case,想放到服务器上跑跑看,结果竟然报"PlatformNotSupportedException",然后仔细一查,竟然发现MMF在Windows和Linux上的用法不一样。。。"mapName"参数仅作为Window平台的一个特性,在Linux平台上只能传"null",于是就有了今天这篇博客
实现
既然“mapName”不能使用,经过测试选定使用了FileStream的这个重载
具体细节就不在介绍了,有疑问的话可以参考我的上一篇博客 https://www.cnblogs.com/y-yp/p/12191258.html,这里直接给实现
先定义“行数据信息记录“,这个用来读取数据的时候用,一行数据只生成一条记录,所以在大数据量的情况下也不会占用很多内存
1
2
3
4
5
6
7
8
9
10
11
12
|
public class RowInfo { /// <summary> /// 行数据体积(单位字节) /// </summary> public long Capacity { get ; set ; } /// <summary> /// 行单元格个数 /// </summary> public int CellQuantity { get ; set ; } } |
然后开始将数据写入MMF文件,并获取到”行数据信息记录“
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
//准备数据 var data = new List< string []>(); for ( var i = 0; i < 100; i++) { var rowData = new string [100]; for ( var j = 0; j < 100; j++) { rowData[j] = $ "{i}-{j}" ; } data.Add(rowData); } //统计mmf文件体积,包含单元格数据的体积Encoding.UTF8.GetBytes(x).Length和默认单元格数据长度int类型占4字节 var mmfCapacity = data.Sum(x => x.Sum(x => Encoding.UTF8.GetBytes(x).Length + 4)); var path = Environment.CurrentDirectory + "\\" + "test.txt" ; using var writerFs = new FileStream(path, FileMode.Create, FileAccess.ReadWrite); using var writerMMF = MemoryMappedFile.CreateFromFile(writerFs, null , mmfCapacity, MemoryMappedFileAccess.ReadWrite, HandleInheritability.Inheritable, true ); //记录行数据信息 var rowInfos = new List<RowInfo>(); var totalWriterOffset = 0; foreach ( var rowData in data) { var rowBuffers = rowData.Select(x => Encoding.UTF8.GetBytes(x)).ToList(); //计算行数据总体积 var capacity = rowBuffers.Sum(x => x.Length + 4); //根据当前偏移和需要读取的长度创建accessor using var accessor = writerMMF.CreateViewAccessor(totalWriterOffset, capacity); //统计同行内单元格偏移 var offset = 0L; foreach ( var cellBuffer in rowBuffers) { if (cellBuffer.Length > 0) { var dataSize = cellBuffer.Length; accessor.Write(offset, dataSize); accessor.WriteArray(offset + 4, cellBuffer, 0, dataSize); offset += 4 + dataSize; } else { accessor.Write(offset, 0); offset += 4; } } //记录行数据信息 var rowInfo = new RowInfo() { Capacity = capacity, CellQuantity = rowBuffers.Count() }; rowInfos.Add(rowInfo); //总位移向前走一行数据的长度 totalWriterOffset += capacity; } return rowInfos; |
通过”行数据信息记录“还原数据,这里可以将读取出来的数据写入自己的excel或者是csv文件,不再赘述
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
var result = new List< string []>(); var path = Environment.CurrentDirectory + "\\" + "test.txt" ; //从行数据信息记录统计mmf文件总体积 var mmfCapacity = rowInfos.Sum(x => x.Capacity); var totalReaderOffset = 0; using var readerFs = new FileStream(path, FileMode.Open, FileAccess.ReadWrite); using var readerMMF = MemoryMappedFile.CreateFromFile(readerFs, null , mmfCapacity, MemoryMappedFileAccess.ReadWrite, HandleInheritability.Inheritable, true ); foreach ( var rowInfo in rowInfos) { var rowData = new string [rowInfo.CellQuantity]; using var accessor = readerMMF.CreateViewAccessor(totalReaderOffset, rowInfo.Capacity); var position = 0; for ( int cellIndex = 0; cellIndex < rowInfo.CellQuantity; cellIndex++) { var cellSize = accessor.ReadInt32(position); var buffer = new byte [cellSize]; accessor.ReadArray(position + 4, buffer, 0, cellSize); rowData[cellIndex] = Encoding.UTF8.GetString(buffer); position += 4 + cellSize; } result.Add(rowData); } if (File.Exists(path)) { File.Delete(path); } |
考虑使用内存映射文件的话,可以先本地测试一下性能,如果是SSD的话性能还是很不错的,综合跑下来跟直接写入内存速度相差不会超过一两倍(内存使用率较高的话会严重降低性能,甚至会OOM),而且这其中还有很多的优化空间
今天的文章只是提了个思路,细节还有很多要考虑,有疑问的话欢迎提问交流~~
文章出处:https://www.cnblogs.com/y-yp/p/12372963.html