Format code

This commit is contained in:
Jonas Kvinge
2020-11-19 18:22:57 +01:00
parent 7e6c16b287
commit 3a27a3e868
6 changed files with 47 additions and 31 deletions

View File

@@ -107,10 +107,10 @@ void _MessageHandlerBase::WriteMessage(const QByteArray &data) {
else if (flush_local_socket_) { else if (flush_local_socket_) {
((qobject_cast<QLocalSocket*>(device_))->*(flush_local_socket_))(); ((qobject_cast<QLocalSocket*>(device_))->*(flush_local_socket_))();
} }
} }
void _MessageHandlerBase::DeviceClosed() { void _MessageHandlerBase::DeviceClosed() {
is_device_closed_ = true; is_device_closed_ = true;
AbortAll(); AbortAll();
} }

View File

@@ -146,6 +146,7 @@ void AbstractMessageHandler<MT>::SendReply(const MessageType &request, MessageTy
template<typename MT> template<typename MT>
bool AbstractMessageHandler<MT>::RawMessageArrived(const QByteArray &data) { bool AbstractMessageHandler<MT>::RawMessageArrived(const QByteArray &data) {
MessageType message; MessageType message;
if (!message.ParseFromArray(data.constData(), data.size())) { if (!message.ParseFromArray(data.constData(), data.size())) {
return false; return false;
@@ -161,14 +162,17 @@ bool AbstractMessageHandler<MT>::RawMessageArrived(const QByteArray &data) {
} }
return true; return true;
} }
template<typename MT> template<typename MT>
void AbstractMessageHandler<MT>::AbortAll() { void AbstractMessageHandler<MT>::AbortAll() {
for (ReplyType *reply : pending_replies_) { for (ReplyType *reply : pending_replies_) {
reply->Abort(); reply->Abort();
} }
pending_replies_.clear(); pending_replies_.clear();
} }
#endif // MESSAGEHANDLER_H #endif // MESSAGEHANDLER_H

View File

@@ -72,9 +72,7 @@ private:
template<typename MessageType> template<typename MessageType>
MessageReply<MessageType>::MessageReply(const MessageType& request_message, QObject *parent) MessageReply<MessageType>::MessageReply(const MessageType &request_message, QObject *parent) : _MessageReplyBase(parent) {
: _MessageReplyBase(parent)
{
request_message_.MergeFrom(request_message); request_message_.MergeFrom(request_message);
} }

View File

@@ -78,7 +78,7 @@ class WorkerPool : public _WorkerPoolBase {
void SetExecutableName(const QString &executable_name); void SetExecutableName(const QString &executable_name);
// Sets the number of worker process to use. Defaults to 1 <= (processors / 2) <= 2. // Sets the number of worker process to use. Defaults to 1 <= (processors / 2) <= 2.
void SetWorkerCount(int count); void SetWorkerCount(const int count);
// Sets the prefix to use for the local server (on unix this is a named pipe in /tmp). // Sets the prefix to use for the local server (on unix this is a named pipe in /tmp).
// Defaults to QApplication::applicationName(). // Defaults to QApplication::applicationName().
@@ -170,6 +170,7 @@ WorkerPool<HandlerType>::WorkerPool(QObject *parent)
template <typename HandlerType> template <typename HandlerType>
WorkerPool<HandlerType>::~WorkerPool() { WorkerPool<HandlerType>::~WorkerPool() {
for (const Worker &worker : workers_) { for (const Worker &worker : workers_) {
if (worker.local_socket_ && worker.process_) { if (worker.local_socket_ && worker.process_) {
disconnect(worker.process_, SIGNAL(errorOccurred(QProcess::ProcessError)), this, SLOT(ProcessError(QProcess::ProcessError))); disconnect(worker.process_, SIGNAL(errorOccurred(QProcess::ProcessError)), this, SLOT(ProcessError(QProcess::ProcessError)));
@@ -193,10 +194,11 @@ WorkerPool<HandlerType>::~WorkerPool() {
for (ReplyType *reply : message_queue_) { for (ReplyType *reply : message_queue_) {
reply->Abort(); reply->Abort();
} }
} }
template <typename HandlerType> template <typename HandlerType>
void WorkerPool<HandlerType>::SetWorkerCount(int count) { void WorkerPool<HandlerType>::SetWorkerCount(const int count) {
Q_ASSERT(workers_.isEmpty()); Q_ASSERT(workers_.isEmpty());
worker_count_ = count; worker_count_ = count;
} }
@@ -220,6 +222,7 @@ void WorkerPool<HandlerType>::Start() {
template <typename HandlerType> template <typename HandlerType>
void WorkerPool<HandlerType>::DoStart() { void WorkerPool<HandlerType>::DoStart() {
Q_ASSERT(workers_.isEmpty()); Q_ASSERT(workers_.isEmpty());
Q_ASSERT(!executable_name_.isEmpty()); Q_ASSERT(!executable_name_.isEmpty());
Q_ASSERT(QThread::currentThread() == thread()); Q_ASSERT(QThread::currentThread() == thread());
@@ -248,10 +251,12 @@ void WorkerPool<HandlerType>::DoStart() {
workers_ << worker; workers_ << worker;
} }
} }
template <typename HandlerType> template <typename HandlerType>
void WorkerPool<HandlerType>::StartOneWorker(Worker *worker) { void WorkerPool<HandlerType>::StartOneWorker(Worker *worker) {
Q_ASSERT(QThread::currentThread() == thread()); Q_ASSERT(QThread::currentThread() == thread());
DeleteQObjectPointerLater(&worker->local_server_); DeleteQObjectPointerLater(&worker->local_server_);
@@ -284,6 +289,7 @@ void WorkerPool<HandlerType>::StartOneWorker(Worker *worker) {
// Start the process // Start the process
worker->process_->setProcessChannelMode(QProcess::ForwardedChannels); worker->process_->setProcessChannelMode(QProcess::ForwardedChannels);
worker->process_->start(executable_path_, QStringList() << worker->local_server_->fullServerName()); worker->process_->start(executable_path_, QStringList() << worker->local_server_->fullServerName());
} }
template <typename HandlerType> template <typename HandlerType>
@@ -338,20 +344,24 @@ void WorkerPool<HandlerType>::ProcessError(QProcess::ProcessError error) {
StartOneWorker(worker); StartOneWorker(worker);
break; break;
} }
} }
template <typename HandlerType> template <typename HandlerType>
typename WorkerPool<HandlerType>::ReplyType* typename WorkerPool<HandlerType>::ReplyType*
WorkerPool<HandlerType>::NewReply(MessageType *message) { WorkerPool<HandlerType>::NewReply(MessageType *message) {
const int id = next_id_.fetchAndAddOrdered(1); const int id = next_id_.fetchAndAddOrdered(1);
message->set_id(id); message->set_id(id);
return new ReplyType(*message); return new ReplyType(*message);
} }
template <typename HandlerType> template <typename HandlerType>
typename WorkerPool<HandlerType>::ReplyType* typename WorkerPool<HandlerType>::ReplyType*
WorkerPool<HandlerType>::SendMessageWithReply(MessageType *message) { WorkerPool<HandlerType>::SendMessageWithReply(MessageType *message) {
ReplyType *reply = NewReply(message); ReplyType *reply = NewReply(message);
// Add the pending reply to the queue // Add the pending reply to the queue
@@ -364,10 +374,12 @@ WorkerPool<HandlerType>::SendMessageWithReply(MessageType *message) {
metaObject()->invokeMethod(this, "SendQueuedMessages", Qt::QueuedConnection); metaObject()->invokeMethod(this, "SendQueuedMessages", Qt::QueuedConnection);
return reply; return reply;
} }
template <typename HandlerType> template <typename HandlerType>
void WorkerPool<HandlerType>::SendQueuedMessages() { void WorkerPool<HandlerType>::SendQueuedMessages() {
QMutexLocker l(&message_queue_mutex_); QMutexLocker l(&message_queue_mutex_);
while (!message_queue_.isEmpty()) { while (!message_queue_.isEmpty()) {
@@ -384,10 +396,12 @@ void WorkerPool<HandlerType>::SendQueuedMessages() {
handler->SendRequest(reply); handler->SendRequest(reply);
} }
} }
template <typename HandlerType> template <typename HandlerType>
HandlerType *WorkerPool<HandlerType>::NextHandler() const { HandlerType *WorkerPool<HandlerType>::NextHandler() const {
for (int i = 0; i < workers_.count(); ++i) { for (int i = 0; i < workers_.count(); ++i) {
const int worker_index = (next_worker_ + i) % workers_.count(); const int worker_index = (next_worker_ + i) % workers_.count();
@@ -398,6 +412,7 @@ HandlerType *WorkerPool<HandlerType>::NextHandler() const {
} }
return nullptr; return nullptr;
} }
#endif // WORKERPOOL_H #endif // WORKERPOOL_H

View File

@@ -27,9 +27,7 @@
#include "tagreaderworker.h" #include "tagreaderworker.h"
TagReaderWorker::TagReaderWorker(QIODevice *socket, QObject *parent) TagReaderWorker::TagReaderWorker(QIODevice *socket, QObject *parent)
: AbstractMessageHandler<pb::tagreader::Message>(socket, parent) : AbstractMessageHandler<pb::tagreader::Message>(socket, parent) {}
{
}
void TagReaderWorker::MessageArrived(const pb::tagreader::Message &message) { void TagReaderWorker::MessageArrived(const pb::tagreader::Message &message) {
@@ -51,6 +49,7 @@ void TagReaderWorker::MessageArrived(const pb::tagreader::Message& message) {
} }
SendReply(message, &reply); SendReply(message, &reply);
} }