package com.ruoyi.im.config;
|
|
import java.util.HashMap;
|
import java.util.Map;
|
import org.json.JSONObject;
|
|
public class DynamicRequestBodyBuilder {
|
private Map<String, Object> bodyData;
|
private Map<String, Object> configurationData;
|
|
public DynamicRequestBodyBuilder() {
|
bodyData = new HashMap<>();
|
configurationData = new HashMap<>();
|
}
|
|
/**
|
* 设置token字段
|
*/
|
public DynamicRequestBodyBuilder setToken(String token) {
|
if (token != null) {
|
bodyData.put("token", token);
|
}
|
return this;
|
}
|
|
/**
|
* 设置enabled配置字段
|
*/
|
public DynamicRequestBodyBuilder setEnabled(boolean enabled) {
|
configurationData.put("enabled", enabled);
|
return this;
|
}
|
|
/**
|
* 设置p2p_chat_banned配置字段
|
*/
|
public DynamicRequestBodyBuilder setP2pChatBanned(boolean banned) {
|
configurationData.put("p2p_chat_banned", banned);
|
return this;
|
}
|
|
/**
|
* 设置team_chat_banned配置字段
|
*/
|
public DynamicRequestBodyBuilder setTeamChatBanned(boolean banned) {
|
configurationData.put("team_chat_banned", banned);
|
return this;
|
}
|
|
/**
|
* 设置chatroom_chat_banned配置字段
|
*/
|
public DynamicRequestBodyBuilder setChatroomChatBanned(boolean banned) {
|
configurationData.put("chatroom_chat_banned", banned);
|
return this;
|
}
|
|
/**
|
* 设置qchat_chat_banned配置字段
|
*/
|
public DynamicRequestBodyBuilder setQchatChatBanned(boolean banned) {
|
configurationData.put("qchat_chat_banned", banned);
|
return this;
|
}
|
|
/**
|
* 设置push_enabled_when_desktop_online配置字段
|
*/
|
public DynamicRequestBodyBuilder setPushEnabledWhenDesktopOnline(boolean enabled) {
|
configurationData.put("push_enabled_when_desktop_online", enabled);
|
return this;
|
}
|
|
/**
|
* 设置need_kick字段
|
*/
|
public DynamicRequestBodyBuilder setNeedKick(boolean needKick) {
|
bodyData.put("need_kick", needKick);
|
return this;
|
}
|
|
/**
|
* 设置kick_notify_extension字段
|
*/
|
public DynamicRequestBodyBuilder setKickNotifyExtension(String extension) {
|
if (extension != null) {
|
bodyData.put("kick_notify_extension", extension);
|
}
|
return this;
|
}
|
|
/**
|
* 批量设置配置项
|
*/
|
public DynamicRequestBodyBuilder setConfiguration(Map<String, Object> config) {
|
if (config != null) {
|
configurationData.putAll(config);
|
}
|
return this;
|
}
|
|
/**
|
* 批量设置所有字段
|
*/
|
public DynamicRequestBodyBuilder setAllFields(Map<String, Object> fields) {
|
if (fields != null) {
|
// 处理顶级字段
|
if (fields.containsKey("token")) {
|
setToken((String) fields.get("token"));
|
}
|
if (fields.containsKey("need_kick")) {
|
setNeedKick((Boolean) fields.get("need_kick"));
|
}
|
if (fields.containsKey("kick_notify_extension")) {
|
setKickNotifyExtension((String) fields.get("kick_notify_extension"));
|
}
|
|
// 处理配置字段
|
if (fields.containsKey("configuration")) {
|
Map<String, Object> config = (Map<String, Object>) fields.get("configuration");
|
setConfiguration(config);
|
}
|
}
|
return this;
|
}
|
|
/**
|
* 构建JSON字符串
|
*/
|
public String build() {
|
// 只有在有配置数据时才添加configuration字段
|
if (!configurationData.isEmpty()) {
|
bodyData.put("configuration", new HashMap<>(configurationData));
|
}
|
|
JSONObject jsonObject = new JSONObject(bodyData);
|
return jsonObject.toString();
|
}
|
|
/**
|
* 获取Map形式的数据
|
*/
|
public Map<String, Object> getBodyData() {
|
Map<String, Object> result = new HashMap<>(bodyData);
|
if (!configurationData.isEmpty()) {
|
result.put("configuration", new HashMap<>(configurationData));
|
}
|
return result;
|
}
|
|
/**
|
* 清空所有已设置的数据
|
*/
|
public void clear() {
|
bodyData.clear();
|
configurationData.clear();
|
}
|
|
/**
|
* 使用示例
|
*/
|
public static void main(String[] args) {
|
// 示例1: 只设置部分字段
|
DynamicRequestBodyBuilder builder = new DynamicRequestBodyBuilder();
|
String partialBody = builder
|
.setToken("custom_token_123")
|
.setEnabled(true)
|
.build();
|
System.out.println("部分字段请求体: " + partialBody);
|
|
// 示例2: 使用Map批量设置
|
builder.clear();
|
Map<String, Object> config = new HashMap<>();
|
config.put("enabled", true);
|
config.put("p2p_chat_banned", true);
|
|
Map<String, Object> fields = new HashMap<>();
|
fields.put("token", "batch_token");
|
fields.put("need_kick", true);
|
fields.put("configuration", config);
|
|
String batchBody = builder.setAllFields(fields).build();
|
System.out.println("批量设置请求体: " + batchBody);
|
|
// 示例3: 完全不设置任何字段
|
builder.clear();
|
String emptyBody = builder.build();
|
System.out.println("空请求体: " + emptyBody);
|
}
|
}
|
