//
|
// RHSocketVariableLengthEncoder.m
|
// RHSocketKitDemo
|
//
|
// Created by zhuruhong on 16/2/15.
|
// Copyright © 2016年 zhuruhong. All rights reserved.
|
//
|
|
#import "RHSocketVariableLengthEncoder.h"
|
#import "RHSocketException.h"
|
#import "RHSocketUtils.h"
|
|
@implementation RHSocketVariableLengthEncoder
|
|
- (instancetype)init
|
{
|
if (self = [super init]) {
|
_maxFrameSize = 65536;
|
_countOfLengthByte = 2;
|
_reverseOfLengthByte = NO;
|
}
|
return self;
|
}
|
|
- (void)encode:(id<RHUpstreamPacket>)upstreamPacket output:(id<RHSocketEncoderOutputProtocol>)output
|
{
|
NSData *data = [upstreamPacket dataWithPacket];
|
|
if (data.length >= _maxFrameSize - _countOfLengthByte) {
|
[RHSocketException raiseWithReason:@"[Encode] Too Long Frame ..."];
|
return;
|
}//
|
|
//可变长度编码，数据块的前两个字节为后续完整数据块的长度
|
NSUInteger dataLen = data.length;
|
NSMutableData *sendData = [[NSMutableData alloc] init];
|
|
//将数据长度转换为长度字节，写入到数据块中。这里根据head占的字节个数转换data长度，默认为2个字节
|
NSData *headData = [RHSocketUtils bytesFromValue:dataLen byteCount:_countOfLengthByte reverse:_reverseOfLengthByte];
|
[sendData appendData:headData];
|
if (data.length > 0) {
|
[sendData appendData:data];
|
}
|
NSTimeInterval timeout = [upstreamPacket timeout];
|
|
RHSocketLog(@"[Log]: timeout: %f, sendData: %@", timeout, sendData);
|
[output didEncode:sendData timeout:timeout];
|
}
|
|
@end
|
