compsci-industry
/
TechEmpower.FrameworkBenchmarks
kopia lustrzana https://github.com/TechEmpower/FrameworkBenchmarks.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
							#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 != 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;
  }
}