vip.justlive.oxygen.web.router.Route Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oxygen-web Show documentation
web support of light framework
There is a newer version: 3.0.9
/*
 * Copyright (C) 2020 the original author or authors.
 *
 * 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.
 */
package vip.justlive.oxygen.web.router;

import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;
import lombok.experimental.Accessors;
import vip.justlive.oxygen.core.util.net.http.HttpMethod;

/**
 * 路由
 *
 * @author wubo
 */
@Getter
@ToString
@Accessors(fluent = true)
public class Route {

  public static final String REGEX_PATH_VAR = "\\{\\w+}";
  public static final String REGEX_PATH_VAR_REPLACE = "([^/?]*)";
  private static final Pattern REGEX_PATH_GROUP = Pattern.compile("\\{(\\w+)[}]");
  /**
   * 方法
   */
  private final Set methods = new HashSet<>(1, 1f);
  private final List pathVars = new LinkedList<>();
  /**
   * 请求路径
   */
  private String path;
  /**
   * 是否为正则请求
   */
  private boolean regex;
  /**
   * 路由处理方式
   */
  @Setter
  private RouteHandler handler;
  @Setter
  private Route next;

  private Pattern pattern;

  Route() {
  }

  /**
   * 设置method
   *
   * @param httpMethod 请求方法
   * @return router
   */
  public Route method(HttpMethod httpMethod) {
    methods.add(httpMethod);
    return this;
  }

  /**
   * add path
   *
   * @param path request path
   * @return router
   */
  public Route path(String path) {
    Matcher matcher = REGEX_PATH_GROUP.matcher(path);
    int start = 0;
    if (matcher.find(start)) {
      do {
        pathVars.add(matcher.group(1));
        start = matcher.end();
      } while (matcher.find(start));
      path = path.replaceAll(REGEX_PATH_VAR, REGEX_PATH_VAR_REPLACE);
      this.regex = true;
      this.pattern = Pattern.compile(path);
    }
    this.path = path;
    return this;
  }

  /**
   * 是否匹配
   *
   * @param path 路径
   * @param method 方法
   * @return 匹配的route
   */
  Route match(String path, HttpMethod method) {
    Route route = this;
    while (route != null) {
      if (regex) {
        if (pattern.matcher(path).matches() && route.methods.contains(method)) {
          return route;
        }
      } else {
        if (route.path.equals(path) && route.methods.contains(method)) {
          return route;
        }
      }
      route = route.next;
    }
    return null;
  }
}