com.dingtalk.open.app.stream.protocol.ProtocolResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dingtalk-stream Show documentation
There is a newer version: 1.3.7
package com.dingtalk.open.app.stream.protocol;

import java.util.HashMap;
import java.util.Map;

/**
 * @author feiyin
 * @date 2022/12/28
 */
public class ProtocolResponse {
    private int code;
    private String message;

    private Map headers;
    private String data;


    public int getCode() {
        return code;
    }

    public void setCode(int code) {
        this.code = code;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public Map getHeaders() {
        return headers;
    }

    public void setHeaders(Map headers) {
        this.headers = headers;
    }

    public String getData() {
        return data;
    }

    public void setData(String data) {
        this.data = data;
    }

    public void setHeader(String headerName, String headerValue) {
        if (headerName == null || headerValue == null) {
            return;
        }
        if (headers == null) {
            this.headers = new HashMap<>();
        }
        this.headers.put(headerName, headerValue);
    }

    public static ProtocolResponse new200Response(ProtocolRequestFacade request) {
        ProtocolResponse response = base(request);
        response.setCode(200);
        response.setMessage("OK");
        response.setHeader(HeaderNames.MESSAGE_ID, request.getMessageId());
        response.setHeader(HeaderNames.CONTENT_TYPE, request.getContentType());
        response.setHeader(HeaderNames.TIME, String.valueOf(System.currentTimeMillis()));
        return response;
    }

    public static ProtocolResponse new500Response(ProtocolRequestFacade request) {
        ProtocolResponse response = base(request);
        response.setCode(500);
        response.setMessage("client internal error");
        return response;
    }


    public static ProtocolResponse newErrorResponse(ProtocolRequestFacade request, int code, String msg) {
        ProtocolResponse response = base(request);
        response.setCode(code);
        response.setMessage(msg);
        return response;
    }

    private static ProtocolResponse base(ProtocolRequestFacade request) {
        ProtocolResponse response = new ProtocolResponse();
        response.setHeader(HeaderNames.MESSAGE_ID, request.getMessageId());
        response.setHeader(HeaderNames.CONTENT_TYPE, ContentType.APPLICATION_JSON.getContentType());
        return response;
    }
}