vip.justlive.oxygen.web.Context Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oxygen-web Show documentation
web support of light framework
There is a newer version: 3.0.9
/*
 * Copyright (C) 2020 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */

package vip.justlive.oxygen.web;

import java.lang.reflect.Parameter;
import java.util.LinkedList;
import java.util.List;
import lombok.experimental.UtilityClass;
import vip.justlive.oxygen.core.bean.Singleton;
import vip.justlive.oxygen.core.exception.Exceptions;
import vip.justlive.oxygen.core.util.base.MoreObjects;
import vip.justlive.oxygen.web.bind.DataBinder;
import vip.justlive.oxygen.web.bind.ParamBinder;
import vip.justlive.oxygen.web.exception.ExceptionHandler;
import vip.justlive.oxygen.web.exception.ExceptionHandlerImpl;
import vip.justlive.oxygen.web.hook.WebHook;
import vip.justlive.oxygen.web.http.Parser;
import vip.justlive.oxygen.web.http.Request;
import vip.justlive.oxygen.web.http.Response;
import vip.justlive.oxygen.web.http.Session;
import vip.justlive.oxygen.web.http.SessionManager;
import vip.justlive.oxygen.web.result.ResultHandler;
import vip.justlive.oxygen.web.router.RouteHandler;
import vip.justlive.oxygen.web.router.RoutingContext;

/**
 * context
 *
 * @author wubo
 */
@UtilityClass
public class Context {

  public final SessionManager SESSION_MANAGER = new SessionManager();
  final List HANDLERS = new LinkedList<>();
  final List BINDERS = new LinkedList<>();
  final List HOOKS = new LinkedList<>();
  final List PARSERS = new LinkedList<>();
  final ExceptionHandler HANDLER = new ExceptionHandlerImpl();


  /**
   * 绑定参数
   *
   * @param parameter 参数
   * @return 数据绑定
   */
  public DataBinder bind(Parameter parameter) {
    for (ParamBinder binder : Context.BINDERS) {
      if (binder.supported(parameter)) {
        return binder.build(parameter);
      }
    }
    return null;
  }

  /**
   * dispatch
   *
   * @param ctx 上下文
   */
  public void dispatch(RoutingContext ctx) {
    try {
      Context.parseRequest(ctx.request());
      RouteHandler handler = ctx.request().getRouteHandler();
      if (handler == null) {
        routeNotFound(ctx);
        return;
      }
      if (!Context.invokeBefore(ctx)) {
        return;
      }
      handler.handle(ctx);
      Context.invokeAfter(ctx);
    } catch (Exception e) {
      routeException(ctx, e);
    } finally {
      Context.invokeFinished(ctx);
      Context.restoreSession(ctx.request(), ctx.response());
      Context.handleResult(ctx);
    }
  }

  public void routeNotFound(RoutingContext ctx) {
    MoreObjects.firstNonNull(Singleton.get(ExceptionHandler.class), HANDLER)
        .handle(ctx, Exceptions.fail("No handle found"), 404);
  }

  public void routeException(RoutingContext ctx, Exception e) {
    MoreObjects.firstNonNull(Singleton.get(ExceptionHandler.class), HANDLER).handle(ctx, e, 500);
  }

  public void routeError(RoutingContext ctx, Exception e) {
    MoreObjects.firstNonNull(Singleton.get(ExceptionHandler.class), HANDLER).error(ctx, e);
  }

  /**
   * finally中执行
   *
   * @param ctx 上下文
   */
  private void invokeFinished(RoutingContext ctx) {
    for (int i = HOOKS.size() - 1; i >= 0; i--) {
      HOOKS.get(i).finished(ctx);
    }
  }

  /**
   * 执行后置处理
   *
   * @param ctx 上下文
   */
  private void invokeAfter(RoutingContext ctx) {
    for (int i = HOOKS.size() - 1; i >= 0; i--) {
      HOOKS.get(i).after(ctx);
    }
  }

  /**
   * 执行前置处理
   *
   * @param ctx 上下文
   * @return 返回false中断处理
   */
  private boolean invokeBefore(RoutingContext ctx) {
    for (WebHook webHook : HOOKS) {
      if (!webHook.before(ctx)) {
        return false;
      }
    }
    return true;
  }

  /**
   * 解析请求
   *
   * @param request 请求
   */
  private void parseRequest(Request request) {
    PARSERS.forEach(parser -> parser.parse(request));
  }

  /**
   * 重新保存session
   *
   * @param request 请求
   * @param response 响应
   */
  private void restoreSession(Request request, Response response) {
    if (request.getSession() != null && !request.getSession().getId()
        .equals(request.getCookieValue(Session.SESSION_COOKIE_KEY))) {
      response.setCookie(Session.SESSION_COOKIE_KEY, request.getSession().getId());
    }
    SESSION_MANAGER.restoreSession(request.getSession());
  }

  /**
   * 处理结果
   *
   * @param ctx 上下文
   */
  private void handleResult(RoutingContext ctx) {
    for (ResultHandler resultHandler : Context.HANDLERS) {
      if (resultHandler.support(ctx.response().getResult())) {
        resultHandler.apply(ctx, ctx.response().getResult());
        break;
      }
    }
  }

}