diff options
Diffstat (limited to 'include/routing.h')
-rw-r--r-- | include/routing.h | 362 |
1 files changed, 22 insertions, 340 deletions
diff --git a/include/routing.h b/include/routing.h index 28700c1..2483bb9 100644 --- a/include/routing.h +++ b/include/routing.h @@ -279,329 +279,42 @@ namespace crow std::array<unsigned, (int)ParamType::MAX> param_childrens{}; std::unordered_map<std::string, unsigned> children; - bool IsSimpleNode() const - { - return - !rule_index && - std::all_of( - std::begin(param_childrens), - std::end(param_childrens), - [](unsigned x){ return !x; }); - } + bool IsSimpleNode() const; }; - Trie() : nodes_(1) - { - } - -private: - void optimizeNode(Node* node) - { - for(auto x : node->param_childrens) - { - if (!x) - continue; - Node* child = &nodes_[x]; - optimizeNode(child); - } - if (node->children.empty()) - return; - bool mergeWithChild = true; - for(auto& kv : node->children) - { - Node* child = &nodes_[kv.second]; - if (!child->IsSimpleNode()) - { - mergeWithChild = false; - break; - } - } - if (mergeWithChild) - { - decltype(node->children) merged; - for(auto& kv : node->children) - { - Node* child = &nodes_[kv.second]; - for(auto& child_kv : child->children) - { - merged[kv.first + child_kv.first] = child_kv.second; - } - } - node->children = std::move(merged); - optimizeNode(node); - } - else - { - for(auto& kv : node->children) - { - Node* child = &nodes_[kv.second]; - optimizeNode(child); - } - } - } - - void optimize() - { - optimizeNode(head()); - } - -public: - void validate() - { - if (!head()->IsSimpleNode()) - throw std::runtime_error("Internal error: Trie header should be simple!"); - optimize(); - } - - std::pair<unsigned, routing_params> find(const request& req, const Node* node = nullptr, unsigned pos = 0, routing_params* params = nullptr) const - { - routing_params empty; - if (params == nullptr) - params = ∅ - - unsigned found{}; - routing_params match_params; - - if (node == nullptr) - node = head(); - if (pos == req.url.size()) - return {node->rule_index, *params}; - - auto update_found = [&found, &match_params](std::pair<unsigned, routing_params>& ret) - { - if (ret.first && (!found || found > ret.first)) - { - found = ret.first; - match_params = std::move(ret.second); - } - }; - - if (node->param_childrens[(int)ParamType::INT]) - { - char c = req.url[pos]; - if ((c >= '0' && c <= '9') || c == '+' || c == '-') - { - char* eptr; - errno = 0; - long long int value = strtoll(req.url.data()+pos, &eptr, 10); - if (errno != ERANGE && eptr != req.url.data()+pos) - { - params->int_params.push_back(value); - auto ret = find(req, &nodes_[node->param_childrens[(int)ParamType::INT]], eptr - req.url.data(), params); - update_found(ret); - params->int_params.pop_back(); - } - } - } - - if (node->param_childrens[(int)ParamType::UINT]) - { - char c = req.url[pos]; - if ((c >= '0' && c <= '9') || c == '+') - { - char* eptr; - errno = 0; - unsigned long long int value = strtoull(req.url.data()+pos, &eptr, 10); - if (errno != ERANGE && eptr != req.url.data()+pos) - { - params->uint_params.push_back(value); - auto ret = find(req, &nodes_[node->param_childrens[(int)ParamType::UINT]], eptr - req.url.data(), params); - update_found(ret); - params->uint_params.pop_back(); - } - } - } - - if (node->param_childrens[(int)ParamType::DOUBLE]) - { - char c = req.url[pos]; - if ((c >= '0' && c <= '9') || c == '+' || c == '-' || c == '.') - { - char* eptr; - errno = 0; - double value = strtod(req.url.data()+pos, &eptr); - if (errno != ERANGE && eptr != req.url.data()+pos) - { - params->double_params.push_back(value); - auto ret = find(req, &nodes_[node->param_childrens[(int)ParamType::DOUBLE]], eptr - req.url.data(), params); - update_found(ret); - params->double_params.pop_back(); - } - } - } - - if (node->param_childrens[(int)ParamType::STRING]) - { - size_t epos = pos; - for(; epos < req.url.size(); epos ++) - { - if (req.url[epos] == '/') - break; - } - - if (epos != pos) - { - params->string_params.push_back(req.url.substr(pos, epos-pos)); - auto ret = find(req, &nodes_[node->param_childrens[(int)ParamType::STRING]], epos, params); - update_found(ret); - params->string_params.pop_back(); - } - } - - if (node->param_childrens[(int)ParamType::PATH]) - { - size_t epos = req.url.size(); + Trie(); - if (epos != pos) - { - params->string_params.push_back(req.url.substr(pos, epos-pos)); - auto ret = find(req, &nodes_[node->param_childrens[(int)ParamType::PATH]], epos, params); - update_found(ret); - params->string_params.pop_back(); - } - } + void validate(); - for(auto& kv : node->children) - { - const std::string& fragment = kv.first; - const Node* child = &nodes_[kv.second]; + void add(const std::string& url, unsigned rule_index); - if (req.url.compare(pos, fragment.size(), fragment) == 0) - { - auto ret = find(req, child, pos + fragment.size(), params); - update_found(ret); - } - } + std::pair<unsigned, routing_params> find( + const request& req, + const Node* node = nullptr, + unsigned pos = 0, + routing_params* params = nullptr) const; - return {found, match_params}; - } + void debug_print(); - void add(const std::string& url, unsigned rule_index) - { - unsigned idx{0}; - - for(unsigned i = 0; i < url.size(); i ++) - { - char c = url[i]; - if (c == '<') - { - static struct ParamTraits - { - ParamType type; - std::string name; - } paramTraits[] = - { - { ParamType::INT, "<int>" }, - { ParamType::UINT, "<uint>" }, - { ParamType::DOUBLE, "<float>" }, - { ParamType::DOUBLE, "<double>" }, - { ParamType::STRING, "<str>" }, - { ParamType::STRING, "<string>" }, - { ParamType::PATH, "<path>" }, - }; - - for(auto& x:paramTraits) - { - if (url.compare(i, x.name.size(), x.name) == 0) - { - if (!nodes_[idx].param_childrens[(int)x.type]) - { - auto new_node_idx = new_node(); - nodes_[idx].param_childrens[(int)x.type] = new_node_idx; - } - idx = nodes_[idx].param_childrens[(int)x.type]; - i += x.name.size(); - break; - } - } - - i --; - } - else - { - std::string piece(&c, 1); - if (!nodes_[idx].children.count(piece)) - { - auto new_node_idx = new_node(); - nodes_[idx].children.emplace(piece, new_node_idx); - } - idx = nodes_[idx].children[piece]; - } - } - if (nodes_[idx].rule_index) - throw std::runtime_error("handler already exists for " + url); - nodes_[idx].rule_index = rule_index; - } private: - void debug_node_print(Node* n, int level) - { - for(int i = 0; i < (int)ParamType::MAX; i ++) - { - if (n->param_childrens[i]) - { - CROW_LOG_DEBUG << std::string(2*level, ' ') /*<< "("<<n->param_childrens[i]<<") "*/; - switch((ParamType)i) - { - case ParamType::INT: - CROW_LOG_DEBUG << "<int>"; - break; - case ParamType::UINT: - CROW_LOG_DEBUG << "<uint>"; - break; - case ParamType::DOUBLE: - CROW_LOG_DEBUG << "<float>"; - break; - case ParamType::STRING: - CROW_LOG_DEBUG << "<str>"; - break; - case ParamType::PATH: - CROW_LOG_DEBUG << "<path>"; - break; - default: - CROW_LOG_DEBUG << "<ERROR>"; - break; - } - - debug_node_print(&nodes_[n->param_childrens[i]], level+1); - } - } - for(auto& kv : n->children) - { - CROW_LOG_DEBUG << std::string(2*level, ' ') /*<< "(" << kv.second << ") "*/ << kv.first; - debug_node_print(&nodes_[kv.second], level+1); - } - } - - public: - void debug_print() - { - debug_node_print(head(), 0); - } + void debug_node_print(Node* n, int level); - private: - const Node* head() const - { - return &nodes_.front(); - } + void optimizeNode(Node* node); + void optimize(); - Node* head() - { - return &nodes_.front(); - } + const Node* head() const; + Node* head(); - unsigned new_node() - { - nodes_.resize(nodes_.size()+1); - return nodes_.size() - 1; - } + unsigned new_node(); + private: std::vector<Node> nodes_; }; class Router { public: - Router() : rules_(1) {} + Router(); template <uint64_t N> typename black_magic::arguments<N>::type::template rebind<TaggedRule>& new_rule_tagged(const std::string& rule) { @@ -612,42 +325,11 @@ public: return *ruleObject; } - void validate() - { - trie_.validate(); - for(auto& rule:rules_) - { - if (rule) - rule->validate(); - } - } - - void handle(const request& req, response& res) - { - auto found = trie_.find(req); - - unsigned rule_index = found.first; - - if (!rule_index) - { - CROW_LOG_DEBUG << "Cannot match rules " << req.url; - res = response(404); - res.end(); - return; - } + void validate(); - if (rule_index >= rules_.size()) - throw std::runtime_error("Trie internal structure corrupted!"); - - CROW_LOG_DEBUG << "Matched rule '" << ((TaggedRule<>*)rules_[rule_index].get())->rule_ << "'"; - - rules_[rule_index]->handle(req, res, found.second); - } - - void debug_print() - { - trie_.debug_print(); - } + void handle(const request& req, response& res); + + void debug_print(); private: std::vector<std::unique_ptr<BaseRule>> rules_; |