HttpServer/HttpResponse.h at main · cellphonef/HttpServer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#ifndef HTTPSERVER_HTTPRESPONSE_H
#define HTTPSERVER_HTTPRESPONSE_H

#include "HttpRequest.h"
#include "ConnectionPool.h"

#include <map>

#include <sys/stat.h>
#include <mysql.h>


class HttpResponse {
public:

    enum class HttpCode {
        k200Ok = 200,
        k300MovedPermanently = 301,
        k400BadRequest = 400,
        k403Forbidden = 403,
        k404NotFound = 404,
        k500InternalServerError = 500,
    };

    HttpResponse();
    void initResponse(ConnectionPool* connPool, char* docRoot);
    void reset();
    void doResponse(HttpRequest& req, Buffer& inBuf, Buffer& outBuf);
    void addResponse(Buffer& buf);

    // getter & setter
    void setHttpCode(HttpCode httpCode);
    HttpCode getHttpCode() const;
    void setIsCloseConn(bool isClose);
    bool getIsCloseConn() const;
    const char* getRealFile() const;
    const struct stat& getFileStat() const;
    std::string getHttpCodeString() const;


private:
    void addResponseLine_(Buffer& buf);
    void addResponseHeader_(Buffer& buf);
    void addBlankLine_(Buffer& buf);
    void addContent_(Buffer& buf);

    // 用于设置响应行
    std::string version_;
    HttpCode httpCode_;

    // 用于设置响应头
    std::map<std::string, std::string> headers_;
    bool isCloseConn_;  // keep-alive ?

    // 用于设置响应体
    // 1. 设置出错情况
    static const char kError400[];
    static const char kError403[];
    static const char kError404[];
    static const char kError500[];
    // 2. 静态资源访问
    static const int kFilePathMaxLen = 200;
    static const int kFileNameMaxLen = 100;
    static char* docRoot_;
    struct stat fileStat_;
    char realFile_[kFilePathMaxLen];
    char url_[kFileNameMaxLen];
    // 3. 登录注册
    MYSQL* mysql_;
    static ConnectionPool* connPool_;


};

#endif  // HTTPSERVER_HTTPRESPONSE_H