io.magician.tcp.codec.impl.http.request.MagicianResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Magician Show documentation
There is a newer version: 2.0.7
package io.magician.tcp.codec.impl.http.request;

import io.magician.common.constant.CommonConstant;

import java.io.InputStream;
import java.net.URLEncoder;

/**
 * 响应管理
 */
public class MagicianResponse {

    private static String contentType = "Content-type";
    private static String contentDis = "Content-Disposition";

    private MagicianHttpExchange httpExchange;

    public MagicianResponse(MagicianHttpExchange httpExchange){
        this.httpExchange = httpExchange;
    }

    /**
     * 设置响应头
     * @param key
     * @param value
     */
    public MagicianResponse setResponseHeader(String key, String value){
        httpExchange.setResponseHeader(key,value);
        return this;
    }

    /**
     * 响应文本数据
     * @param code
     * @param data
     */
    public void sendText(int code, String data){
        setResponseHeader(contentType, "text/plain;charset=UTF-8");
        httpExchange.sendText(code, data);
    }

    /**
     * 响应html数据
     * @param code
     * @param data
     */
    public void sendHtml(int code, String data){
        setResponseHeader(contentType, "text/html;charset=UTF-8");
        httpExchange.sendText(code, data);
    }

    /**
     * 响应自定义格式的数据
     * 需要自己设置 content-type
     * @param code
     * @param data
     */
    public void sendData(int code, String data){
        httpExchange.sendText(code, data);
    }

    /**
     * 响应json数据
     * @param code
     * @param data
     */
    public void sendJson(int code, String data){
        setResponseHeader(contentType, "application/json;charset=UTF-8");
        httpExchange.sendText(code, data);
    }

    /**
     * 响应二进制
     * @param bytes
     * @throws Exception
     */
    public void sendStream(String fileName, byte[] bytes) throws Exception {
        setResponseHeader(contentType, "application/octet-stream");
        setResponseHeader(contentDis, "attachment; filename=" + URLEncoder.encode(fileName, CommonConstant.ENCODING));
        httpExchange.setResponseBody(bytes);
    }

    /**
     * 响应文件流
     * @param inputStream
     * @throws Exception
     */
    public void sendStream(String fileName, InputStream inputStream) throws Exception {
        setResponseHeader(contentType, "application/octet-stream");
        setResponseHeader(contentDis, "attachment; filename=" + URLEncoder.encode(fileName, CommonConstant.ENCODING));
        httpExchange.setResponseBody(inputStream);
    }
}