add bodyEntity
This commit is contained in:
@@ -1,40 +1,143 @@
|
||||
package com.imyeyu.network;
|
||||
|
||||
import com.imyeyu.utils.Encoder;
|
||||
import org.apache.hc.client5.http.entity.mime.MultipartEntityBuilder;
|
||||
import org.apache.hc.client5.http.fluent.Form;
|
||||
import org.apache.hc.core5.http.ContentType;
|
||||
import org.apache.hc.core5.http.HttpEntity;
|
||||
import org.apache.hc.core5.http.NameValuePair;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.io.File;
|
||||
import java.io.InputStream;
|
||||
import java.util.AbstractMap;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.function.BiConsumer;
|
||||
|
||||
/**
|
||||
* HashMap 构建 {@link org.apache.hc.client5.http.fluent.Form} 参数列表
|
||||
* 请求参数列表。
|
||||
* <p>
|
||||
* 该实现允许同名 key 重复出现,适用于 query 参数和表单参数场景。
|
||||
*
|
||||
* @author 夜雨
|
||||
* @since 2025-06-26 15:41
|
||||
*/
|
||||
public class ArgMap<K, V> extends HashMap<K, V> {
|
||||
public class ArgMap<K, V> {
|
||||
|
||||
private final List<Map.Entry<K, V>> entries = new ArrayList<>();
|
||||
|
||||
/**
|
||||
* 追加一个参数。
|
||||
*
|
||||
* @param key 参数名
|
||||
* @param value 参数值
|
||||
* @return 当前对象
|
||||
*/
|
||||
public ArgMap<K, V> put(K key, V value) {
|
||||
entries.add(new AbstractMap.SimpleImmutableEntry<>(key, value));
|
||||
return this;
|
||||
}
|
||||
|
||||
/**
|
||||
* 按插入顺序遍历全部参数。
|
||||
*
|
||||
* @param consumer 处理逻辑
|
||||
*/
|
||||
public void forEach(BiConsumer<? super K, ? super V> consumer) {
|
||||
entries.forEach(entry -> consumer.accept(entry.getKey(), entry.getValue()));
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回参数数量。
|
||||
*
|
||||
* @return 参数数量
|
||||
*/
|
||||
public int size() {
|
||||
return entries.size();
|
||||
}
|
||||
|
||||
/**
|
||||
* 判断参数列表是否为空。
|
||||
*
|
||||
* @return 是否为空
|
||||
*/
|
||||
public boolean isEmpty() {
|
||||
return entries.isEmpty();
|
||||
}
|
||||
|
||||
/**
|
||||
* 清空全部参数。
|
||||
*/
|
||||
public void clear() {
|
||||
entries.clear();
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回只读参数视图。
|
||||
*
|
||||
* @return 参数条目列表
|
||||
*/
|
||||
public List<Map.Entry<K, V>> entries() {
|
||||
return Collections.unmodifiableList(entries);
|
||||
}
|
||||
|
||||
/**
|
||||
* 转换为 multipart 实体。
|
||||
*
|
||||
* @return multipart 请求体
|
||||
*/
|
||||
public HttpEntity toEntity() {
|
||||
MultipartEntityBuilder builder = MultipartEntityBuilder.create();
|
||||
forEach((k, v) -> {
|
||||
String key = k.toString();
|
||||
switch (v) {
|
||||
case File file -> builder.addBinaryBody(key, file);
|
||||
case byte[] bytes -> builder.addBinaryBody(key, bytes);
|
||||
case InputStream stream -> builder.addBinaryBody(key, stream, ContentType.APPLICATION_OCTET_STREAM, key);
|
||||
default -> builder.addTextBody(key, v.toString());
|
||||
}
|
||||
});
|
||||
return builder.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 转换为表单参数列表。
|
||||
*
|
||||
* @return 表单参数列表
|
||||
*/
|
||||
public List<NameValuePair> toNameValuePair() {
|
||||
Form form = Form.form();
|
||||
for (Map.Entry<K, V> item : entrySet()) {
|
||||
form.add(item.getKey().toString(), item.getValue().toString());
|
||||
}
|
||||
forEach((k, v) -> form.add(k.toString(), v.toString()));
|
||||
return form.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 转换为 query string。
|
||||
*
|
||||
* @return query string
|
||||
*/
|
||||
public String toURL() {
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (Map.Entry<K, V> item : entrySet()) {
|
||||
sb.append(item.getKey().toString()).append('=').append(Encoder.urlArg(item.getValue().toString()));
|
||||
sb.append('&');
|
||||
if (isEmpty()) {
|
||||
return "";
|
||||
}
|
||||
StringBuilder sb = new StringBuilder();
|
||||
forEach((k, v) -> sb.append(k.toString()).append('=').append(Encoder.urlArg(v.toString())).append('&'));
|
||||
sb.deleteCharAt(sb.length() - 1);
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* 将参数拼接到指定 URL。
|
||||
*
|
||||
* @param url 原始 URL
|
||||
* @return 拼接后的 URL
|
||||
*/
|
||||
public String toURL(String url) {
|
||||
if (isEmpty()) {
|
||||
return url;
|
||||
}
|
||||
StringBuilder sb = new StringBuilder(url);
|
||||
if (url.contains("?")) {
|
||||
if (!url.endsWith("?") && !url.endsWith("&")) {
|
||||
@@ -42,29 +145,52 @@ public class ArgMap<K, V> extends HashMap<K, V> {
|
||||
}
|
||||
sb.append(toURL());
|
||||
return sb.toString();
|
||||
} else {
|
||||
return url + "?" + toURL();
|
||||
}
|
||||
return url + "?" + toURL();
|
||||
}
|
||||
|
||||
/**
|
||||
* 使用一组参数创建对象。
|
||||
*
|
||||
* @param key 参数名
|
||||
* @param value 参数值
|
||||
* @param <K> key 类型
|
||||
* @param <V> value 类型
|
||||
* @return 参数对象
|
||||
*/
|
||||
public static <K, V> ArgMap<K, V> of(K key, V value) {
|
||||
ArgMap<K, V> map = new ArgMap<>();
|
||||
map.put(key, value);
|
||||
return map;
|
||||
return new ArgMap<K, V>().put(key, value);
|
||||
}
|
||||
|
||||
/**
|
||||
* 使用两组参数创建对象。
|
||||
*
|
||||
* @param key1 参数名 1
|
||||
* @param value1 参数值 1
|
||||
* @param key2 参数名 2
|
||||
* @param value2 参数值 2
|
||||
* @param <K> key 类型
|
||||
* @param <V> value 类型
|
||||
* @return 参数对象
|
||||
*/
|
||||
public static <K, V> ArgMap<K, V> of(K key1, V value1, K key2, V value2) {
|
||||
ArgMap<K, V> map = new ArgMap<>();
|
||||
map.put(key1, value1);
|
||||
map.put(key2, value2);
|
||||
return map;
|
||||
return new ArgMap<K, V>().put(key1, value1).put(key2, value2);
|
||||
}
|
||||
|
||||
/**
|
||||
* 使用三组参数创建对象。
|
||||
*
|
||||
* @param key1 参数名 1
|
||||
* @param value1 参数值 1
|
||||
* @param key2 参数名 2
|
||||
* @param value2 参数值 2
|
||||
* @param key3 参数名 3
|
||||
* @param value3 参数值 3
|
||||
* @param <K> key 类型
|
||||
* @param <V> value 类型
|
||||
* @return 参数对象
|
||||
*/
|
||||
public static <K, V> ArgMap<K, V> of(K key1, V value1, K key2, V value2, K key3, V value3) {
|
||||
ArgMap<K, V> map = new ArgMap<>();
|
||||
map.put(key1, value1);
|
||||
map.put(key2, value2);
|
||||
map.put(key3, value3);
|
||||
return map;
|
||||
return new ArgMap<K, V>().put(key1, value1).put(key2, value2).put(key3, value3);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user