feat(routing): multi-segment wildcard matches
							parent
							
								
									3ae1b62dd0
								
							
						
					
					
						commit
						115bf74456
					
				| 
						 | 
				
			
			@ -23,6 +23,25 @@ lnm_http_step_err print_step(lnm_http_conn *conn) {
 | 
			
		|||
  return lnm_http_step_err_done;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
lnm_http_step_err print_step2(lnm_http_conn *conn) {
 | 
			
		||||
  lnm_http_loop_ctx *ctx = conn->ctx;
 | 
			
		||||
 | 
			
		||||
  const char *key;
 | 
			
		||||
  size_t key_len = lnm_http_req_route_segment(&key, &ctx->req, "key") ;
 | 
			
		||||
  lnm_linfo("main", "yuhh key: %.*s", key_len, key);
 | 
			
		||||
  return lnm_http_step_err_done;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
lnm_http_step_err print_step3(lnm_http_conn *conn) {
 | 
			
		||||
  lnm_http_loop_ctx *ctx = conn->ctx;
 | 
			
		||||
 | 
			
		||||
  const char *key;
 | 
			
		||||
  size_t key_len = lnm_http_req_route_segment(&key, &ctx->req, "cool") ;
 | 
			
		||||
  lnm_linfo("main", "cool: %.*s", key_len, key);
 | 
			
		||||
  return lnm_http_step_err_done;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
int main() {
 | 
			
		||||
  lnm_http_loop *hl;
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -34,8 +53,13 @@ int main() {
 | 
			
		|||
  lnm_http_router_init(&router);
 | 
			
		||||
 | 
			
		||||
  lnm_http_route *route;
 | 
			
		||||
  lnm_http_router_add(&route, router, lnm_http_method_get, "/emma");
 | 
			
		||||
  lnm_http_router_add(&route, router, lnm_http_method_get, "/:key");
 | 
			
		||||
  lnm_http_route_step_append(route, print_step, false);
 | 
			
		||||
  lnm_http_router_add(&route, router, lnm_http_method_get, "/:key/two");
 | 
			
		||||
  lnm_http_route_step_append(route, print_step2, false);
 | 
			
		||||
  lnm_http_router_add(&route, router, lnm_http_method_get, "/*cool");
 | 
			
		||||
  lnm_http_route_step_append(route, print_step3, false);
 | 
			
		||||
 | 
			
		||||
  lnm_http_loop_router_set(hl, router);
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -5,19 +5,25 @@
 | 
			
		|||
 | 
			
		||||
#include "lnm/http/loop.h"
 | 
			
		||||
 | 
			
		||||
struct lnm_http_router {
 | 
			
		||||
  struct lnm_http_router *exact_children[128];
 | 
			
		||||
  struct lnm_http_router *single_segment_child;
 | 
			
		||||
  lnm_http_route *routes[lnm_http_method_total];
 | 
			
		||||
  bool represents_route;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
typedef struct lnm_http_route_segment_trie {
 | 
			
		||||
  struct lnm_http_route_segment_trie *children[128];
 | 
			
		||||
  size_t index;
 | 
			
		||||
  bool represents_segment;
 | 
			
		||||
} lnm_http_route_segment_trie;
 | 
			
		||||
 | 
			
		||||
struct lnm_http_route {
 | 
			
		||||
  lnm_http_route_segment_trie *key_segments;
 | 
			
		||||
  lnm_http_step *step;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
struct lnm_http_router {
 | 
			
		||||
  struct lnm_http_router *exact_children[128];
 | 
			
		||||
  struct lnm_http_router *single_segment_child;
 | 
			
		||||
  lnm_http_route *multi_segment_routes[lnm_http_method_total];
 | 
			
		||||
  lnm_http_route *routes[lnm_http_method_total];
 | 
			
		||||
  bool represents_route;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
lnm_err lnm_http_route_segment_trie_init(lnm_http_route_segment_trie **out);
 | 
			
		||||
 | 
			
		||||
void lnm_http_route_segment_trie_free(lnm_http_route_segment_trie *trie);
 | 
			
		||||
| 
						 | 
				
			
			@ -39,11 +45,6 @@ typedef struct lnm_http_step {
 | 
			
		|||
 */
 | 
			
		||||
lnm_err lnm_http_step_init(lnm_http_step **out);
 | 
			
		||||
 | 
			
		||||
struct lnm_http_route {
 | 
			
		||||
  lnm_http_route_segment_trie *key_segments;
 | 
			
		||||
  lnm_http_step *step;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
lnm_err lnm_http_route_init(lnm_http_route **out);
 | 
			
		||||
 | 
			
		||||
void lnm_http_route_free(lnm_http_route *route);
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -93,9 +93,42 @@ lnm_err lnm_http_router_add(lnm_http_route **out, lnm_http_router *http_router,
 | 
			
		|||
      http_router = http_router->single_segment_child;
 | 
			
		||||
      path = new_path;
 | 
			
		||||
    } break;
 | 
			
		||||
    case '*':
 | 
			
		||||
      // TODO multi-segment wildcard
 | 
			
		||||
      break;
 | 
			
		||||
    case '*': {
 | 
			
		||||
      const char *next_slash_ptr = strchr(path + 1, '/');
 | 
			
		||||
 | 
			
		||||
      // Star match should be at end of route
 | 
			
		||||
      if (next_slash_ptr != NULL) {
 | 
			
		||||
        res = lnm_err_invalid_route;
 | 
			
		||||
 | 
			
		||||
        goto end;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      const char *end = strchr(path + 1, '\0');
 | 
			
		||||
      size_t key_len = end - (path + 1);
 | 
			
		||||
 | 
			
		||||
      if (key_len == 0) {
 | 
			
		||||
        res = lnm_err_invalid_route;
 | 
			
		||||
 | 
			
		||||
        goto end;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      res = lnm_http_route_key_segment_insert(route, path + 1, key_len,
 | 
			
		||||
                                              key_segments_count);
 | 
			
		||||
 | 
			
		||||
      if (res != lnm_err_ok) {
 | 
			
		||||
        goto end;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      key_segments_count++;
 | 
			
		||||
 | 
			
		||||
      if (http_router->multi_segment_routes[method] != NULL) {
 | 
			
		||||
        res = lnm_err_overlapping_route;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      http_router->multi_segment_routes[method] = route;
 | 
			
		||||
      goto end;
 | 
			
		||||
 | 
			
		||||
    } break;
 | 
			
		||||
    default:
 | 
			
		||||
      if (http_router->exact_children[c] == NULL) {
 | 
			
		||||
        res = lnm_http_router_init(&http_router->exact_children[c]);
 | 
			
		||||
| 
						 | 
				
			
			@ -107,6 +140,7 @@ lnm_err lnm_http_router_add(lnm_http_route **out, lnm_http_router *http_router,
 | 
			
		|||
 | 
			
		||||
      http_router = http_router->exact_children[c];
 | 
			
		||||
      path++;
 | 
			
		||||
 | 
			
		||||
      break;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
| 
						 | 
				
			
			@ -195,6 +229,26 @@ static lnm_http_route_err __lnm_http_router_route(lnm_http_route_match *out,
 | 
			
		|||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  const lnm_http_route *multi_segment_route =
 | 
			
		||||
      router->multi_segment_routes[method];
 | 
			
		||||
 | 
			
		||||
  if (multi_segment_route != NULL) {
 | 
			
		||||
    const char *end_ptr = strchr(path + path_index, '\0');
 | 
			
		||||
    size_t segment_len = end_ptr - (path + path_index);
 | 
			
		||||
 | 
			
		||||
    // TODO do 405's make sense for star matches?
 | 
			
		||||
 | 
			
		||||
    if (segment_len > 0) {
 | 
			
		||||
      if (out != NULL) {
 | 
			
		||||
        out->route = multi_segment_route;
 | 
			
		||||
        out->key_segments[matched_key_segments].start = path_index;
 | 
			
		||||
        out->key_segments[matched_key_segments].len = segment_len;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      return lnm_http_route_err_match;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  return res;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue