SimpleController.h

Go to the documentation of this file.

/*
 *  Copyright (c) 2015-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */
#pragma once

#include <proxygen/lib/http/session/HTTPSessionController.h>
#include <string>

namespace proxygen {

class HTTPErrorPage;
class HTTPSessionAcceptor;

class SimpleController : public HTTPSessionController {
 public:
  explicit SimpleController(HTTPSessionAcceptor* acceptor);

  HTTPTransactionHandler* getRequestHandler(HTTPTransaction& txn,
                                            HTTPMessage* msg) override;

  HTTPTransactionHandler* getParseErrorHandler(
    HTTPTransaction* txn,
    const HTTPException& error,
    const folly::SocketAddress& localAddress) override;

  HTTPTransactionHandler* getTransactionTimeoutHandler(
    HTTPTransaction* txn,
    const folly::SocketAddress& localAddress) override;

  void attachSession(HTTPSessionBase*) override;
  void detachSession(const HTTPSessionBase*) override;

  std::chrono::milliseconds getGracefulShutdownTimeout() const override;
 protected:
  HTTPTransactionHandler* createErrorHandler(
      uint32_t statusCode,
      const std::string& statusMessage,
      const HTTPErrorPage* errorPage);

  HTTPSessionAcceptor* const acceptor_{nullptr};
};

}