com.zuoxiaolong.blog.web.controller.AbstractWebController.java Source code

Introduction

Here is the source code for com.zuoxiaolong.blog.web.controller.AbstractWebController.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 com.zuoxiaolong.blog.web.controller;

import com.zuoxiaolong.blog.common.authorization.AuthorizationException;
import com.zuoxiaolong.blog.common.bean.Attachment;
import com.zuoxiaolong.blog.common.bean.JsonResponse;
import com.zuoxiaolong.blog.common.utils.JsonUtils;
import com.zuoxiaolong.blog.common.utils.ObjectUtils;
import com.zuoxiaolong.blog.common.web.AbstractController;
import com.zuoxiaolong.blog.model.dto.WebUserDTO;
import com.zuoxiaolong.blog.sdk.Api;
import com.zuoxiaolong.blog.sdk.BlogSdk;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.ModelAttribute;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Map;

/**
 * @author Boren You
 * @dateTime 2016/6/11 0:23
 * @since 1.0.0
 */
public abstract class AbstractWebController extends AbstractController {

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

    protected static final String TOKEN_ATTRIBUTE_NAME = "token";

    protected static final String USERNAME_ATTRIBUTE_NAME = "username";

    protected static final String NICKNAME_ATTRIBUTE_NAME = "nickname";

    @Autowired
    private BlogSdk blogSdk;

    protected JsonResponse invokeApi(Api api) {
        JsonResponse jsonResponse = blogSdk.invokeApi(getToken(), api);
        checkJsonResponse(jsonResponse);
        return jsonResponse;
    }

    protected JsonResponse invokeApi(Api api, Object params) {
        JsonResponse jsonResponse = blogSdk.invokeApi(getToken(), api, ObjectUtils.objectToMap(params));
        checkJsonResponse(jsonResponse);
        return jsonResponse;
    }

    protected JsonResponse invokeApi(Api api, Map<String, String> params) {
        JsonResponse jsonResponse = blogSdk.invokeApi(getToken(), api, params);
        checkJsonResponse(jsonResponse);
        return jsonResponse;
    }

    protected JsonResponse invokeApi(Api api, String attachmentKey, Attachment[] attachments) {
        JsonResponse jsonResponse = blogSdk.invokeApi(getToken(), api, attachmentKey, attachments);
        checkJsonResponse(jsonResponse);
        return jsonResponse;
    }

    protected JsonResponse invokeApi(Api api, Map<String, String> params, String attachmentKey,
            Attachment[] attachments) {
        JsonResponse jsonResponse = blogSdk.invokeApi(getToken(), api, params, attachmentKey, attachments);
        checkJsonResponse(jsonResponse);
        return jsonResponse;
    }

    private void checkJsonResponse(JsonResponse jsonResponse) {
        if (jsonResponse.authorizationError()) {
            throw new AuthorizationException();
        }
    }

    /**
     * response
     *
     * @param response
     */
    @ModelAttribute
    protected void setThreadLocal(HttpServletResponse response) {
        httpServletResponseThreadLocal.set(response);
    }

    /**
     * ???session
     *
     * @param token
     */
    protected void loginSuccess(String token) {
        setSessionAttribute(TOKEN_ATTRIBUTE_NAME, token);
    }

    protected void afterLoginSuccess(WebUserDTO loginWebUser) {
        setSessionAttribute(USERNAME_ATTRIBUTE_NAME, loginWebUser.getUsername());
        setSessionAttribute(NICKNAME_ATTRIBUTE_NAME, loginWebUser.getNickname());
    }

    /**
     * ???session
     */
    protected void logoutSuccess() {
        setSessionAttribute(USERNAME_ATTRIBUTE_NAME, null);
        setSessionAttribute(NICKNAME_ATTRIBUTE_NAME, null);
        setSessionAttribute(TOKEN_ATTRIBUTE_NAME, null);
    }

    /**
     * ?username
     *
     * @return
     */
    protected String getUsername() {
        return (String) getSessionAttribute(USERNAME_ATTRIBUTE_NAME);
    }

    /**
     * ?token
     *
     * @return
     */
    protected String getToken() {
        return (String) getSessionAttribute(TOKEN_ATTRIBUTE_NAME);
    }

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

    /**
     * JSON
     *
     * @param object
     * @return
     */
    protected void renderJson(Object object) {
        renderText(JsonUtils.toJson(object), "application/json");
    }

    /**
     * 
     *
     * @param string
     * @return
     */
    protected void renderText(String string, String type) {
        try {
            HttpServletResponse response = getResponse();
            response.reset();
            response.setContentType(type);
            response.setCharacterEncoding("utf-8");
            response.getWriter().print(string);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

}