com.litongjava.tio.boot.websocket.DefaultWebSocketRouter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tio-boot Show documentation

Java High Performance Web Development Framework

There is a newer version: 1.8.6

package com.litongjava.tio.boot.websocket;

import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import com.litongjava.tio.websocket.server.handler.IWebSocketHandler;

public class DefaultWebSocketRouter implements WebSocketRouter {
  Map routes = new ConcurrentHashMap<>();

  public void add(String path, IWebSocketHandler wsHandler) {
    routes.put(path, wsHandler);
  }

  /**
   * /* 表示匹配任何以特定路径开始的路径，/** 表示匹配该路径及其下的任何子路径
   * @param path
   * @return
   */
  public IWebSocketHandler find(String path) {
    // Direct match
    if (routes.containsKey(path)) {
      return routes.get(path);
    }

    // Check for wildcard matches
    Set> entrySet = routes.entrySet();

    for (Map.Entry entry : entrySet) {
      String key = entry.getKey();

      if (key.endsWith("/*")) {
        String baseRoute = key.substring(0, key.length() - 1);
        if (path.startsWith(baseRoute)) {
          return entry.getValue();
        }
      } else if (key.endsWith("/**")) {
        String baseRoute = key.substring(0, key.length() - 2);
        if (path.startsWith(baseRoute)) {
          return entry.getValue();
        }
      }
    }

    return null;
  }

  @Override
  public Map all() {
    return routes;
  }
}