thrift_server/uwsgi_handler.cpp

Go to the documentation of this file.

/*
 * Copyright (c) 2013-2014 ADTECH GmbH
 * Licensed under MIT (https://github.com/adtechlabs/libtasks/blob/master/COPYING)
 *
 * Author: Andreas Pohl
 */

#include <tasks/dispatcher.h>
#include <tasks/net/acceptor.h>
#include <tasks/net/http_response.h>
#include <tasks/net/uwsgi_thrift_transport.h>
#include <tasks/tools/thrift_server_writer.h>

#include <iostream>
#include <arpa/inet.h>
#include <thrift/protocol/TBinaryProtocol.h>

#ifdef PROFILER
#include <google/profiler.h>
#endif

#include "uwsgi_handler.h"
#include "stats.h"

bool uwsgi_handler::handle_request() {
    // Do something with the url for example
    // const std::string& url = request().var("REQUEST_URI");
    // Or print all incoming variables
    // request().print_vars();

    typedef tasks::net::uwsgi_thrift_transport<tasks::net::http_response> transport_type;
    typedef apache::thrift::protocol::TBinaryProtocol protocol_type;

    boost::shared_ptr<transport_type> transport(new transport_type(response_p()));
    boost::shared_ptr<protocol_type> protocol(new protocol_type(transport));
    tasks::tools::thrift_server_writer<id_name, transport_type, protocol_type> writer("lookup", transport, protocol);

    id_name m;
    m.id = 33;
    m.name = "testtest";
    writer.write(m);

    // Now send back a response
    response().set_status("200 OK");
    response().set_header("Content-Type", "application/x-thrift");
    send_response();

    stats::inc_req();

    return true;
}

int main(int argc, char** argv) {
#ifdef PROFILER
    ProfilerStart("uwsgi_server.prof");
#endif
    stats s;
    tasks::net::acceptor<uwsgi_handler> srv(12345);
    auto tasks = std::vector<tasks::task*>{&srv, &s};
    tasks::dispatcher::instance()->run(tasks);
#ifdef PROFILER
    ProfilerStop();
#endif
    return 0;
}