大家好，我是 V 哥，在鸿蒙HarmonyOS NEXT开发中，跨线程对象传递可以通过拷贝形式实现，确保两个线程的对象内容一致，但各自指向线程的隔离内存区间。以下是使用SharedArrayBuffer实现跨线程共享内存的完整案例代码，包括详细解释，整理的学习笔记，分享给大家。关注威哥不迷路，学习鸿蒙就很酷。
案例代码
1. 主线程代码
@Component
export struct LockUsage {taskNum: number = 10; // 任务数，实际并行线程数依设备而定baseDir: string = getContext().filesDir + '/TextDir'; // 文件写入的应用沙箱路径sabInLock: SharedArrayBuffer = new SharedArrayBuffer(4); // 在主线程，初始化子线程锁标志位，所使用的共享内存sabForLine: SharedArrayBuffer = new SharedArrayBuffer(4); // 在主线程，初始化子线程偏移位，所使用的共享内存@State result: string = "";build() {Row() {Column() {Button($r('app.string.not_use_lock')).width("80%").fontSize(30).fontWeight(FontWeight.Bold).margin({ top: 30 }).onClick(async () => {this.startWrite(false);})Button($r('app.string.use_lock')).width("80%").fontSize(30).fontWeight(FontWeight.Bold).margin({ top: 30 }).onClick(async () => {this.startWrite(true);})Text(this.result).width("80%").fontSize(30).fontWeight(FontWeight.Bold).fontColor(Color.Blue).margin({ top: 30 })}.width('100%')}.height('100%')}startWrite(useLock: boolean): void {this.result = getContext().resourceManager.getStringSync($r('app.string.write_file_start'));let whichLineToWrite: Int32Array = new Int32Array(this.sabForLine);Atomics.store(whichLineToWrite, 0, 0);let taskPoolGroup: taskpool.TaskGroup = new taskpool.TaskGroup();for (let i: number = 0; i < this.taskNum; i++) {taskPoolGroup.addTask(new taskpool.Task(createWriteTask, this.baseDir, i, this.sabInLock, this.sabForLine, useLock));}taskpool.execute(taskPoolGroup).then(() => {this.result = getContext().resourceManager.getStringSync($r('app.string.write_file_success'));}).catch(() => {this.result = getContext().resourceManager.getStringSync($r('app.string.write_file_failed'));})}
}
2. 子线程代码
@Concurrent
async function createWriteTask(baseDir: string, writeText: number, sabInLock: SharedArrayBuffer, sabForLine: SharedArrayBuffer, useLock: boolean): Promise<void> {class Option { offset: number = 0;length: number = 0;encoding: string = 'utf-8';constructor(offset: number, length: number) {this.offset = offset;this.length = length;}}let filePath: string | undefined = undefined;filePath = baseDir + useLock ? "/useLock.txt" : "/unusedLock.txt";if (!fs.accessSync(baseDir)) {fs.mkdirSync(baseDir);}let nrl: NonReentrantLock | undefined = undefined;if (useLock) {nrl = new NonReentrantLock(sabInLock);}let whichLineToWrite: Int32Array = new Int32Array(sabForLine);let str: string = writeText + '\n';for (let i: number = 0; i < 100; i++) {if (useLock && nrl !== undefined) {nrl.lock();}let file: fs.File = fs.openSync(filePath, fs.OpenMode.READ_WRITE | fs.OpenMode.CREATE);try {fs.writeSync(file.fd, str, new Option(whichLineToWrite[0], str.length));} catch (err) {logger.error(`errorCode : ${err.code},errMessage : ${err.message}`);}fs.closeSync(file);whichLineToWrite[0] += str.length;if (useLock && nrl !== undefined) {nrl.unlock();}}
}
详细解释
-
主线程初始化共享内存：
sabInLock和sabForLine是两个SharedArrayBuffer对象，分别用于子线程锁标志位和偏移位。它们在主线程中被初始化，并将被传递给子线程，实现跨线程共享内存。
-
子线程写入文件：
- 子线程根据主线程传入的
SharedArrayBuffer初始化锁和偏移量。 - 使用锁确保线程安全，避免多个线程同时写入文件时出现数据竞争。
- 通过
Atomics.store和Atomics.load操作共享内存，实现线程间的同步。
- 子线程根据主线程传入的
-
线程间参数传递：
- 使用
taskpool.Task创建子线程任务，并通过taskpool.execute执行。 - 子线程任务通过
createWriteTask函数实现，该函数接收主线程传递的参数，包括文件路径、写入内容、锁标志位和偏移位。
- 使用
-
线程安全写入：
- 在写入文件前，如果启用锁，则获取锁；写入完成后释放锁，确保线程安全。
- 通过修改共享内存中的偏移量，指定下次写入的位置，实现线程间的协作。
这个案例展示了如何在鸿蒙HarmonyOS NEXT开发中实现跨线程对象传递和共享内存，确保线程安全和数据一致性。通过使用 SharedArrayBuffer 和线程间参数传递，可以实现高效的并发编程。关注威哥爱编程，一起向鸿蒙出发。
