ssgx_http_t_filter.h

Go to the documentation of this file.

#ifndef SSGXLIB_SSGX_HTTP_T_FILTER_H
#define SSGXLIB_SSGX_HTTP_T_FILTER_H
 
#include <functional>
#include <mutex>
#include <string>
#include <unordered_map>
#include <vector>
 
#include "ssgx_http_t_structs.h"
 
namespace ssgx {
namespace http_t {
class Filter {
  public:
    virtual bool Before(Request& request, Response& response) {
        return true;
    }
 
    virtual void After(Request& request, Response& response) {
    }
 
    virtual ~Filter() = default;
};
 
class FilterChain {
  public:
    void AddFilter(std::unique_ptr<Filter>&& filter) {
        filters_.emplace_back(std::move(filter));
    }
    Response Execute(Request& request, const std::function<void(Request&, Response&)>& handler) {
        Response response;
 
        for (auto& filter : filters_) {
            if (!filter->Before(request, response)) {
                return response;
            }
        }
 
        handler(request, response);
 
        for (auto& filter : filters_) {
            filter->After(request, response);
        }
 
        return response;
    }
 
  private:
    std::vector<std::unique_ptr<Filter>> filters_;
};
 
} // namespace http_t
} // namespace ssgx
 
#endif // SSGXLIB_SSGX_HTTP_T_FILTER_H