Controller.h

浏览该文件的文档.

// Copyright 2024 libsese
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
#pragma once
 
#include <sese/net/http/HttpServletContext.h>
 
#include <set>
#include <vector>
#include <functional>
#include <utility>
 
namespace sese::net::http {
 
class Servlet {
public:
    using Ptr = std::shared_ptr<Servlet>;
    using Callback = std::function<void(HttpServletContext &)>;
 
    Servlet();
 
    Servlet(RequestType expect, const std::string &url);
 
    void setCallback(Callback callback);
 
    [[nodiscard]] const std::string &getUri() const { return uri; }
 
    [[nodiscard]] RequestType getExpectType() const { return expect_type; }
 
    [[nodiscard]] auto &getExpectQueryArgs() const { return expect_query_args; }
 
    [[nodiscard]] auto &getExpectHeaders() const { return expect_headers; }
 
    void requiredQueryArg(const std::string &arg);
 
    void requiredHeader(const std::string &arg);
 
    void invoke(HttpServletContext &ctx) const;
 
    Servlet &operator=(Callback callback) {
        setCallback(std::move(callback));
        return *this;
    }
 
private:
    std::string uri;
    RequestType expect_type;
    std::set<std::string> expect_query_args;
    std::set<std::string> expect_headers;
    Callback callback;
};
 
class Controller {
public:
    using Ptr = std::shared_ptr<Controller>;
 
    Controller() = default;
    virtual ~Controller() = default;
 
    virtual void init() = 0;
 
    auto begin() { return servlets.begin(); }
    auto end() { return servlets.end(); }
 
protected:
    std::vector<Servlet> servlets;
};
 
} // namespace sese::net::http
 
#define SESE_CTRL(name, ...)                                \
    class name final : public sese::net::http::Controller { \
    public:                                                 \
        using RequestType = sese::net::http::RequestType;   \
        name() : Controller() {                             \
        }                                                   \
        void init() override;                               \
                                                            \
    private:                                                \
        __VA_ARGS__;                                        \
    };                                                      \
    void name::init()
 
#define SESE_URL(name, method, url)                                 \
    this->servlets.emplace_back(method, url);                       \
    sese::net::http::Servlet &name = servlets[servlets.size() - 1]; \
    name = [this](sese::net::http::HttpServletContext &ctx)