compsci-industry
/
TechEmpower.FrameworkBenchmarks
mirror of https://github.com/TechEmpower/FrameworkBenchmarks.git


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
							#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("&amp;");       break;
            case '\"': buffer.append("&quot;");      break;
            case '\'': buffer.append("&apos;");      break;
            case '<':  buffer.append("&lt;");        break;
            case '>':  buffer.append("&gt;");        break;
            default:   buffer.append(&data[pos], 1); break;
        }
    }
    return std::move(buffer);
}

int main(int argc, char* argv[])
{

  if (argc != 4)
  {
    std::cerr << "Usage: " << argv[0] << " mysql_host port pid_file" << std::endl;
    return 1;
  }
  
  auto hello_api = make_api(

    _plaintext = [] () { return "Hello, World!"; },
    _json = [] () { return D(_message = "Hello, World!"); },

    _db = [] (rn_orm& orm) {
      random_number r;
      orm.find_by_id(1245, r);
      return 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 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 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 ss.str();
    }
  
    ).bind_factories(
      mysql_connection_factory(argv[1], "benchmarkdbuser", "benchmarkdbpass", "hello_world"),
      fortune_orm_factory("Fortune"),
      rn_orm_factory("World")
      );
  
  try
  {
    // Demonize the process.
    if (daemon(0,0))
    {
      std::cerr << "Cannot start the daemon." << std::endl;
      exit(1);
    }

    // Write the pid.
    std::ofstream pidfile(argv[3]);
    pidfile << getpid() << std::endl;
    pidfile.close();

    // Start the server.
    sl::mhd_json_serve(hello_api, atoi(argv[2]));
  }
  catch (std::exception& e)
  {
    std::cerr << e.what() << std::endl;
  }
  catch (sl::error::error& e)
  {
    std::cerr << e.what() << std::endl;
  }
}