Exemplo n.º 1
0
 @Override
 public void modifyHandshake(
     ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
   super.modifyHandshake(sec, request, response);
   HttpSession httpSession = (HttpSession) request.getHttpSession();
   sec.getUserProperties().put("httpSession", httpSession);
 }
  @Override
  public void modifyHandshake(
      ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {

    HttpSession httpSession = (HttpSession) request.getHttpSession();

    sec.getUserProperties().put("header", request.getHeaders().toString());
    sec.getUserProperties().put(HttpSession.class.getName(), httpSession);

    if (request.getHeaders().containsKey("origin")) {
      sec.getUserProperties().put("origin", request.getHeaders().get("origin").get(0));
    }
    if (request.getHeaders().containsKey("user-agent")) {
      sec.getUserProperties().put("user-agent", request.getHeaders().get("user-agent").get(0));
    }
  }
 @Override
 public void modifyHandshake(
     ServerEndpointConfig config, HandshakeRequest request, HandshakeResponse response) {
   HttpSession session = (HttpSession) request.getHttpSession();
   if (null != session) {
     config.getUserProperties().put("demo", 1L);
   }
 }
    @Override
    public void modifyHandshake(
        ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
      delegate.modifyHandshake(sec, request, response);

      // do not store null keys/values because Tomcat 8 uses ConcurrentMap for UserProperties

      Map<String, Object> userProperties = sec.getUserProperties();
      Object httpSession = request.getHttpSession();
      LOG.trace("httpSession: {}", httpSession);
      if (httpSession != null) {
        userProperties.put("session", httpSession);
      }

      Map<String, List<String>> headers = request.getHeaders();
      LOG.trace("headers: {}", headers);
      if (headers != null) {
        userProperties.put("headers", headers);
      }

      Map<String, List<String>> parameterMap = request.getParameterMap();
      LOG.trace("parameterMap: {}", parameterMap);
      if (parameterMap != null) {
        userProperties.put("parameterMap", parameterMap);
      }

      String queryString = request.getQueryString();
      LOG.trace("queryString: {}", queryString);
      if (queryString != null) {
        userProperties.put("queryString", queryString);
      }

      URI requestURI = request.getRequestURI();
      LOG.trace("requestURI: {}", requestURI);
      if (requestURI != null) {
        userProperties.put("requestURI", requestURI);
      }

      Principal userPrincipal = request.getUserPrincipal();
      LOG.trace("userPrincipal: {}", userPrincipal);
      if (userPrincipal != null) {
        userProperties.put("userPrincipal", userPrincipal);
      }
    }
  /**
   * Provides the equivalent of {@link #addEndpoint(ServerEndpointConfig)} for publishing plain old
   * java objects (POJOs) that have been annotated as WebSocket endpoints.
   *
   * @param pojo The annotated POJO
   */
  @Override
  public void addEndpoint(Class<?> pojo) throws DeploymentException {

    ServerEndpoint annotation = pojo.getAnnotation(ServerEndpoint.class);
    if (annotation == null) {
      throw new DeploymentException(
          sm.getString("serverContainer.missingAnnotation", pojo.getName()));
    }
    String path = annotation.value();

    // Validate encoders
    validateEncoders(annotation.encoders());

    // Method mapping
    PojoMethodMapping methodMapping = new PojoMethodMapping(pojo, annotation.decoders(), path);

    // ServerEndpointConfig
    ServerEndpointConfig sec;
    Class<? extends Configurator> configuratorClazz = annotation.configurator();
    Configurator configurator = null;
    if (!configuratorClazz.equals(Configurator.class)) {
      try {
        configurator = annotation.configurator().newInstance();
      } catch (InstantiationException | IllegalAccessException e) {
        throw new DeploymentException(
            sm.getString(
                "serverContainer.configuratorFail",
                annotation.configurator().getName(),
                pojo.getClass().getName()),
            e);
      }
    }
    sec =
        ServerEndpointConfig.Builder.create(pojo, path)
            .decoders(Arrays.asList(annotation.decoders()))
            .encoders(Arrays.asList(annotation.encoders()))
            .subprotocols(Arrays.asList(annotation.subprotocols()))
            .configurator(configurator)
            .build();
    sec.getUserProperties().put(PojoEndpointServer.POJO_METHOD_MAPPING_KEY, methodMapping);

    addEndpoint(sec);
  }
 @Override
 public void modifyHandshake(
     ServerEndpointConfig config, HandshakeRequest request, HandshakeResponse response) {
   HttpSession httpSession = (HttpSession) request.getHttpSession();
   config.getUserProperties().put(HttpSession.class.getName(), httpSession);
 }
 @Override
 public Map<String, Object> getUserProperties() {
   return delegate.getUserProperties();
 }
 @Override
 public void modifyHandshake(
     ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {
   super.modifyHandshake(sec, request, response);
   sec.getUserProperties().put("request-headers", request.getHeaders());
 }