Skip to content

Commit

Permalink
fix: App 端 websocket 支持 arraybuffer,支持创建多个连接
Browse files Browse the repository at this point in the history
  • Loading branch information
zhetengbiji committed Sep 12, 2019
1 parent 1b29984 commit f464c78
Show file tree
Hide file tree
Showing 2 changed files with 55 additions and 26 deletions.
72 changes: 48 additions & 24 deletions src/platforms/app-plus/service/api/network/socket.js
Original file line number Diff line number Diff line change
@@ -1,70 +1,84 @@
import {
unpack,
publish,
requireNativePlugin
requireNativePlugin,
base64ToArrayBuffer,
arrayBufferToBase64
} from '../../bridge'

let socketTaskId = 0
const socketTasks = {}

const publishStateChange = (res) => {
publish('onSocketTaskStateChange', res)
}

const createSocketTaskById = function (socketTaskId, {
url,
data,
header,
method,
protocols
} = {}) {
// fixed by hxy 需要测试是否支持 arraybuffer
const socket = requireNativePlugin('webSocket')
socket.WebSocket(url, Array.isArray(protocols) ? protocols.join(',') : protocols)
// socket.binaryType = 'arraybuffer'
socketTasks[socketTaskId] = socket

let socket
function getSocket () {
if (socket) {
return socket
}
socket = requireNativePlugin('uni-webSocket')
socket.onopen(function (e) {
publishStateChange({
socketTaskId,
socketTaskId: e.id,
state: 'open'
})
})
socket.onmessage(function (e) {
const data = e.data
publishStateChange({
socketTaskId,
socketTaskId: e.id,
state: 'message',
data: e.data
data: typeof data === 'object' ? base64ToArrayBuffer(data.base64) : data
})
})
socket.onerror(function (e) {
publishStateChange({
socketTaskId,
socketTaskId: e.id,
state: 'error',
errMsg: e.message
errMsg: e.data
})
})
socket.onclose(function (e) {
const socketTaskId = e.id
delete socketTasks[socketTaskId]
publishStateChange({
socketTaskId,
state: 'close'
})
})
return socket
}

const createSocketTaskById = function (socketTaskId, {
url,
data,
header,
method,
protocols
} = {}) {
const socket = getSocket()
socket.WebSocket({
id: socketTaskId,
url,
protocol: Array.isArray(protocols) ? protocols.join(',') : protocols
})
socketTasks[socketTaskId] = socket
return {
socketTaskId,
errMsg: 'createSocketTask:ok'
}
}

export function createSocketTask (args) {
return createSocketTaskById(++socketTaskId, args)
return createSocketTaskById(String(Date.now()), args)
}

export function operateSocketTask (args) {
const {
operationType,
code,
reason,
data,
socketTaskId
} = unpack(args)
Expand All @@ -77,13 +91,23 @@ export function operateSocketTask (args) {
switch (operationType) {
case 'send':
if (data) {
socket.send(data)
socket.send({
id: socketTaskId,
data: typeof data === 'object' ? {
'@type': 'binary',
base64: arrayBufferToBase64(data)
} : data
})
}
return {
errMsg: 'operateSocketTask:ok'
}
case 'close':
socket.close(code)
socket.close({
id: socketTaskId,
code,
reason
})
delete socketTasks[socketTaskId]
return {
errMsg: 'operateSocketTask:ok'
Expand All @@ -92,4 +116,4 @@ export function operateSocketTask (args) {
return {
errMsg: 'operateSocketTask:fail'
}
}
}
9 changes: 7 additions & 2 deletions src/platforms/app-plus/service/bridge.js
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import {
decode
decode,
encode
} from 'base64-arraybuffer'

export {
Expand Down Expand Up @@ -56,7 +57,7 @@ export function isTabBarPage (path = '') {
return false
}
return page.$page.meta.isTabBar
}
}
const route = __uniRoutes.find(route => route.path.slice(1) === path)
return route && route.meta.isTabBar
} catch (e) {
Expand All @@ -70,3 +71,7 @@ export function isTabBarPage (path = '') {
export function base64ToArrayBuffer (data) {
return decode(data)
}

export function arrayBufferToBase64 (data) {
return encode(data)
}

0 comments on commit f464c78

Please sign in to comment.