avbot_log_search.cpp

#include <string>
#include <vector>
#include <memory>
#include <utility>
#include <functional>

#include <boost/locale.hpp>
#include <boost/locale/encoding_utf.hpp>

#include <boost/property_tree/ptree.hpp>
namespace pt = boost::property_tree;
#include <boost/asio.hpp>

#include <boost/timer/timer.hpp>

#include <soci-sqlite3.h>
#include <boost-optional.h>
#include <boost-tuple.h>
#include <boost-gregorian-date.h>
#include <soci.h>

#include <avhttp/detail/escape_string.hpp>

#include "boost/logger.hpp"

void avlog_do_search(boost::asio::io_service & io_service, boost::logger& logger,
	std::string c, std::string q, std::string date,
	std::function<void (boost::system::error_code, pt::wptree)> handler,
	soci::session & db)
{
	pt::wptree outjson;
	std::string q_escaped;
	// 根据 channel_name , query string , date 像数据库查找
	logger.dbg() << " c = " << c << " q =  " << q << " date= " << date ;

	std::vector<std::string>	r_date(1000);
	std::vector<std::string>	r_channel(1000);
	std::vector<std::string>	r_nick(1000);
	std::vector<std::string>	r_message(1000);
	std::vector<std::string>	r_rowid(1000);

	avhttp::detail::unescape_path(q, q_escaped);

	boost::timer::cpu_timer cputimer;

	cputimer.start();

	db << "select date,channel,nick,message,rowid from avlog where channel=:c "
		"and message like \"%" << q_escaped << "%\" order  by strftime(`date`) DESC"
		, soci::into(r_date)
		, soci::into(r_channel)
		, soci::into(r_nick)
		, soci::into(r_message)
		, soci::into(r_rowid)
		, soci::use(c);

	pt::wptree results;
	// print out the result
	for (int i = 0; i < r_date.size() ; i ++)
	{
		pt::wptree onemsg;
		onemsg.put(L"date", boost::locale::conv::utf_to_utf<wchar_t>(r_date[i]));
		onemsg.put(L"channel", boost::locale::conv::utf_to_utf<wchar_t>(r_channel[i]));
		onemsg.put(L"nick", boost::locale::conv::utf_to_utf<wchar_t>(r_nick[i]));
		onemsg.put(L"message", boost::locale::conv::utf_to_utf<wchar_t>(r_message[i]));
		onemsg.put(L"id", boost::locale::conv::utf_to_utf<wchar_t>(r_rowid[i]));

		results.push_back(std::make_pair(L"", onemsg));
	}

	outjson.put(L"params.num_results", r_date.size());
	outjson.put_child(L"data", results);

	outjson.put(L"params.time_used", boost::locale::conv::utf_to_utf<wchar_t>(boost::timer::format(cputimer.elapsed(), 6, "%w")));

	io_service.post(
		boost::asio::detail::bind_handler(handler,
			boost::system::error_code(),
			outjson
		)
	);
};