157 lines
4.1 KiB
C++
157 lines
4.1 KiB
C++
#include "stompframe.h"
|
|
|
|
STOMPFrame::STOMPFrame(Command command, std::initializer_list<QPair<QString, QString>> headers, const QByteArray& body):
|
|
command_(command), body_(body) {
|
|
|
|
for (const auto& header: headers) {
|
|
headers_[header.first] = header.second;
|
|
}
|
|
}
|
|
|
|
qint64 STOMPFrame::send(QIODevice& device) {
|
|
|
|
QByteArray encoded;
|
|
QTextStream out {&encoded};
|
|
out << command_ << "\n";
|
|
|
|
for (const auto& key: headers_.keys()) {
|
|
out << key << ":" << headers_[key] << "\n";
|
|
}
|
|
|
|
if (!headers_.contains("content-length")) {
|
|
out << "content-length" << ":" << body_.size() << "\n";
|
|
}
|
|
|
|
out << "\n" << body_ << '\0';
|
|
out.flush();
|
|
return device.write(encoded);
|
|
}
|
|
|
|
STOMPFrame STOMPFrame::receive(QIODevice& device) {
|
|
STOMPFrame frame;
|
|
QTextStream in(&device);
|
|
|
|
QString commandStr;
|
|
if (!in.readLineInto(&commandStr)) {
|
|
return {};
|
|
}
|
|
if (commandStr == "STOMP") {
|
|
frame.command_ = STOMPFrame::STOMP;
|
|
} else if (commandStr == "SEND") {
|
|
frame.command_ = STOMPFrame::SEND;
|
|
} else if (commandStr == "SUBSCRIBE") {
|
|
frame.command_ = STOMPFrame::SUBSCRIBE;
|
|
} else if (commandStr == "UNSUBSCRIBE") {
|
|
frame.command_ = STOMPFrame::UNSUBSCRIBE;
|
|
} else if (commandStr == "BEGIN") {
|
|
frame.command_ = STOMPFrame::BEGIN;
|
|
} else if (commandStr == "COMMIT") {
|
|
frame.command_ = STOMPFrame::COMMIT;
|
|
} else if (commandStr == "ABORT") {
|
|
frame.command_ = STOMPFrame::ABORT;
|
|
} else if (commandStr == "ACK") {
|
|
frame.command_ = STOMPFrame::ACK;
|
|
} else if (commandStr == "NACK") {
|
|
frame.command_ = STOMPFrame::NACK;
|
|
} else if (commandStr == "DISCONNECT") {
|
|
frame.command_ = STOMPFrame::DISCONNECT;
|
|
} else if (commandStr == "CONNECTED") {
|
|
frame.command_ = STOMPFrame::CONNECTED;
|
|
} else if (commandStr == "MESSAGE") {
|
|
frame.command_ = STOMPFrame::MESSAGE;
|
|
} else if (commandStr == "RECEIPT") {
|
|
frame.command_ = STOMPFrame::RECEIPT;
|
|
} else if (commandStr == "ERROR") {
|
|
frame.command_ = STOMPFrame::ERROR;
|
|
} else {
|
|
return {};
|
|
}
|
|
|
|
QString headerLine;
|
|
while (in.readLineInto(&headerLine) && headerLine != "") {
|
|
auto components = headerLine.split(':');
|
|
if (components.count() != 2) {
|
|
return {};
|
|
}
|
|
frame.headers_.insert(components[0].trimmed(), components[1].trimmed());
|
|
}
|
|
|
|
if (frame.headers_.contains("content-length")) {
|
|
bool conversionOk;
|
|
qint64 length = frame.headers_["content-length"].toLongLong(&conversionOk);
|
|
if (!conversionOk) {
|
|
return {};
|
|
}
|
|
while (frame.body_.length() < length) {
|
|
frame.body_ += in.read(length - frame.body_.length()).toUtf8();
|
|
}
|
|
} else {
|
|
in >> frame.body_;
|
|
}
|
|
|
|
return frame;
|
|
}
|
|
|
|
QTextStream& operator <<(QTextStream& out, STOMPFrame::Command command) {
|
|
switch (command) {
|
|
case STOMPFrame::INVALID:
|
|
break;
|
|
|
|
case STOMPFrame::STOMP:
|
|
out << "STOMP";
|
|
break;
|
|
|
|
case STOMPFrame::SEND:
|
|
out << "SEND";
|
|
break;
|
|
|
|
case STOMPFrame::SUBSCRIBE:
|
|
out << "SUBSCRIBE";
|
|
break;
|
|
|
|
case STOMPFrame::UNSUBSCRIBE:
|
|
out << "UNSUBSCRIBE";
|
|
break;
|
|
|
|
case STOMPFrame::BEGIN:
|
|
out << "BEGIN";
|
|
break;
|
|
|
|
case STOMPFrame::COMMIT:
|
|
out << "COMMIT";
|
|
break;
|
|
|
|
case STOMPFrame::ABORT:
|
|
out << "ABORT";
|
|
break;
|
|
|
|
case STOMPFrame::ACK:
|
|
out << "ACK";
|
|
break;
|
|
|
|
case STOMPFrame::NACK:
|
|
out << "NACK";
|
|
break;
|
|
|
|
case STOMPFrame::DISCONNECT:
|
|
out << "DISCONNECT";
|
|
break;
|
|
|
|
case STOMPFrame::CONNECTED:
|
|
out << "CONNECTED";
|
|
break;
|
|
|
|
case STOMPFrame::MESSAGE:
|
|
out << "MESSAGE";
|
|
break;
|
|
|
|
case STOMPFrame::RECEIPT:
|
|
out << "RECEIPT";
|
|
break;
|
|
case STOMPFrame::ERROR:
|
|
out << "ERROR";
|
|
break;
|
|
}
|
|
return out;
|
|
}
|