cn.cug.laboratory.web.BaseController.java Source code

Introduction

Here is the source code for cn.cug.laboratory.web.BaseController.java

Source

/*
 * Copyright 2002-2016 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 cn.cug.laboratory.web;

import cn.cug.laboratory.utils.JsonMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.ModelAttribute;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;

/**
 * @author Boren You
 * @date 2016/5/12 21:28
 * @since 1.0.0
 */
public abstract class BaseController {

    /**
     * 
     */
    protected Logger logger = LoggerFactory.getLogger(getClass());

    /**
     * ?HttpServletRequest
     */
    private static ThreadLocal<HttpServletRequest> httpServletRequestThreadLocal = new ThreadLocal<HttpServletRequest>();

    /**
     * ?HttpServletResponse
     */
    private static ThreadLocal<HttpServletResponse> httpServletResponseThreadLocal = new ThreadLocal<HttpServletResponse>();

    /**
     * ?Model
     */
    private static ThreadLocal<Model> modelThreadLocal = new ThreadLocal<Model>();

    /**
     * @ModelAttribute??
     * @param request
     * @param response
     * @param model
     */
    @ModelAttribute
    protected void setThreadLocal(HttpServletRequest request, HttpServletResponse response, Model model) {
        BaseController.httpServletRequestThreadLocal.set(request);
        BaseController.httpServletResponseThreadLocal.set(response);
        BaseController.modelThreadLocal.set(model);
    }

    /**
     * ??HttpServletRequest
     * @return
     */
    protected HttpServletRequest getHttpServletRequest() {
        return BaseController.httpServletRequestThreadLocal.get();
    }

    /**
     * ??HttpServletResponse
     * @return
     */
    protected HttpServletResponse getHttpServletResponse() {
        return BaseController.httpServletResponseThreadLocal.get();
    }

    /**
     * ??HttpSession
     * @return
     */
    protected HttpSession getHttpSession() {
        return getHttpServletRequest().getSession();
    }

    /**
     * ??Model
     * @return
     */
    protected Model getModel() {
        return BaseController.modelThreadLocal.get();
    }

    /**
     * ? Model 
     * @param name
     * @param value
     */
    protected void setModelAttribute(String name, Object value) {
        getModel().addAttribute(name, value);
    }

    /**
     * ? HttpServletRequest 
     * @param name
     * @param value
     */
    protected void setHttpServletRequestAttribute(String name, Object value) {
        HttpServletRequest request = this.getHttpServletRequest();
        request.setAttribute(name, value);
    }

    /**
     * ? HttpSession 
     * @param name
     * @param value
     */
    public void setHttpSessionAttribute(String name, Object value) {
        HttpSession session = this.getHttpSession();
        session.setAttribute(name, value);
    }

    /**
     *  HttpSession ?
     * @param name
     * @return
     */
    protected Object getHttpSessionAttribute(String name) {
        HttpSession session = this.getHttpSession();
        Object value = session.getAttribute(name);
        return value;
    }

    /**
     *  HttpServletRequest ?
     * @param name
     * @return
     */
    protected Object getHttpServletRequestAttribute(String name) {
        HttpServletRequest request = this.getHttpServletRequest();
        Object value = request.getAttribute(name);
        return value;
    }

    /**
     * JSON
     * @param response
     * @param object
     * @return
     */
    protected String renderString(HttpServletResponse response, Object object) {
        return renderString(response, JsonMapper.toJsonString(object), "application/json");
    }

    /**
     * 
     * @param response
     * @param string
     * @return
     */
    protected String renderString(HttpServletResponse response, String string, String type) {
        try {
            response.reset();
            response.setContentType(type);
            response.setCharacterEncoding("utf-8");
            response.getWriter().print(string);
            return null;
        } catch (IOException e) {
            return null;
        }
    }
}