2019-05-04 23:21:33 -07:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <functional>
|
|
|
|
|
#include <sys/uio.h>
|
|
|
|
|
#include <unordered_map>
|
|
|
|
|
#include <unordered_set>
|
|
|
|
|
|
2019-05-04 23:42:03 -07:00
|
|
|
#include "firebuf/stream_buffer.h"
|
|
|
|
|
|
2019-05-04 23:21:33 -07:00
|
|
|
#include "request.h"
|
|
|
|
|
|
|
|
|
|
namespace firecgi {
|
|
|
|
|
|
|
|
|
|
class Connection {
|
|
|
|
|
public:
|
|
|
|
|
Connection(int sock, const sockaddr_in6& client_addr, const std::function<void(std::unique_ptr<Request>)>& callback, const std::unordered_set<std::string_view>& headers);
|
|
|
|
|
~Connection();
|
|
|
|
|
|
|
|
|
|
[[nodiscard]] int Read();
|
|
|
|
|
[[nodiscard]] bool Write(const std::vector<iovec>& vecs);
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
const int sock_;
|
|
|
|
|
const std::function<void(std::unique_ptr<Request>)>& callback_;
|
|
|
|
|
const std::unordered_set<std::string_view>& headers_;
|
|
|
|
|
|
|
|
|
|
uint64_t requests_ = 0;
|
|
|
|
|
|
2019-05-04 23:42:03 -07:00
|
|
|
firebuf::StreamBuffer buf_;
|
2019-05-04 23:21:33 -07:00
|
|
|
|
|
|
|
|
std::unique_ptr<Request> request_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
} // namespace firecgi
|