1 /*
2  * JBoss, Home of Professional Open Source.
3  * Copyright 2014 Red Hat, Inc., and individual contributors
4  * as indicated by the @author tags.
5  *
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  *  Unless required by applicable law or agreed to in writing, software
13  *  distributed under the License is distributed on an "AS IS" BASIS,
14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  *  See the License for the specific language governing permissions and
16  *  limitations under the License.
17  */
18
19 package io.undertow.servlet.core;
20
21 import io.undertow.server.HttpServerExchange;
22 import io.undertow.servlet.api.ThreadSetupHandler;
23 import io.undertow.servlet.handlers.ServletRequestContext;
24
25 /**
26  * @author Stuart Douglas
27  */
28 class ServletRequestContextThreadSetupAction implements ThreadSetupHandler {
29
30     static final ServletRequestContextThreadSetupAction INSTANCE = new ServletRequestContextThreadSetupAction();
31
32     private ServletRequestContextThreadSetupAction() {
33
34     }
35
36     @Override
37     public <T, C> Action<T, C> create(final Action<T, C> action) {
38         return new Action<T, C>() {
39             @Override
40             public T call(HttpServerExchange exchange, C context) throws Exception {
41                 if (exchange == null) {
42                     return action.call(null, context);
43                 } else {
44                     ServletRequestContext servletRequestContext = exchange.getAttachment(ServletRequestContext.ATTACHMENT_KEY);
45                     final ServletRequestContext old = ServletRequestContext.current();
46                     SecurityActions.setCurrentRequestContext(servletRequestContext);
47                     try {
48                         return action.call(exchange, context);
49                     } finally {
50                         ServletRequestContext.setCurrentRequestContext(old);
51                     }
52                 }
53             }
54         };
55     }
56 }
57