c48c1a728a
Summary: For one thing, this file is too big so it's good to split it up. Also, having this responder defined in the .cpp file makes it hard to test. Extracting it for testability. Reviewed By: passy Differential Revision: D14000079 fbshipit-source-id: 8da4e0e325f48c0ada8efc7cd6fffcb3440c6e26
78 lines
2.6 KiB
C++
78 lines
2.6 KiB
C++
/**
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the LICENSE
|
|
* file in the root directory of this source tree.
|
|
*/
|
|
#include "FlipperRSocketResponder.h"
|
|
#include <folly/json.h>
|
|
#include <rsocket/RSocket.h>
|
|
#include "FireAndForgetBasedFlipperResponder.h"
|
|
#include "FlipperConnectionManagerImpl.h"
|
|
#include "FlipperResponderImpl.h"
|
|
#include "Log.h"
|
|
|
|
using folly::dynamic;
|
|
|
|
namespace facebook {
|
|
namespace flipper {
|
|
|
|
rsocket::Payload toRSocketPayload(dynamic data);
|
|
|
|
void FlipperRSocketResponder::handleFireAndForget(
|
|
rsocket::Payload request,
|
|
rsocket::StreamId streamId) {
|
|
const auto payload = request.moveDataToString();
|
|
std::unique_ptr<FireAndForgetBasedFlipperResponder> responder;
|
|
auto message = folly::parseJson(payload);
|
|
if (message.find("id") != message.items().end()) {
|
|
auto id = message["id"].getInt();
|
|
responder =
|
|
std::make_unique<FireAndForgetBasedFlipperResponder>(websocket_, id);
|
|
}
|
|
|
|
websocket_->callbacks_->onMessageReceived(
|
|
folly::parseJson(payload), std::move(responder));
|
|
}
|
|
|
|
std::shared_ptr<yarpl::single::Single<rsocket::Payload>>
|
|
FlipperRSocketResponder::handleRequestResponse(
|
|
rsocket::Payload request,
|
|
rsocket::StreamId streamId) {
|
|
const auto requestString = request.moveDataToString();
|
|
|
|
auto dynamicSingle = yarpl::single::Single<folly::dynamic>::create(
|
|
[payload = std::move(requestString), this](auto observer) {
|
|
auto responder = std::make_unique<FlipperResponderImpl>(observer);
|
|
websocket_->callbacks_->onMessageReceived(
|
|
folly::parseJson(payload), std::move(responder));
|
|
});
|
|
|
|
auto rsocketSingle = yarpl::single::Single<rsocket::Payload>::create(
|
|
[payload = std::move(requestString), dynamicSingle, this](auto observer) {
|
|
observer->onSubscribe(yarpl::single::SingleSubscriptions::empty());
|
|
dynamicSingle->subscribe(
|
|
[observer, this](folly::dynamic d) {
|
|
websocket_->connectionEventBase_->runInEventBaseThread(
|
|
[observer, d]() {
|
|
try {
|
|
observer->onSuccess(toRSocketPayload(d));
|
|
|
|
} catch (std::exception& e) {
|
|
log(e.what());
|
|
observer->onError(e);
|
|
}
|
|
});
|
|
},
|
|
[observer, this](folly::exception_wrapper e) {
|
|
websocket_->connectionEventBase_->runInEventBaseThread(
|
|
[observer, e]() { observer->onError(e); });
|
|
});
|
|
});
|
|
|
|
return rsocketSingle;
|
|
}
|
|
|
|
} // namespace flipper
|
|
} // namespace facebook
|