|
@@ -0,0 +1,137 @@
|
|
|
+#include <algorithm>
|
|
|
+#include <unistd.h>
|
|
|
+#include <iostream>
|
|
|
+#include <silicon/backends/mhd.hh>
|
|
|
+#include <silicon/api.hh>
|
|
|
+#include <silicon/middlewares/mysql_connection.hh>
|
|
|
+#include <silicon/middlewares/mysql_orm.hh>
|
|
|
+#include "symbols.hh"
|
|
|
+
|
|
|
+using namespace s;
|
|
|
+using namespace sl;
|
|
|
+
|
|
|
+typedef decltype(D(_id(_auto_increment, _primary_key) = int(),
|
|
|
+ _randomNumber = int())) random_number;
|
|
|
+
|
|
|
+typedef decltype(D(_id(_auto_increment, _primary_key) = int(),
|
|
|
+ _message = std::string())) fortune;
|
|
|
+
|
|
|
+typedef mysql_orm_factory<random_number> rn_orm_factory;
|
|
|
+typedef mysql_orm<random_number> rn_orm;
|
|
|
+
|
|
|
+typedef mysql_orm_factory<fortune> fortune_orm_factory;
|
|
|
+typedef mysql_orm<fortune> fortune_orm;
|
|
|
+
|
|
|
+std::string escape_html_entities(const std::string& data)
|
|
|
+{
|
|
|
+ std::string buffer;
|
|
|
+ buffer.reserve(data.size());
|
|
|
+ for(size_t pos = 0; pos != data.size(); ++pos) {
|
|
|
+ switch(data[pos]) {
|
|
|
+ case '&': buffer.append("&"); break;
|
|
|
+ case '\"': buffer.append("""); break;
|
|
|
+ case '\'': buffer.append("'"); break;
|
|
|
+ case '<': buffer.append("<"); break;
|
|
|
+ case '>': buffer.append(">"); break;
|
|
|
+ default: buffer.append(&data[pos], 1); break;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return std::move(buffer);
|
|
|
+}
|
|
|
+
|
|
|
+int main(int argc, char* argv[])
|
|
|
+{
|
|
|
+
|
|
|
+ if (argc != 3)
|
|
|
+ {
|
|
|
+ std::cerr << "Usage: " << argv[0] << " mysql_host port" << std::endl;
|
|
|
+ return 1;
|
|
|
+ }
|
|
|
+
|
|
|
+ auto hello_api = make_api(
|
|
|
+
|
|
|
+ _plaintext = [] () { return response(_content_type = "text/plain",
|
|
|
+ _body = "Hello, World!"); },
|
|
|
+
|
|
|
+ _json = [] () { return response(_content_type = "application/json",
|
|
|
+ _body = D(_message = "Hello, World!")); },
|
|
|
+
|
|
|
+ _db = [] (rn_orm& orm) {
|
|
|
+ random_number r;
|
|
|
+ orm.find_by_id(1245, r);
|
|
|
+ return response(_content_type = "application/json",
|
|
|
+ _body = r);
|
|
|
+ },
|
|
|
+
|
|
|
+ _queries = [] (rn_orm& orm, get_parameters& get_params) {
|
|
|
+ int N = atoi(get_params["queries"].c_str());
|
|
|
+ N = std::max(1, std::min(N, 500));
|
|
|
+
|
|
|
+ std::vector<random_number> qs(N);
|
|
|
+ for (int i = 0; i < N; i++)
|
|
|
+ orm.find_by_id(1 + rand() % 9999, qs[i]);
|
|
|
+ return response(_content_type = "application/json",
|
|
|
+ _body = std::move(qs));
|
|
|
+ },
|
|
|
+
|
|
|
+ _updates = [] (rn_orm& orm, get_parameters& get_params) {
|
|
|
+ int N = atoi(get_params["queries"].c_str());
|
|
|
+ N = std::max(1, std::min(N, 500));
|
|
|
+
|
|
|
+ std::vector<random_number> qs(N);
|
|
|
+ for (int i = 0; i < N; i++)
|
|
|
+ {
|
|
|
+ orm.find_by_id(1 + rand() % 9999, qs[i]);
|
|
|
+ qs[i].randomNumber = 1 + rand() % 9999;
|
|
|
+ orm.update(qs[i]);
|
|
|
+ }
|
|
|
+ return response(_content_type = "application/json",
|
|
|
+ _body = std::move(qs));
|
|
|
+ },
|
|
|
+
|
|
|
+ _fortunes = [] (fortune_orm& orm) {
|
|
|
+ std::vector<fortune> table;
|
|
|
+ orm.forall([&] (fortune& f) { table.push_back(f); });
|
|
|
+ table.push_back(fortune(0, "Additional fortune added at request time."));
|
|
|
+
|
|
|
+ std::sort(table.begin(), table.end(),
|
|
|
+ [] (const fortune& a, const fortune& b) { return a.message < b.message; });
|
|
|
+
|
|
|
+ std::stringstream ss;
|
|
|
+
|
|
|
+ ss << "<!DOCTYPE html><html><head><title>Fortunes</title></head><body><table><tr><th>id</th><th>message</th></tr>";
|
|
|
+ for(auto& f : table)
|
|
|
+ ss << "<tr><td>" << f.id << "</td><td>" << escape_html_entities(f.message) << "</td></tr>";
|
|
|
+ ss << "</table></body></html>";
|
|
|
+
|
|
|
+ return response(_content_type = "text/html",
|
|
|
+ _body = ss.str());
|
|
|
+ }
|
|
|
+
|
|
|
+ ).bind_factories(
|
|
|
+ mysql_connection_factory(argv[1], "benchmarkdbuser", "benchmarkdbpass", "hello_world"),
|
|
|
+ fortune_orm_factory("Fortune"),
|
|
|
+ rn_orm_factory("World")
|
|
|
+ );
|
|
|
+
|
|
|
+ try
|
|
|
+ {
|
|
|
+
|
|
|
+ // Start the server.
|
|
|
+ sl::mhd_json_serve(hello_api, atoi(argv[2])
|
|
|
+#ifdef TFB_USE_EPOLL
|
|
|
+ , _linux_epoll, _nthreads = 1000
|
|
|
+#else
|
|
|
+ , _one_thread_per_connection
|
|
|
+#endif
|
|
|
+ );
|
|
|
+ }
|
|
|
+ catch (std::exception& e)
|
|
|
+ {
|
|
|
+ std::cerr << e.what() << std::endl;
|
|
|
+ }
|
|
|
+ catch (sl::error::error& e)
|
|
|
+ {
|
|
|
+ std::cerr << e.what() << std::endl;
|
|
|
+ }
|
|
|
+}
|