Author: geechorama
Date: Wed Feb 10 00:56:09 2010
New Revision: 908301
URL: http://svn.apache.org/viewvc?rev=908301&view=rev
Log:
THRIFT-686. Adding TMemoryBuffer implementation to Cocoa
Added:
incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.h
incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.m
Added: incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.h
URL:
http://svn.apache.org/viewvc/incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.h?rev=908301&view=auto
==============================================================================
--- incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.h (added)
+++ incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.h Wed Feb 10
00:56:09 2010
@@ -0,0 +1,28 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#import <Foundation/Foundation.h>
+#import "TTransport.h"
+
+...@interface TMemoryBuffer : NSObject <TTransport> {
+ NSMutableData *mBuffer;
+ NSUInteger mOffset;
+}
+- (id)initWithData:(NSData *)data;
+...@end
Added: incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.m
URL:
http://svn.apache.org/viewvc/incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.m?rev=908301&view=auto
==============================================================================
--- incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.m (added)
+++ incubator/thrift/trunk/lib/cocoa/src/transport/TMemoryBuffer.m Wed Feb 10
00:56:09 2010
@@ -0,0 +1,59 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#import "TMemoryBuffer.h"
+#import "TTransportException.h"
+
+#define GARBAGE_BUFFER_SIZE 4096 // 4KiB
+
+...@implementation TMemoryBuffer
+- (id)initWithData:(NSData *)data {
+ if (self = [super init]) {
+ mBuffer = [data mutableCopy];
+ mOffset = 0;
+ }
+ return self;
+}
+
+- (int)readAll:(uint8_t *)buf offset:(int)off length:(int)len {
+ if ([mBuffer length] - mOffset < len) {
+ @throw [TTransportException exceptionWithReason:@"Not enough
bytes remain in buffer"];
+ }
+ [mBuffer getBytes:buf range:NSMakeRange(mOffset, len)];
+ mOffset += len;
+ if (mOffset >= GARBAGE_BUFFER_SIZE) {
+ [mBuffer replaceBytesInRange:NSMakeRange(0, mOffset)
withBytes:NULL length:0];
+ mOffset = 0;
+ }
+ return len;
+}
+
+- (void)write:(const uint8_t *)data offset:(unsigned int)offset
length:(unsigned int)length {
+ [mBuffer appendBytes:data+offset length:length];
+}
+
+- (void)flush {
+ // noop
+}
+
+- (void)dealloc {
+ [mBuffer release];
+ [super dealloc];
+}
+...@end
