133 lines
3.9 KiB
C++
133 lines
3.9 KiB
C++
#include "restapiserver.h"
|
|
|
|
#include <QJsonArray>
|
|
#include <QJsonDocument>
|
|
#include <QJsonObject>
|
|
#include <QJsonValue>
|
|
|
|
#include <database/author_s-odb.hxx> // Должен быть здесь
|
|
#include <database/author_s.h>
|
|
#include <database/book_s-odb.hxx> // Должен быть здесь
|
|
#include <database/book_s.h>
|
|
|
|
#include <odb/core.hxx>
|
|
#include <odb/database.hxx>
|
|
#include <odb/query.hxx>
|
|
|
|
RestApiServer::RestApiServer(odb::database& db, QObject* parent) :
|
|
QTcpServer(parent), m_db(db) {}
|
|
|
|
void RestApiServer::start(quint16 port)
|
|
{
|
|
if (!this->listen(QHostAddress::Any, port))
|
|
{
|
|
qWarning() << "Ошибка запуска сервера!";
|
|
}
|
|
else
|
|
{
|
|
qDebug() << "REST API сервер запущен на порту:" << port;
|
|
}
|
|
}
|
|
|
|
void RestApiServer::incomingConnection(qintptr socketDescriptor)
|
|
{
|
|
QTcpSocket* socket = new QTcpSocket(this);
|
|
socket->setSocketDescriptor(socketDescriptor);
|
|
|
|
connect(socket, &QTcpSocket::readyRead, this, &RestApiServer::handleRequest);
|
|
connect(socket, &QTcpSocket::disconnected, socket, &QTcpSocket::deleteLater);
|
|
}
|
|
|
|
void RestApiServer::handleRequest()
|
|
{
|
|
QTcpSocket* socket = qobject_cast<QTcpSocket*>(sender());
|
|
if (!socket)
|
|
return;
|
|
|
|
QString request = QString::fromUtf8(socket->readAll());
|
|
QByteArray response = processRequest(request);
|
|
|
|
socket->write(response);
|
|
socket->flush();
|
|
socket->disconnectFromHost();
|
|
}
|
|
|
|
QByteArray RestApiServer::processRequest(const QString& request)
|
|
{
|
|
if (request.startsWith("GET /books/author/"))
|
|
{
|
|
QString author = request.section(' ', 1, 1).section('/', 3, 3).replace("%20", " ");
|
|
|
|
qWarning() << "THIS";
|
|
// Books books;
|
|
// QList<BookRecord> results = books.getBooksByAuthor(author);
|
|
|
|
QByteArray jsonResponse = "{ \"books\": [";
|
|
// for (const BookRecord& book : results)
|
|
// {
|
|
// jsonResponse += QString("{ \"id\": %1, \"title\": \"%2\", \"author\": \"%3\", \"year\": %4 },")
|
|
// .arg(book.id)
|
|
// .arg(book.title)
|
|
// .arg(book.author)
|
|
// .arg(book.year)
|
|
// .toUtf8();
|
|
// }
|
|
// if (results.size() > 0)
|
|
// jsonResponse.chop(1); // Убираем последнюю запятую
|
|
jsonResponse += "] }";
|
|
|
|
return "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\n\r\n" + jsonResponse;
|
|
}
|
|
|
|
if (request.startsWith("GET /books"))
|
|
{
|
|
odb::transaction t(m_db.begin());
|
|
odb::result<Book_S> res(m_db.query<Book_S>());
|
|
|
|
QJsonArray jArray;
|
|
|
|
for (auto it = res.begin(); it != res.end(); ++it)
|
|
{
|
|
const auto& book = *it;
|
|
|
|
QJsonObject j;
|
|
j["id"] = QString::number(book.id());
|
|
j["title"] = book.name();
|
|
j["author"] = book.author()->full_name();
|
|
j["year"] = book.year();
|
|
jArray.push_back(j);
|
|
}
|
|
auto jDoc = QJsonDocument(jArray);
|
|
|
|
t.commit();
|
|
|
|
return "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\n\r\n" + jDoc.toJson(QJsonDocument::Indented);
|
|
}
|
|
|
|
if (request.startsWith("GET /author"))
|
|
{
|
|
odb::transaction t(m_db.begin());
|
|
odb::result<Author_S> res(m_db.query<Author_S>());
|
|
|
|
QJsonArray jArray;
|
|
|
|
for (auto it = res.begin(); it != res.end(); ++it)
|
|
{
|
|
const auto& author = *it;
|
|
|
|
QJsonObject j;
|
|
j["id"] = QString::number(author.id());
|
|
j["first"] = author.first();
|
|
j["last"] = author.last();
|
|
j["age"] = author.age();
|
|
jArray.push_back(j);
|
|
}
|
|
auto jDoc = QJsonDocument(jArray);
|
|
|
|
t.commit();
|
|
return "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\n\r\n" + jDoc.toJson(QJsonDocument::Indented);
|
|
}
|
|
|
|
return "HTTP/1.1 404 Not Found\r\nContent-Type: text/plain\r\n\r\nNot Found";
|
|
}
|