add latest viewed url query.
All checks were successful
Deploy / PullDocker (push) Successful in 3s
Deploy / Build (push) Successful in 55s

This commit is contained in:
luocai 2024-08-06 09:48:25 +08:00
parent 4757f8e644
commit 0b6ccfd841
3 changed files with 77 additions and 10 deletions

View File

@ -194,6 +194,40 @@ Application::Application(const std::string &path)
session.reply(std::move(s)); session.reply(std::move(s));
}); });
m_router->insert("/api/v1/latest_viewed_urls", [this](HttpSession &session, const Request &request,
const boost::urls::matches &matches) {
using namespace boost::beast;
int size = 5;
std::error_code error;
if (!request.body().empty()) {
auto rootJson = boost::json::parse(request.body(), error);
if (error) {
LOG(info) << "<" << request.body() << "> parse json error: " << error.message();
} else {
auto &root = rootJson.as_object();
if (root.contains("size")) {
size = root.at("size").as_int64();
}
}
}
auto data = Amass::Singleton<Database>::instance()->latestViewedUrls(size);
boost::json::array reply;
for (auto &d : data) {
boost::json::object object;
object["url"] = d.url;
object["time"] = d.lastViewTime;
reply.push_back(std::move(object));
}
http::response<boost::beast::http::string_body> s{boost::beast::http::status::ok, request.version()};
s.set(http::field::server, BOOST_BEAST_VERSION_STRING);
s.set(http::field::content_type, "application/json;charset=UTF-8");
s.keep_alive(request.keep_alive());
s.body() = boost::json::serialize(reply);
s.prepare_payload();
session.reply(std::move(s));
});
m_ioContext = Amass::Singleton<IoContext>::instance<Amass::Construct>(getThreads()); m_ioContext = Amass::Singleton<IoContext>::instance<Amass::Construct>(getThreads());
m_timer = std::make_shared<boost::asio::system_timer>(*m_ioContext->ioContext()); m_timer = std::make_shared<boost::asio::system_timer>(*m_ioContext->ioContext());

View File

@ -186,13 +186,14 @@ VisitAnalysis Database::siteVisitAnalysisData() {
return ret; return ret;
} }
static std::vector<std::string> urlFilter = {
"/",
"/LoginPage",
"/我的笔记",
"/我的博客",
};
std::list<VisitAnalysis> Database::mostViewedUrls(int size) { std::list<VisitAnalysis> Database::mostViewedUrls(int size) {
std::vector<std::string> filter = {
"/",
"/LoginPage",
"/我的笔记",
"/我的博客",
};
const char *sql = "SELECT url, SUM(page_view_count) AS total_page_view_count " const char *sql = "SELECT url, SUM(page_view_count) AS total_page_view_count "
"FROM visit_analysis " "FROM visit_analysis "
"GROUP BY url " "GROUP BY url "
@ -206,13 +207,11 @@ std::list<VisitAnalysis> Database::mostViewedUrls(int size) {
return ret; return ret;
} }
sqlite3_bind_int(stmt, 1, size + filter.size()); sqlite3_bind_int(stmt, 1, size + urlFilter.size());
// Execute the SQL statement and fetch the results
while ((rc = sqlite3_step(stmt)) == SQLITE_ROW) { while ((rc = sqlite3_step(stmt)) == SQLITE_ROW) {
VisitAnalysis pv; VisitAnalysis pv;
pv.url = reinterpret_cast<const char *>(sqlite3_column_text(stmt, 0)); pv.url = reinterpret_cast<const char *>(sqlite3_column_text(stmt, 0));
if (std::find(filter.cbegin(), filter.cend(), pv.url) != filter.cend()) continue; if (std::find(urlFilter.cbegin(), urlFilter.cend(), pv.url) != urlFilter.cend()) continue;
pv.pageViewCount = sqlite3_column_int(stmt, 1); pv.pageViewCount = sqlite3_column_int(stmt, 1);
ret.push_back(pv); ret.push_back(pv);
} }
@ -225,6 +224,38 @@ std::list<VisitAnalysis> Database::mostViewedUrls(int size) {
return ret; return ret;
} }
std::list<VisitAnalysis> Database::latestViewedUrls(int size) {
std::list<VisitAnalysis> ret;
sqlite3_stmt *stmt;
const char *sql = R"(
SELECT url, MAX(last_view_time) as last_view_time
FROM visit_analysis
GROUP BY url
ORDER BY last_view_time DESC
LIMIT ?
)";
if (sqlite3_prepare_v2(m_sqlite3, sql, -1, &stmt, nullptr) != SQLITE_OK) {
LOG(error) << "Failed to prepare statement: " << sqlite3_errmsg(m_sqlite3);
return ret;
}
if (sqlite3_bind_int(stmt, 1, size + urlFilter.size()) != SQLITE_OK) {
LOG(error) << "Failed to bind parameter: " << sqlite3_errmsg(m_sqlite3);
sqlite3_finalize(stmt);
return ret;
}
while (sqlite3_step(stmt) == SQLITE_ROW) {
VisitAnalysis visit;
visit.url = reinterpret_cast<const char *>(sqlite3_column_text(stmt, 0));
if (std::find(urlFilter.cbegin(), urlFilter.cend(), visit.url) != urlFilter.cend()) continue;
visit.lastViewTime = sqlite3_column_int64(stmt, 1);
ret.push_back(visit);
}
return ret;
}
VisitAnalysis Database::visitAnalysisData(const std::string &url) { VisitAnalysis Database::visitAnalysisData(const std::string &url) {
VisitAnalysis ret; VisitAnalysis ret;
sqlite3_stmt *stmt; sqlite3_stmt *stmt;

View File

@ -11,6 +11,7 @@ public:
std::string url; std::string url;
int pageViewCount = 0; int pageViewCount = 0;
int uniqueVisitorCount = 0; int uniqueVisitorCount = 0;
int lastViewTime = 0;
}; };
struct sqlite3; struct sqlite3;
@ -33,6 +34,7 @@ public:
VisitAnalysis visitAnalysisData(const std::string &url); VisitAnalysis visitAnalysisData(const std::string &url);
VisitAnalysis siteVisitAnalysisData(); VisitAnalysis siteVisitAnalysisData();
std::list<VisitAnalysis> mostViewedUrls(int size); std::list<VisitAnalysis> mostViewedUrls(int size);
std::list<VisitAnalysis> latestViewedUrls(int size);
HomeBox::Items homeBoxItems(); HomeBox::Items homeBoxItems();
bool addHomeBoxItem(const std::string &name, const std::string &location, int cost); bool addHomeBoxItem(const std::string &name, const std::string &location, int cost);