提交 a4ab2e33 编写于 作者: Q qiang

Merge branch 'fix-websocket' into dev

import { import {
unpack, unpack,
publish, publish,
requireNativePlugin requireNativePlugin,
base64ToArrayBuffer,
arrayBufferToBase64
} from '../../bridge' } from '../../bridge'
let socketTaskId = 0
const socketTasks = {} const socketTasks = {}
const publishStateChange = (res) => { const publishStateChange = (res) => {
publish('onSocketTaskStateChange', res) publish('onSocketTaskStateChange', res)
} }
const createSocketTaskById = function (socketTaskId, { let socket
url, function getSocket () {
data, if (socket) {
header, return socket
method, }
protocols socket = requireNativePlugin('uni-webSocket')
} = {}) {
// fixed by hxy 需要测试是否支持 arraybuffer
const socket = requireNativePlugin('webSocket')
socket.WebSocket(url, Array.isArray(protocols) ? protocols.join(',') : protocols)
// socket.binaryType = 'arraybuffer'
socketTasks[socketTaskId] = socket
socket.onopen(function (e) { socket.onopen(function (e) {
publishStateChange({ publishStateChange({
socketTaskId, socketTaskId: e.id,
state: 'open' state: 'open'
}) })
}) })
socket.onmessage(function (e) { socket.onmessage(function (e) {
const data = e.data
publishStateChange({ publishStateChange({
socketTaskId, socketTaskId: e.id,
state: 'message', state: 'message',
data: e.data data: typeof data === 'object' ? base64ToArrayBuffer(data.base64) : data
}) })
}) })
socket.onerror(function (e) { socket.onerror(function (e) {
publishStateChange({ publishStateChange({
socketTaskId, socketTaskId: e.id,
state: 'error', state: 'error',
errMsg: e.message errMsg: e.data
}) })
}) })
socket.onclose(function (e) { socket.onclose(function (e) {
const socketTaskId = e.id
delete socketTasks[socketTaskId] delete socketTasks[socketTaskId]
publishStateChange({ publishStateChange({
socketTaskId, socketTaskId,
state: 'close' 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 { return {
socketTaskId, socketTaskId,
errMsg: 'createSocketTask:ok' errMsg: 'createSocketTask:ok'
...@@ -58,13 +71,14 @@ const createSocketTaskById = function (socketTaskId, { ...@@ -58,13 +71,14 @@ const createSocketTaskById = function (socketTaskId, {
} }
export function createSocketTask (args) { export function createSocketTask (args) {
return createSocketTaskById(++socketTaskId, args) return createSocketTaskById(String(Date.now()), args)
} }
export function operateSocketTask (args) { export function operateSocketTask (args) {
const { const {
operationType, operationType,
code, code,
reason,
data, data,
socketTaskId socketTaskId
} = unpack(args) } = unpack(args)
...@@ -77,13 +91,23 @@ export function operateSocketTask (args) { ...@@ -77,13 +91,23 @@ export function operateSocketTask (args) {
switch (operationType) { switch (operationType) {
case 'send': case 'send':
if (data) { if (data) {
socket.send(data) socket.send({
id: socketTaskId,
data: typeof data === 'object' ? {
'@type': 'binary',
base64: arrayBufferToBase64(data)
} : data
})
} }
return { return {
errMsg: 'operateSocketTask:ok' errMsg: 'operateSocketTask:ok'
} }
case 'close': case 'close':
socket.close(code) socket.close({
id: socketTaskId,
code,
reason
})
delete socketTasks[socketTaskId] delete socketTasks[socketTaskId]
return { return {
errMsg: 'operateSocketTask:ok' errMsg: 'operateSocketTask:ok'
...@@ -92,4 +116,4 @@ export function operateSocketTask (args) { ...@@ -92,4 +116,4 @@ export function operateSocketTask (args) {
return { return {
errMsg: 'operateSocketTask:fail' errMsg: 'operateSocketTask:fail'
} }
} }
import { import {
decode decode,
encode
} from 'base64-arraybuffer' } from 'base64-arraybuffer'
export { export {
...@@ -56,7 +57,7 @@ export function isTabBarPage (path = '') { ...@@ -56,7 +57,7 @@ export function isTabBarPage (path = '') {
return false return false
} }
return page.$page.meta.isTabBar return page.$page.meta.isTabBar
} }
const route = __uniRoutes.find(route => route.path.slice(1) === path) const route = __uniRoutes.find(route => route.path.slice(1) === path)
return route && route.meta.isTabBar return route && route.meta.isTabBar
} catch (e) { } catch (e) {
...@@ -70,3 +71,7 @@ export function isTabBarPage (path = '') { ...@@ -70,3 +71,7 @@ export function isTabBarPage (path = '') {
export function base64ToArrayBuffer (data) { export function base64ToArrayBuffer (data) {
return decode(data) return decode(data)
} }
export function arrayBufferToBase64 (data) {
return encode(data)
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册