cloud.agileframework.mvc.base.AbstractResponseFormat Maven / Gradle / Ivy

Go to download
package cloud.agileframework.mvc.base;

import cloud.agileframework.common.constant.Constant;
import cloud.agileframework.mvc.param.AgileReturn;
import org.springframework.web.servlet.ModelAndView;

import java.util.LinkedHashMap;
import java.util.Map;

/**
 * @author 佟盟 on 2018/11/2
 */
public abstract class AbstractResponseFormat extends LinkedHashMap {
    private RETURN sourceHead;
    private Object sourceResult;

    /**
     * 构建返回数据
     *
     * @param head   头部信息
     * @param result 体
     * @return 格式化信息
     */
    public abstract Map buildResponseData(RETURN head, Object result);

    /**
     * 构建响应报文体
     *
     * @param head   头信息
     * @param result 体信息
     * @return 返回ModelAndView
     */
    public ModelAndView buildResponse(RETURN head, Object result) {
        if (head == null) {
            head = RETURN.SUCCESS;
        }
        sourceHead = head;
        sourceResult = result;
        ModelAndView modelAndView = new ModelAndView();
        if (result == null) {
            modelAndView.addAllObjects(buildResponseData(head, null));
            return modelAndView;
        }
        if (Map.class.isAssignableFrom(result.getClass())) {
            Map resultMap = ((Map) result);
            boolean isResult = ((Map) result).containsKey(Constant.ResponseAbout.RESULT);
            if (isResult) {
                modelAndView.addAllObjects(buildResponseData(head, resultMap.get(Constant.ResponseAbout.RESULT)));
                return modelAndView;
            }
        }
        modelAndView.addAllObjects(buildResponseData(head, result));
        return modelAndView;
    }

    /**
     * 根据报文模板数据转换为对应得Agile报文
     */
    public void initAgileReturn() {
        AgileReturn.init(sourceHead, sourceResult);
    }
}