HTTP中的Cookie和Session¶

void login(HttpRequest &req, HttpResponse &resp)
{
    LOG(LogLevel::INFO) << "进入登录模块";
    req.getParamKv();
}

void login(HttpRequest &req, HttpResponse &resp)
{   
    // ...

    // 添加Cookie
    resp.insertRespHead("Set-Cookie", "testCookie=newCookie");
}

void login(HttpRequest &req, HttpResponse &resp)
{
    // ...

    // 设置响应行
    std::string line = default_http_ver + " " + std::to_string(200) + resp.setStatusCodeDesc(200);
    resp.setRespLine(line);

    // 设置响应体
    std::string body = "<h1>Hello Linux</h1>";
    resp.setRespBody(body);
}

void insertRespCookies(const std::string &cookie_str)
{
    _cookies.push_back(cookie_str);
}

// 序列化
bool serialize(std::string &out_str)
{
    // ...

    // 单独为Cookie添加\r\n
    std::for_each(_cookies.begin(), _cookies.end(), [&](std::string s){
        s += default_sep;
        out_str += s;
    });

    // ...

    return true;
}

void login(HttpRequest &req, HttpResponse &resp)
{
    // ...

    // 添加Cookie
    // resp.insertRespHead("Set-Cookie", "testCookie=newCookie");
    resp.insertRespCookies("Set-Cookie: testCookie=newCookie");
    resp.insertRespCookies("Set-Cookie: testCookie1=newCookie1");
    resp.insertRespCookies("Set-Cookie: testCookie2=newCookie2");

    // ...
}

// 键值对分隔符
const std::string default_kv_sep = "=";
// Cookie分隔符
const std::string default_cookie_sep = "; ";

// HTTP请求
class HttpRequest
{
public:
    // ...

    // 获取Cookie键值对
    bool getCookiePairFromReqHead()
    {

    }
    // ...

private:
    // ...

    std::unordered_map<std::string, std::string> _cookie_kv; // Cookie键值对

    // ...
};

// 获取Cookie键值对
bool getCookiePairFromReqHead()
{
    // 先提取到Cookie
    auto keyPos = _kv.find("Cookie");
    if (keyPos == _kv.end())
        return false;

    // 在对应的value中找到Cookie值
    std::string cookie_value = _kv["Cookie"];

    auto pos = size_t(-1);

    while (true)
    {
        // 找到;
        auto pos1 = cookie_value.find(default_cookie_sep, pos + 1);
        if (pos1 == std::string::npos)
        {
            // 最后一个参数键值对
            auto pos_t = cookie_value.find(default_kv_sep, pos + 1);
            if (pos_t == std::string::npos)
                return false;

            std::string key = cookie_value.substr(pos + 1, pos_t - pos - 1);
            std::string value = cookie_value.substr(pos_t + 1);

            _cookie_kv.insert({key, value});
            break;
        }

        // 找到=
        auto pos2 = cookie_value.find(default_kv_sep, pos + 1);
        if (pos2 == std::string::npos)
            return false;

        std::string key = cookie_value.substr(pos + 1, pos2 - pos - 1);
        std::string value = cookie_value.substr(pos2 + 1, pos1 - pos2 - 1);

        _cookie_kv.insert({key, value});

        // 修改起始位置
        pos = pos1 + 1;
    }

    return true;
}

void getCookieKv()
{
    std::for_each(_cookie_kv.begin(), _cookie_kv.end(), [&](std::pair<std::string, std::string> kv)
                { std::cout << kv.first << ":" << kv.second << std::endl; });
}

// 反序列化
bool deserialize(std::string &in_str)
{
    // ...
    // 获取Cookie
    getCookiePairFromReqHead();
    // 打印获取到的Cookie
    getCookieKv();

    // ...

    return true;
}

struct tm *gmtime(const time_t *timep);

void login(HttpRequest &req, HttpResponse &resp)
{
    // ...

    std::string cookie_val = "testCookie=newCookie; expires=" + resp.getCookieExpiresTime(60);
    resp.insertRespCookies("Set-Cookie: " + cookie_val);

    // ...
}

void login(HttpRequest &req, HttpResponse &resp)
{
    // ...

    resp.insertRespCookies("Set-Cookie: testCookie=newCookie; path=/login.html");

    // ...
}

class Session
{
public:
    Session(const std::string &username, const std::string &status)
    :_username(username), _status(status)
    {
        _create_time = time(nullptr); // 获取时间戳就行了，后面实际需要，就转化就转换一下
    }
    ~Session()
    {}
public:
    std::string _username;
    std::string _status;
    uint64_t _create_time;
};

using session_ptr = std::shared_ptr<Session>;

class SessionManager
{
public:
    SessionManager()
    {
    }
    std::string addSession(session_ptr s)
    {
        // 使用boost库生成uuid
        // 创建一个随机数生成器
        boost::uuids::random_generator generator;

        // 生成一个随机 UUID
        boost::uuids::uuid id = generator();
        std::string sessionid = boost::uuids::to_string(id);
        _sessions.insert(std::make_pair(sessionid, s));
        return sessionid;
    }
    session_ptr getSession(const std::string sessionid)
    {
        if (_sessions.find(sessionid) == _sessions.end())
            return nullptr;
        return _sessions[sessionid];
    }
    ~SessionManager()
    {
    }

private:
    std::unordered_map<std::string, session_ptr> _sessions;
};

bool getReqParams()
{
    if (_req_method == "GET")
    {
        // 处理GET请求中的参数
        if (getReqParamsFromReqLine())
        {
            _hasArgs = true;
            return true;
        }
    }
    else if (_req_method == "POST")
    {
        // 处理POST请求中的参数
        if (getReqParamsFromBody())
        {
            _hasArgs = true;
            return true;
        }
    }

    return false;
}

bool deserialize(std::string &in_str)
{
    // ...

    // 获取参数内容
    if(getReqParams()) 
    {
        // 判断制定的sessionid是否存在，不存在则创建
        if (!_sm->getSession(getSessionIdFromCookiePair()))
        {
            std::string username = _param_kv["username"];
            // 根据参数的username创建Session对象，并处于活跃状态
            std::shared_ptr<Session> s = std::make_shared<Session>(username, "active");
            // 插入到SessionManager中
            _sessionid = _sm->addSession(s);
        }
    }

    return true;
}

// 获取_sessionid
std::string getSessionId()
{
    return _sessionid;
}

void login(HttpRequest &req, HttpResponse &resp)
{
    // ...

    std::string sessionid = req.getSessionId();
    std::string cookie_val = "sessionid=" + sessionid;
    resp.insertRespCookies("Set-Cookie: " + cookie_val);

    // ...
}