当前位置：首页 > news >正文

用 Go + HTML 实现 OpenHarmony 投屏（hdckit-go + WebSocket + Canvas 实战）

news 2025/9/10 9:05:32

本文带你用 Go + HTML/WebSocket 从零实现一个 OpenHarmony 设备投屏 Demo：Go 侧用 hdckit-go 连接设备并抓取屏幕帧（UiDriver），通过 WebSocket 二进制实时推送到浏览器，前端用 Canvas 渲染，并根据设备分辨率自适应显示。

源码：GitHub - airhandsome/hdckit-go-demo

hdckit-go 项目地址：GitHub - airhandsome/hdckit-go

一、功能概览

设备发现：通过 hdckit-go 列出 HDC 可连接设备

屏幕捕获：UiDriver 启动后回调获取 JPEG 帧

实时传输：WebSocket 二进制帧推送

前端渲染：Canvas 使用 createImageBitmap + requestAnimationFrame 平滑绘制

自适应缩放：根据设备原始分辨率动态调整 Canvas 样式尺寸

可扩展：可继续加按键/触控映射、录屏、帧率调节等

二、整体架构

Go 后端

hdc.NewClient → Target → CreateUiDriver → Start → StartCaptureScreen

WebSocket 端点 /ws 推送二进制 JPEG 帧；REST 接口 /api/devices 获取设备列表，/api/devices/{key}/start|stop 控制投屏

前端

WebSocket 连接服务器（binaryType = 'arraybuffer'）

收到 ArrayBuffer（二进制帧）→ createImageBitmap → requestAnimationFrame → Canvas 绘制

按容器大小与原始分辨率等比缩放显示

三、后端实现要点（Go）

依赖：

github.com/airhandsome/hdckit-go/hdc

github.com/gorilla/websocket

核心流程：

1) 初始化 HDC Client

cli := hdc.NewClient(hdc.Options{}) // 可配置 Bin/Host/Port
hdcClient = cli

2) 列出设备

targets, _ := hdcClient.ListTargets(ctx)
// 组装为 Device 列表返回给前端

3) UiDriver 投屏

drv := hdcClient.Target(deviceKey).CreateUiDriver()
_ = drv.Start(ctx)
_, _ = drv.StartCaptureScreen(ctx, func(frame []byte) {// 直接推送二进制帧（JPEG）broadcastBinaryFrame(frame)
}, 1)

4) WebSocket 广播二进制帧

func broadcastBinaryFrame(frame []byte) {for c := range clients {_ = c.WriteMessage(websocket.BinaryMessage, frame)}
}

5) 停止投屏

_ = drv.StopCaptureScreen(ctx)
drv.Stop() // 无参

常见坑：

drv.Stop() 不接受 context.Context，注意函数签名

初次使用 UiDriver 需确保 uitest agent 可推送到设备（uitestkit_sdk/uitest_agent_v1.1.0.so）

四、前端实现要点（HTML/JS）

1) WebSocket 设置为二进制

ws = new WebSocket(`ws://${location.host}/ws`);
ws.binaryType = 'arraybuffer';

2) 消息处理：区分 JSON（设备列表）与二进制帧

ws.onmessage = (event) => {const data = event.data;if (typeof data === 'string') {// 设备列表等 JSON 文本const msg = JSON.parse(data);handleMessage(msg);} else if (data instanceof ArrayBuffer) {renderFrame(data);}
};

3) Canvas 渲染（平滑且防闪）

async function renderFrame(buffer) {if (!isCapturing) return;const blob = new Blob([buffer], { type: 'image/jpeg' });const bitmap = await createImageBitmap(blob);// 源尺寸imgNaturalWidth = bitmap.width;imgNaturalHeight = bitmap.height;// 仅在尺寸变化时调整样式尺寸ensureCanvasDisplaySize();// rAF 合批绘制，避免频繁同步 drawscheduleDraw(bitmap); // 内部 drawImage(bitmap, 0, 0)
}

4) 自适应缩放