79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
112 lines
2.8 KiB
Plaintext
112 lines
2.8 KiB
Plaintext
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#if FB_SONARKIT_ENABLED
|
|
|
|
#import <vector>
|
|
|
|
#import <FlipperKit/FlipperConnection.h>
|
|
#import "SKBufferingPlugin+CPPInitialization.h"
|
|
#import "SKBufferingPlugin.h"
|
|
#import "SKDispatchQueue.h"
|
|
|
|
static const NSUInteger bufferSize = 500;
|
|
|
|
@interface SKBufferingPlugin ()
|
|
|
|
@property(assign, nonatomic) std::vector<CachedEvent> ringBuffer;
|
|
@property(assign, nonatomic) std::shared_ptr<facebook::flipper::DispatchQueue>
|
|
connectionAccessQueue;
|
|
@property(strong, nonatomic) id<FlipperConnection> connection;
|
|
|
|
@end
|
|
|
|
@implementation SKBufferingPlugin
|
|
// {
|
|
// std::vector<CachedEvent> _ringBuffer;
|
|
// std::shared_ptr<facebook::flipper::DispatchQueue> _connectionAccessQueue;
|
|
//
|
|
// id<FlipperConnection> _connection;
|
|
// }
|
|
|
|
- (instancetype)initWithQueue:(dispatch_queue_t)queue {
|
|
if (self = [super init]) {
|
|
_ringBuffer.reserve(bufferSize);
|
|
_connectionAccessQueue =
|
|
std::make_shared<facebook::flipper::GCDQueue>(queue);
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (NSString*)identifier {
|
|
// Note: This must match with the javascript pulgin identifier!!
|
|
return @"Network";
|
|
}
|
|
|
|
- (void)didConnect:(id<FlipperConnection>)connection {
|
|
_connectionAccessQueue->async(^{
|
|
self->_connection = connection;
|
|
[self sendBufferedEvents];
|
|
});
|
|
}
|
|
|
|
- (void)didDisconnect {
|
|
_connectionAccessQueue->async(^{
|
|
self->_connection = nil;
|
|
});
|
|
}
|
|
|
|
- (BOOL)runInBackground {
|
|
return YES;
|
|
}
|
|
|
|
- (void)send:(NSString*)method
|
|
sonarObject:(NSDictionary<NSString*, id>*)sonarObject {
|
|
_connectionAccessQueue->async(^{
|
|
if (self->_connection) {
|
|
[self->_connection send:method withParams:sonarObject];
|
|
} else {
|
|
if (self->_ringBuffer.size() == bufferSize) {
|
|
return;
|
|
}
|
|
self->_ringBuffer.push_back(
|
|
{.method = method, .sonarObject = sonarObject});
|
|
}
|
|
});
|
|
}
|
|
|
|
- (void)sendBufferedEvents {
|
|
NSAssert(_connection, @"connection object cannot be nil");
|
|
for (const auto& event : _ringBuffer) {
|
|
[_connection send:event.method withParams:event.sonarObject];
|
|
}
|
|
_ringBuffer.clear();
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation SKBufferingPlugin (CPPInitialization)
|
|
|
|
- (instancetype)initWithVectorEventSize:(NSUInteger)size
|
|
connectionAccessQueue:
|
|
(std::shared_ptr<facebook::flipper::DispatchQueue>)
|
|
connectionAccessQueue {
|
|
if (self = [super init]) {
|
|
_ringBuffer.reserve(size);
|
|
_connectionAccessQueue = connectionAccessQueue;
|
|
}
|
|
return self;
|
|
}
|
|
- (instancetype)initWithDispatchQueue:
|
|
(std::shared_ptr<facebook::flipper::DispatchQueue>)queue {
|
|
return [self initWithVectorEventSize:bufferSize connectionAccessQueue:queue];
|
|
}
|
|
|
|
@end
|
|
|
|
#endif
|