Android WebRTC利用DataChannel收發文本數據和傳輸文件

一、概述

　　利用WebRTC的DataChannel發送文本數據以及傳輸文件

二、實例代碼演示

　　1.初始化DataChannel以及注冊DataChannel的回調函數

 /**
     * 創建DataChannel
     *
     * @param socketId 用戶id
     * @return 數據通道
     */
    public DataChannel createDataChannel(String socketId, PeerConnection peerConnection) {
        if (dataChannelInit == null) {
            /**
             DataChannel.Init 可配參數說明：
             ordered：是否保證順序傳輸；
             maxRetransmitTimeMs：重傳允許的最長時間；
             maxRetransmits：重傳允許的最大次數；
             */
            dataChannelInit = new DataChannel.Init();
            dataChannelInit.ordered = true; // 消息的傳遞是否有序 true代表有序
            dataChannelInit.negotiated = true; // 協商方式
            dataChannelInit.id = 0; // 通道ID
        }
        if (dataChannel == null) {
            dataChannel = peerConnection.createDataChannel("dataChannel", dataChannelInit);
            //注冊DataChannel的回調函數
            dataChannel.registerObserver(new DataChannel.Observer() {
                boolean isHeader = true;
                String suffix = null;
                int fileLength = 0;
                long currentLength = 0;
                boolean isFinish = false;
                List<byte[]> queue = new ArrayList<>();

                //
                @Override
                public void onBufferedAmountChange(long l) {

                }

                //狀態發生改變
                @Override
                public void onStateChange() {

                }

                /**
                 * 接收二進制消息時需要定義一個簡單的header用於存放文件信息
                 * 1.filename 文件名
                 * 2.suffix  后綴名
                 * 3.totalLength 文件總大小
                 * @param buffer
                 */
                //接收消息
                @Override
                public void onMessage(DataChannel.Buffer buffer) {
                    try {
                        ByteBuffer data = buffer.data;
                        byte[] bytes = new byte[data.capacity()];
                        Log.e(TAG, "initDataChannel----->onMessage--->" + bytes.length);
                        data.get(bytes);
                        if (dataChannelListener != null) {
                            if (buffer.binary) {//是二進制數據
                                if (isHeader) {
                                    isHeader = false;//為false時就不是第一次，只有第一次需要檢測文件后綴
                                    //檢測文件后綴
                                    byte[] headerPayload = new byte[200];
                                    System.arraycopy(bytes, 0, headerPayload, 0, 200);
                                    String filePath = ByteUtil.getInstance().getStringFromByteArray(headerPayload);
                                    byte[] lengthPayload = new byte[200];
                                    System.arraycopy(bytes, 200, lengthPayload, 0, 20);
                                    String length = ByteUtil.getInstance().getStringFromByteArray(lengthPayload);
                                    Log.e(TAG, "initDataChannel----->onMessage--->filePath---->" + filePath);
                                    Log.e(TAG, "initDataChannel----->onMessage--->length---->" + length);
                                    suffix = FileUtils.getInstance().getFileSuffix(filePath);
                                    fileLength = Integer.parseInt(length) - 220;
                                    Log.e(TAG, "initDataChannel----->onMessage--->suffix---->" + suffix);
                                }
                                if (!isHeader) {
                                    currentLength += bytes.length;
                                    if ((currentLength - 220) >= fileLength) {
                                        isFinish = true;
                                    }
                                    queue.add(bytes);
                                    float progress = (currentLength / (float) fileLength) * 100;
                                    dataChannelListener.onReceiveFileProgress(progress);
                                }
                                if (isFinish) {
                                    String realPath = null;
//                                    for (int i = 0; i < queue.size(); i++) {
//                                        isFinish = false;
//                                        if (i == queue.size() - 1) {
//                                            isFinish = true;
//                                        }
//
////
//                                    }

                                    queue.remove(0);
                                    realPath = FileUtils.getInstance().writeBytesToFile(context, suffix, queue);
                                    if (realPath != null) {
                                        Log.e(TAG, "initDataChannel----->onMessage--->realPath----> 執行了多少次");
                                        dataChannelListener.onReceiveBinaryMessage(socketId, realPath);
                                    }

                                }

                            } else {//不是二進制數據
                                //此處接收的是非二進制數據
                                String msg = new String(bytes);
                                dataChannelListener.onReceiveMessage(socketId, msg);
                            }

                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    }

                }
            });
        }
        return dataChannel;
    }

　　2.接收及發送文本消息　

/**
     * 使用DataChannel發送普通消息
     *
     * @param message
     */
    public void sendMsg(String message) {
        if (dataChannel != null) {
            if (message != null) {
                byte[] msg = message.getBytes();
                DataChannel.Buffer buffer = new DataChannel.Buffer(ByteBuffer.wrap(msg), false);
                dataChannel.send(buffer);
            }
        }


    }　

//接收普通文本消息
client.setDataChannelListener(new DataChannelListenerImpl() {
            @Override
            public void onReceiveMessage(String userId, String message) {
                mainHandler.post(() -> {
                    sb.append(userId).append(":").append(message).append("\n");
                    tvContent.setText(sb.toString());
                });

            }


        });

 

　　3.接收及發送文件

 /**
     * 使用DataChannel發送二進制消息
     *
     * @param bytes
     */
    public void sendBinary(byte[] bytes) {
        if (dataChannel != null) {
            DataChannel.Buffer buffer = new DataChannel.Buffer(ByteBuffer.wrap(bytes), true);
            dataChannel.send(buffer);
        }
    }　

client.setDataChannelListener(new DataChannelListenerImpl() {

           
//接收文件消息成功回調以及，顯示文件接收的進度
 @Override
            public void onReceiveBinaryMessage(String userId, String realPath) {
                handler.post(() -> {
                    tvReceiveFileName.setText(realPath);
                });
            }

            @Override
            public void onReceiveFileProgress(float progress) {
                handler.post(() -> {
                    tvReceiveFileName.setText(progress+"%");
                });

            }


        });