75 lines
2.1 KiB
C
75 lines
2.1 KiB
C
|
#ifndef _PB_RPC_CHANNEL_H
|
||
|
|
#define _PB_RPC_CHANNEL_H
|
||
|
|
|
||
|
|
#include <QTcpServer>
|
||
|
|
#include <QTcpSocket>
|
||
|
|
|
||
|
|
#include <google/protobuf/message.h>
|
||
|
|
#include <google/protobuf/descriptor.h>
|
||
|
|
#include <google/protobuf/service.h>
|
||
|
|
|
||
|
|
#include "pbrpccommon.h"
|
||
|
|
#include "pbrpccontroller.h"
|
||
|
|
|
||
|
|
class PbRpcChannel : public QObject, public ::google::protobuf::RpcChannel
|
||
|
|
{
|
||
|
|
Q_OBJECT
|
||
|
|
|
||
|
|
// If isPending is TRUE, then controller, done, response
|
||
|
|
// and pendingMethodId correspond to the last method called by
|
||
|
|
// the service stub
|
||
|
|
bool isPending;
|
||
|
|
int pendingMethodId;
|
||
|
|
|
||
|
|
// controller, done, response are set to the corresponding values
|
||
|
|
// passed by the stub to CallMethod(). They are reset to NULL when
|
||
|
|
// we get a response back from the server in on_mpSocket_readyRead()
|
||
|
|
// after calling done->Run().
|
||
|
|
//
|
||
|
|
// TODO(?): change controller, done and response to references
|
||
|
|
// instead of pointers?
|
||
|
|
::google::protobuf::RpcController *controller;
|
||
|
|
::google::protobuf::Closure *done;
|
||
|
|
::google::protobuf::Message *response;
|
||
|
|
|
||
|
|
QHostAddress mServerAddress;
|
||
|
|
quint16 mServerPort;
|
||
|
|
QTcpSocket *mpSocket;
|
||
|
|
|
||
|
|
public:
|
||
|
|
PbRpcChannel(QHostAddress ip, quint16 port);
|
||
|
|
~PbRpcChannel();
|
||
|
|
|
||
|
|
void establish();
|
||
|
|
void establish(QHostAddress ip, quint16 port);
|
||
|
|
void tearDown();
|
||
|
|
|
||
|
|
const QHostAddress& serverAddress() const { return mServerAddress; }
|
||
|
|
quint16 serverPort() const { return mServerPort; }
|
||
|
|
|
||
|
|
QAbstractSocket::SocketState state() const
|
||
|
|
{ return mpSocket->state(); }
|
||
|
|
|
||
|
|
void CallMethod(const ::google::protobuf::MethodDescriptor *method,
|
||
|
|
::google::protobuf::RpcController *controller,
|
||
|
|
const ::google::protobuf::Message *req,
|
||
|
|
::google::protobuf::Message *response,
|
||
|
|
::google::protobuf::Closure* done);
|
||
|
|
|
||
|
|
signals:
|
||
|
|
void connected();
|
||
|
|
void disconnected();
|
||
|
|
void error(QAbstractSocket::SocketError socketError);
|
||
|
|
void stateChanged(QAbstractSocket::SocketState socketState);
|
||
|
|
|
||
|
|
private slots:
|
||
|
|
void on_mpSocket_connected();
|
||
|
|
void on_mpSocket_disconnected();
|
||
|
|
void on_mpSocket_stateChanged(QAbstractSocket::SocketState socketState);
|
||
|
|
void on_mpSocket_error(QAbstractSocket::SocketError socketError);
|
||
|
|
|
||
|
|
void on_mpSocket_readyRead();
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|