001/* 002 * Copyright 2012-2017 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package org.springframework.boot.autoconfigure.websocket.servlet; 018 019import io.undertow.servlet.api.DeploymentInfo; 020import io.undertow.websockets.jsr.WebSocketDeploymentInfo; 021 022import org.springframework.boot.web.embedded.undertow.UndertowDeploymentInfoCustomizer; 023import org.springframework.boot.web.embedded.undertow.UndertowServletWebServerFactory; 024import org.springframework.boot.web.server.WebServerFactoryCustomizer; 025import org.springframework.core.Ordered; 026 027/** 028 * WebSocket customizer for {@link UndertowServletWebServerFactory}. 029 * 030 * @author Phillip Webb 031 * @since 2.0.0 032 */ 033public class UndertowWebSocketServletWebServerCustomizer 034 implements WebServerFactoryCustomizer<UndertowServletWebServerFactory>, Ordered { 035 036 @Override 037 public void customize(UndertowServletWebServerFactory factory) { 038 WebsocketDeploymentInfoCustomizer customizer = new WebsocketDeploymentInfoCustomizer(); 039 factory.addDeploymentInfoCustomizers(customizer); 040 } 041 042 @Override 043 public int getOrder() { 044 return 0; 045 } 046 047 private static class WebsocketDeploymentInfoCustomizer 048 implements UndertowDeploymentInfoCustomizer { 049 050 @Override 051 public void customize(DeploymentInfo deploymentInfo) { 052 WebSocketDeploymentInfo info = new WebSocketDeploymentInfo(); 053 deploymentInfo.addServletContextAttribute( 054 WebSocketDeploymentInfo.ATTRIBUTE_NAME, info); 055 } 056 057 } 058 059}