ProtoServiceBase.hpp

Go to the documentation of this file.

#ifndef _IDENTT_HRPC_PROTO_SERVICE_BASE_HPP_
#define _IDENTT_HRPC_PROTO_SERVICE_BASE_HPP_

#include "utils/BaseUtils.hpp"
#include "../proto/Hrpc.pb.h"

namespace identt {
namespace hrpc {

template <class HttpServerT>
class ProtoServiceBase {
protected:

    bool ProtoRequest(typename HttpServerT::ReqPtr request)
    {
        auto it=request->header.find("Content-Type");
        if(it==request->header.end())it=request->header.find("Accept");
        if(it==request->header.end()) return false;
        if ( it->second.compare(0,17,"application/proto",17)!=0) return false;
        it=request->header.find("Proto-Transfer-Encoding");
        return ( it->second.compare(0,6,"base64",6)==0);
    }

    bool SharedSecretOK(typename HttpServerT::ReqPtr request, std::string sharedkey)
    {
        auto it=request->header.find("Shared-Secret");
        if(it==request->header.end()) return false;
        return ( it->second == sharedkey );
    }

    int ServiceName(typename HttpServerT::ReqPtr request)
    {
        auto it=request->header.find("Service-Name");
        return (it==request->header.end()) ?  0 : std::stoi(it->second);
    }

    void ServiceOKAction(
        typename HttpServerT::RespPtr response,
        typename HttpServerT::ReqPtr request,
        std::string& payload)
    {
        *response << "HTTP/" << request->http_version << " 200 OK\r\n";
        *response << "Service-Name: " << ServiceName(request) << "\r\n";
        *response << "Content-Type: " << "application/proto" << "\r\n";
        *response << "Proto-Transfer-Encoding: " << "base64" << "\r\n";
        *response << "Content-Length: " << payload.length() << "\r\n";
        *response << "\r\n" << payload.c_str();
    }

    void ServiceOKAction(
        typename HttpServerT::RespPtr response,
        typename HttpServerT::ReqPtr request,
        std::string&& payload)
    {
        *response << "HTTP/" << request->http_version << " 200 OK\r\n";
        *response << "Service-Name: " << ServiceName(request) << "\r\n";
        *response << "Content-Type: " << "application/proto" << "\r\n";
        *response << "Proto-Transfer-Encoding: " << "base64" << "\r\n";
        *response << "Content-Length: " << payload.length() << "\r\n";
        *response << "\r\n" << payload.c_str();
    }

    void ServiceErrAction(
        typename HttpServerT::RespPtr response,
        typename HttpServerT::ReqPtr request,
        int ecode, std::string& emsg)
    {
        *response << "HTTP/" << request->http_version << " 200 OK\r\n";
        *response << "Service-Name: " << ServiceName(request) << "\r\n";
        *response << "Service-Error: " << ecode << "\r\n";
        *response << "Content-Type: application/proto" << "\r\n";
        *response << "Content-Length: " << emsg.length() << "\r\n";
        *response << "\r\n" << emsg.c_str();
    }

};
} // namespace hrpc
} // namespace identt

#endif /* _IDENTT_HRPC_PROTO_SERVICE_BASE_HPP_ */