Examples with WebApplicationContext cn.taketoday.web.WebApplicationContext used on opensource projects

Example 1 with WebApplicationContext

use of cn.taketoday.web.WebApplicationContext in project today-infrastructure by TAKETODAY.

the class CorsFilter method doFilterInternal.

@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws IOException, ServletException {
    RequestContext context = RequestContextHolder.get();
    if (context == null) {
        WebApplicationContext webApplicationContext = ServletUtils.findWebApplicationContext(request);
        context = new ServletRequestContext(webApplicationContext, request, response);
        RequestContextHolder.set(context);
    }
    try {
        CorsConfiguration corsConfiguration = this.configSource.getCorsConfiguration(context);
        if (!processor.process(corsConfiguration, context) || CorsUtils.isPreFlightRequest(context)) {
            return;
        }
        // handle next
        filterChain.doFilter(request, response);
    } finally {
        RequestContextHolder.remove();
    }
}

Example 2 with WebApplicationContext

use of cn.taketoday.web.WebApplicationContext in project today-infrastructure by TAKETODAY.

the class RootWacEarTests method verifyRootWacConfig.

@Test
void verifyRootWacConfig() {
    ApplicationContext parent = wac.getParent();
    assertThat(parent).isNotNull();
    boolean condition = parent instanceof WebApplicationContext;
    assertThat(condition).isFalse();
    assertThat(ear).isEqualTo("ear");
    assertThat(root).isEqualTo("root");
}

Example 3 with WebApplicationContext

use of cn.taketoday.web.WebApplicationContext in project today-framework by TAKETODAY.

the class HTTPServer method handleConnection.

/**
 * Handles communications for a single connection over the given streams.
 * Multiple subsequent transactions are handled on the connection,
 * until the streams are closed, an error occurs, or the request
 * contains a "Connection: close" header which explicitly requests
 * the connection be closed after the transaction ends.
 *
 * @throws IOException if an error occurs
 */
protected void handleConnection(Socket socket) throws IOException {
    // BufferedInputStream in = new BufferedInputStream(socket.getInputStream(), 4096);
    final ByteArrayOutputStream inString = new ByteArrayOutputStream();
    BufferedInputStream in = new BufferedInputStream(socket.getInputStream(), 4096) {

        @Override
        public synchronized int read() throws IOException {
            final int read = super.read();
            inString.write(read);
            return read;
        }

        @Override
        public synchronized int read(@NonNull byte[] b, int off, int len) throws IOException {
            final int read = super.read(b, off, len);
            inString.write(b, off, len);
            return read;
        }
    };
    final OutputStream socketOutputStream = socket.getOutputStream();
    BufferedOutputStream out = new BufferedOutputStream(socketOutputStream, 4096);
    // final ByteArrayOutputStream out = new ByteArrayOutputStream() {
    // @Override
    // public synchronized void write(byte[] b, int off, int len) {
    // super.write(b, off, len);
    // try {
    // socketOutputStream.write(b, off, len);
    // }
    // catch (IOException e) {
    // e.printStackTrace();
    // }
    // }
    // };
    HttpRequest req = null;
    HttpResponse resp = new HttpResponse(out);
    InetAddress localAddress = socket.getLocalAddress();
    // create request and response and handle transaction
    try {
        req = new HttpRequest(in, socket, config);
        resp.setClientCapabilities(req);
        if (preprocessTransaction(req, resp)) {
            String method = req.getMethod();
            if ("TRACE".equals(method)) {
                // default TRACE handler
                handleTrace(req, resp);
            } else {
                WebApplicationContext webApplicationContext = httpHandler.getWebApplicationContext();
                final LightRequestContext context = new LightRequestContext(webApplicationContext, req, resp, config, localAddress);
                httpHandler.dispatch(context);
                context.sendIfNotCommitted();
            }
        }
    } catch (Throwable t) {
        log.error("Catch throwable", t);
        // unhandled errors (not normal error responses like 404)
        if (req == null) {
            // error reading request
            if (t instanceof IOException && t.getMessage().contains("missing request line"))
                // we're not in the middle of a transaction - so just disconnect
                return;
            // about to close connection
            resp.getHeaders().add("Connection", "close");
            if (// e.g. SocketTimeoutException
            t instanceof InterruptedIOException)
                resp.sendError(HttpStatus.REQUEST_TIMEOUT, "Timeout waiting for client request");
            else
                resp.sendError(HttpStatus.BAD_REQUEST, "Invalid request: " + t.getMessage());
        } else if (!resp.committed()) {
            // if headers or status line were not already sent, we can send an error response
            // ignore whatever headers may have already been set
            resp = new HttpResponse(out);
            // about to close connection
            resp.getHeaders().add("Connection", "close");
            resp.sendError(HttpStatus.INTERNAL_SERVER_ERROR, "Error processing request: " + t.getMessage());
        }
        // proceed to close connection
        return;
    } finally {
        // System.out.println(out);
        // final String name = StandardCharsets.ISO_8859_1.name();
        // System.out.println(inString.toString(name));
        System.out.println(inString);
        // close response and flush output
        resp.close();
    }
}

Example 4 with WebApplicationContext

use of cn.taketoday.web.WebApplicationContext in project today-framework by TAKETODAY.

the class WebApplicationLoader method configureHandlerAdapter.

/**
 * Configure {@link HandlerAdapter}
 *
 * @param adapters {@link HandlerAdapter}s
 * @param mvcConfiguration {@link WebMvcConfiguration}
 */
protected void configureHandlerAdapter(List<HandlerAdapter> adapters, WebMvcConfiguration mvcConfiguration) {
    // 先看有的
    DispatcherHandler obtainDispatcher = obtainDispatcher();
    HandlerAdapter[] handlerAdapters = obtainDispatcher.getHandlerAdapters();
    if (handlerAdapters != null) {
        Collections.addAll(adapters, handlerAdapters);
    }
    // 添加默认的
    adapters.add(new RequestHandlerAdapter(Ordered.HIGHEST_PRECEDENCE));
    WebApplicationContext context = obtainApplicationContext();
    // ViewControllerHandlerRegistry must configured
    ViewControllerHandlerRegistry viewControllerRegistry = context.getBean(ViewControllerHandlerRegistry.class);
    if (viewControllerRegistry != null) {
        ViewControllerHandlerAdapter bean = context.getBean(ViewControllerHandlerAdapter.class);
        if (bean == null) {
            ViewControllerHandlerAdapter viewControllerHandlerAdapter = null;
            for (HandlerAdapter adapter : adapters) {
                if (adapter instanceof ViewControllerHandlerAdapter) {
                    viewControllerHandlerAdapter = (ViewControllerHandlerAdapter) adapter;
                    break;
                }
            }
            if (viewControllerHandlerAdapter == null) {
                adapters.add(new ViewControllerHandlerAdapter(Ordered.HIGHEST_PRECEDENCE));
            }
        }
    }
    // 排序
    sort(adapters);
    // apply request handler
    obtainDispatcher.setHandlerAdapters(adapters.toArray(new HandlerAdapter[adapters.size()]));
}

Example 5 with WebApplicationContext

use of cn.taketoday.web.WebApplicationContext in project today-framework by TAKETODAY.

the class WebApplicationLoader method configViewControllerHandlerRegistry.

/**
 * Initialize framework.
 *
 * @throws Throwable if any Throwable occurred
 */
protected ViewControllerHandlerRegistry configViewControllerHandlerRegistry(@Nullable ViewControllerHandlerRegistry registry) throws Throwable {
    // find the configure file
    log.info("TODAY WEB Framework Is Looking For ViewController Configuration File.");
    String webMvcConfigLocation = getWebMvcConfigLocation();
    if (StringUtils.isEmpty(webMvcConfigLocation)) {
        log.info("Configuration File does not exist.");
        return registry;
    }
    if (registry == null) {
        WebApplicationContext context = obtainApplicationContext();
        context.unwrap(BeanDefinitionRegistrar.class).registerBean(ViewControllerHandlerRegistry.DEFAULT_BEAN_NAME, ViewControllerHandlerRegistry.class);
        registry = context.getBean(ViewControllerHandlerRegistry.DEFAULT_BEAN_NAME, ViewControllerHandlerRegistry.class);
    }
    registry.configure(webMvcConfigLocation);
    return registry;
}