InfoService.hpp

Go to the documentation of this file.

#ifndef _IDENTT_QUERY_INFO_SERVICE_HPP_
#define _IDENTT_QUERY_INFO_SERVICE_HPP_

#include <query/QueryBase.hpp>

namespace identt {
namespace query {

template <class HttpServerT>
class InfoService : protected identt::query::ServiceBase<HttpServerT> {
public:

    InfoService(
        identt::utils::SharedTable::pointer stptr,
        typename std::shared_ptr<HttpServerT> server,
        ::identt::query::HelpQuery::pointer helpquery,
        const unsigned int scope)
        : identt::query::ServiceBase<HttpServerT>(IDENTT_SERVICE_SCOPE_HTTP | IDENTT_SERVICE_SCOPE_HTTPS)
    {
        if (!(this->myscope & scope)) return; // scope mismatch

        // Endpoint : GET help
        helpquery->add({scope,"GET help", { "This query, Gets info about endpoints this server" } });

        server->resource["/help$"]["GET"]
        =[this,stptr,scope,helpquery](typename HttpServerT::RespPtr response, typename HttpServerT::ReqPtr request) {
            IDENTT_PARALLEL_ONE([this,stptr,scope,helpquery,response,request] {
                try {
                    DLOG(INFO) << request->path;
                    std::string output;
                    std::stringstream content_stream;
                    for (auto& var: helpquery->get(scope))
                    {
                        content_stream << "\r\n\r\n" << var.route ;
                        size_t counter=0;
                        for (auto& desc: var.desc) content_stream << "\r\n -- " << desc;
                    }
                    output = content_stream.str();
                    *response << "HTTP/1.1 200 OK\r\n";
                    *response << "Content-type: text/plain\r\n";
                    *response << "Content-Length: " << output.length() << "\r\n\r\n" << output.c_str();
                    this->HttpOKAction(response,request,200,"OK","application/json",output,true);
                } catch (...)
                {
                    this->HttpErrorAction(response,request,500,"INTERNAL SERVER ERROR");
                }
            });
        };


        // Endpoint : GET info
        helpquery->add({scope,"GET info", { "Gets info about this server" } });

        server->resource["/info$"]["GET"]
        =[this,stptr](typename HttpServerT::RespPtr response, typename HttpServerT::ReqPtr request) {
            IDENTT_PARALLEL_ONE([this,stptr,response,request] {
                try {
                    LOG(INFO) << request->path;
                    ::identt::query::StateT sstate;
                    sstate.set_ts( IDENTT_CURRTIME_MS );
                    sstate.set_is_ready( stptr->is_ready.Get() );
                    // get other params if ready
                    if (sstate.is_ready())
                    {
                        sstate.set_hostname( stptr->hostname.Get() );
                        sstate.set_is_master( stptr->is_master.Get() );
                        if(!sstate.is_master())
                            sstate.set_master( stptr->master.Get() );
                    }

                    // aftermath
                    std::string output;
                    pb2json(&sstate , output);
                    this->HttpOKAction(response,request,200,"OK","application/json",output,true);
                } catch (...)
                {
                    this->HttpErrorAction(response,request,500,"INTERNAL SERVER ERROR");
                }
            });
        };

    }

private:

};
} // namespace query
} // namespace identt

#endif // _IDENTT_QUERY_INFO_SERVICE_HPP_