hutool/hutool-http/src/main/java/cn/hutool/http/MultipartOutputStream.java

package cn.hutool.http;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.io.IORuntimeException;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.io.resource.MultiResource;
import cn.hutool.core.io.resource.Resource;
import cn.hutool.core.io.resource.StringResource;
import cn.hutool.core.util.StrUtil;

import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.Charset;

/**
 * Multipart/form-data输出流封装<br>
 * 遵循RFC2388规范
 *
 * @author looly
 * @since 5.7.17
 */
public class MultipartOutputStream extends OutputStream {

	private static final String CONTENT_DISPOSITION_TEMPLATE = "Content-Disposition: form-data; name=\"{}\"\r\n";
	private static final String CONTENT_DISPOSITION_FILE_TEMPLATE = "Content-Disposition: form-data; name=\"{}\"; filename=\"{}\"\r\n";

	private static final String CONTENT_TYPE_FILE_TEMPLATE = "Content-Type: {}\r\n";

	private final OutputStream out;
	private final Charset charset;
	private final String boundary;

	private boolean isFinish;

	/**
	 * 构造，使用全局默认的边界字符串
	 *
	 * @param out     HTTP写出流
	 * @param charset 编码
	 */
	public MultipartOutputStream(OutputStream out, Charset charset) {
		this(out, charset, HttpGlobalConfig.getBoundary());
	}

	/**
	 * 构造
	 *
	 * @param out      HTTP写出流
	 * @param charset  编码
	 * @param boundary 边界符
	 * @since 5.7.17
	 */
	public MultipartOutputStream(OutputStream out, Charset charset, String boundary) {
		this.out = out;
		this.charset = charset;
		this.boundary = boundary;
	}

	/**
	 * 添加Multipart表单的数据项<br>
	 * <pre>
	 *     --分隔符(boundary)[换行]
	 *     Content-Disposition: form-data; name="参数名"[换行]
	 *     [换行]
	 *     参数值[换行]
	 * </pre>
	 * <p>
	 * 或者：
	 *
	 * <pre>
	 *     --分隔符(boundary)[换行]
	 *     Content-Disposition: form-data; name="表单名"; filename="文件名"[换行]
	 *     Content-Type: MIME类型[换行]
	 *     [换行]
	 *     文件的二进制内容[换行]
	 * </pre>
	 *
	 * @param formFieldName 表单名
	 * @param value         值，可以是普通值、资源（如文件等）
	 * @return this
	 * @throws IORuntimeException IO异常
	 */
	public MultipartOutputStream write(String formFieldName, Object value) throws IORuntimeException {
		// 多资源
		if (value instanceof MultiResource) {
			for (Resource subResource : (MultiResource) value) {
				write(formFieldName, subResource);
			}
			return this;
		}

		// --分隔符(boundary)[换行]
		beginPart();

		if (value instanceof Resource) {
			appendResource(formFieldName, (Resource) value);
		} else {
			appendResource(formFieldName,
					new StringResource(Convert.toStr(value), null, this.charset));
		}

		write(StrUtil.CRLF);
		return this;
	}

	@Override
	public void write(int b) throws IOException {
		this.out.write(b);
	}

	/**
	 * 上传表单结束
	 *
	 * @throws IORuntimeException IO异常
	 */
	public void finish() throws IORuntimeException {
		if (false == isFinish) {
			write(StrUtil.format("--{}--\r\n", boundary));
			this.isFinish = true;
		}
	}

	@Override
	public void close() {
		finish();
		IoUtil.close(this.out);
	}

	/**
	 * 添加Multipart表单的Resource数据项，支持包括{@link HttpResource}资源格式
	 *
	 * @param formFieldName 表单名
	 * @param resource      资源
	 * @throws IORuntimeException IO异常
	 */
	private void appendResource(String formFieldName, Resource resource) throws IORuntimeException {
		final String fileName = resource.getName();

		// Content-Disposition
		if (null == fileName) {
			// Content-Disposition: form-data; name="参数名"[换行]
			write(StrUtil.format(CONTENT_DISPOSITION_TEMPLATE, formFieldName));
		} else {
			// Content-Disposition: form-data; name="参数名"; filename="文件名"[换行]
			write(StrUtil.format(CONTENT_DISPOSITION_FILE_TEMPLATE, formFieldName, fileName));
		}

		// Content-Type
		if (resource instanceof HttpResource) {
			final String contentType = ((HttpResource) resource).getContentType();
			if (StrUtil.isNotBlank(contentType)) {
				// Content-Type: 类型[换行]
				write(StrUtil.format(CONTENT_TYPE_FILE_TEMPLATE, contentType));
			}
		} else if (StrUtil.isNotEmpty(fileName)) {
			// 根据name的扩展名指定互联网媒体类型，默认二进制流数据
			write(StrUtil.format(CONTENT_TYPE_FILE_TEMPLATE,
					HttpUtil.getMimeType(fileName, ContentType.OCTET_STREAM.getValue())));
		}

		// 内容
		write("\r\n");
		resource.writeTo(this);
	}

	/**
	 * part开始，写出:<br>
	 * <pre>
	 *     --分隔符(boundary)[换行]
	 * </pre>
	 */
	private void beginPart() {
		// --分隔符(boundary)[换行]
		write("--", boundary, StrUtil.CRLF);
	}

	/**
	 * 写出对象
	 *
	 * @param objs 写出的对象（转换为字符串）
	 */
	private void write(Object... objs) {
		IoUtil.write(this, this.charset, false, objs);
	}
}