File size: 2,463 Bytes
8d4dde5
 
 
 
d356c9c
 
 
e22740d
 
 
 
 
d356c9c
32a0c60
d356c9c
8d4dde5
8c86632
40497bc
 
8c86632
8d4dde5
 
 
 
 
 
d356c9c
 
31be65d
d356c9c
 
 
32a0c60
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d356c9c
32a0c60
307c022
 
 
 
 
 
32a0c60
d356c9c
 
 
 
8d4dde5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package ai.giskard.learnspringwebsockets;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Collections;
import java.util.Enumeration;

@SpringBootApplication(exclude = {
        org.springframework.boot.autoconfigure.security.servlet.SecurityAutoConfiguration.class,
        org.springframework.boot.actuate.autoconfigure.security.servlet.ManagementWebSecurityAutoConfiguration.class
})
public class LearnSpringWebsocketsApplication {

    public static void main(String[] args) {
        SpringApplication.run(LearnSpringWebsocketsApplication.class, args);
    }


    @Component
    static class RequestWrapperFilter extends OncePerRequestFilter {
        @Override
        protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
                throws ServletException, IOException {

            final HttpServletRequestWrapper reqWrapper = new HttpServletRequestWrapper(request) {
                @Override
                public Enumeration<String> getHeaders(String name) {
                    if ("connection".equals(name) && "UPGRADE".equals(super.getHeaders(name).nextElement())) {
                        return Collections.enumeration(Collections.singleton("Upgrade"));
                    }
                    return super.getHeaders(name);
                }

                @Override
                public String getHeader(String name) {
                    if ("connection".equals(name) && "UPGRADE".equals(super.getHeader(name))) {
                        return "Upgrade";
                    }
                    return super.getHeader(name);
                }
            };

            System.out.println("HEADERS: " + request.getServletPath());
            request.getHeaderNames().asIterator().forEachRemaining(s -> {
                System.out.println("HEADER: " + s + " = " + request.getHeader(s) + " ---> " + reqWrapper.getHeader(s));
            });

            filterChain.doFilter(reqWrapper, response);
        }
    }


}