101 lines
2.6 KiB
C
101 lines
2.6 KiB
C
|
#ifndef HV_HTTP_RESPONSE_WRITER_H_
|
||
|
|
#define HV_HTTP_RESPONSE_WRITER_H_
|
||
|
|
|
||
|
|
#include "Channel.h"
|
||
|
|
#include "HttpMessage.h"
|
||
|
|
|
||
|
|
namespace hv {
|
||
|
|
|
||
|
|
class HV_EXPORT HttpResponseWriter : public SocketChannel {
|
||
|
|
public:
|
||
|
|
HttpResponsePtr response;
|
||
|
|
enum State {
|
||
|
|
SEND_BEGIN = 0,
|
||
|
|
SEND_HEADER,
|
||
|
|
SEND_BODY,
|
||
|
|
SEND_CHUNKED,
|
||
|
|
SEND_CHUNKED_END,
|
||
|
|
SEND_END,
|
||
|
|
} state: 8, end: 8;
|
||
|
|
HttpResponseWriter(hio_t* io, const HttpResponsePtr& resp)
|
||
|
|
: SocketChannel(io)
|
||
|
|
, response(resp)
|
||
|
|
, state(SEND_BEGIN)
|
||
|
|
, end(SEND_BEGIN)
|
||
|
|
{}
|
||
|
|
~HttpResponseWriter() {}
|
||
|
|
|
||
|
|
// Begin -> End
|
||
|
|
// Begin -> WriteResponse -> End
|
||
|
|
// Begin -> WriteStatus -> WriteHeader -> WriteBody -> End
|
||
|
|
// Begin -> EndHeaders("Content-Type", "text/event-stream") -> write -> write -> ... -> close
|
||
|
|
// Begin -> EndHeaders("Content-Length", content_length) -> WriteBody -> WriteBody -> ... -> End
|
||
|
|
// Begin -> EndHeaders("Transfer-Encoding", "chunked") -> WriteChunked -> WriteChunked -> ... -> End
|
||
|
|
|
||
|
|
int Begin() {
|
||
|
|
state = end = SEND_BEGIN;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int WriteStatus(http_status status_codes) {
|
||
|
|
response->status_code = status_codes;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int WriteHeader(const char* key, const char* value) {
|
||
|
|
response->SetHeader(key, value);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
int WriteHeader(const char* key, T num) {
|
||
|
|
response->SetHeader(key, hv::to_string(num));
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int WriteCookie(const HttpCookie& cookie) {
|
||
|
|
response->cookies.push_back(cookie);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int EndHeaders(const char* key = NULL, const char* value = NULL);
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
int EndHeaders(const char* key, T num) {
|
||
|
|
std::string value = hv::to_string(num);
|
||
|
|
return EndHeaders(key, value.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
int WriteChunked(const char* buf, int len = -1);
|
||
|
|
|
||
|
|
int WriteChunked(const std::string& str) {
|
||
|
|
return WriteChunked(str.c_str(), str.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
int EndChunked() {
|
||
|
|
return WriteChunked(NULL, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
int WriteBody(const char* buf, int len = -1);
|
||
|
|
|
||
|
|
int WriteBody(const std::string& str) {
|
||
|
|
return WriteBody(str.c_str(), str.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
int WriteResponse(HttpResponse* resp);
|
||
|
|
|
||
|
|
int SSEvent(const std::string& data, const char* event = "message");
|
||
|
|
|
||
|
|
int End(const char* buf = NULL, int len = -1);
|
||
|
|
|
||
|
|
int End(const std::string& str) {
|
||
|
|
return End(str.c_str(), str.size());
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
typedef std::shared_ptr<hv::HttpResponseWriter> HttpResponseWriterPtr;
|
||
|
|
|
||
|
|
#endif // HV_HTTP_RESPONSE_WRITER_H_
|