package com.yami.trading.huobi.websocket.service.huobi.signature;
|
|
import java.io.UnsupportedEncodingException;
|
import java.math.BigDecimal;
|
import java.net.URLEncoder;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.TreeMap;
|
|
import com.alibaba.fastjson.JSON;
|
import okhttp3.MediaType;
|
import okhttp3.RequestBody;
|
|
import com.yami.trading.huobi.websocket.exception.SDKException;
|
|
|
public class UrlParamsBuilder {
|
|
class ParamsMap {
|
|
final Map<String, String> map = new HashMap<>();
|
final Map<String, List> stringListMap = new HashMap<>();
|
|
void put(String name, String value) {
|
|
if (name == null || "".equals(name)) {
|
throw new SDKException(SDKException.RUNTIME_ERROR,
|
"[URL] Key can not be null");
|
}
|
if (value == null || "".equals(value)) {
|
return;
|
}
|
|
map.put(name, value);
|
}
|
|
void put(String name, Integer value) {
|
put(name, value != null ? Integer.toString(value) : null);
|
}
|
|
void put(String name, Date value, String format) {
|
SimpleDateFormat dateFormatter = new SimpleDateFormat(format);
|
put(name, value != null ? dateFormatter.format(value) : null);
|
}
|
|
void put(String name, Long value) {
|
put(name, value != null ? Long.toString(value) : null);
|
}
|
|
void put(String name, BigDecimal value) {
|
put(name, value != null ? value.toPlainString() : null);
|
}
|
|
}
|
|
private static final MediaType JSON_TYPE = MediaType.parse("application/json");
|
private final ParamsMap paramsMap = new ParamsMap();
|
private final ParamsMap postBodyMap = new ParamsMap();
|
private boolean postMode = false;
|
|
public static UrlParamsBuilder build() {
|
return new UrlParamsBuilder();
|
}
|
|
public UrlParamsBuilder() {
|
}
|
|
public UrlParamsBuilder setPostMode(boolean mode) {
|
postMode = mode;
|
return this;
|
}
|
|
public <T extends Enum> UrlParamsBuilder putToUrl(String name, T value) {
|
if (value != null) {
|
paramsMap.put(name, value.toString());
|
}
|
return this;
|
}
|
|
public UrlParamsBuilder putToUrl(String name, String value) {
|
paramsMap.put(name, value);
|
return this;
|
}
|
|
public UrlParamsBuilder putToUrl(String name, Date value, String format) {
|
paramsMap.put(name, value, format);
|
return this;
|
}
|
|
public UrlParamsBuilder putToUrl(String name, Integer value) {
|
paramsMap.put(name, value);
|
return this;
|
}
|
|
public UrlParamsBuilder putToUrl(String name, Long value) {
|
paramsMap.put(name, value);
|
return this;
|
}
|
|
public UrlParamsBuilder putToUrl(String name, BigDecimal value) {
|
paramsMap.put(name, value);
|
return this;
|
}
|
|
public UrlParamsBuilder putToPost(String name, String value) {
|
postBodyMap.put(name, value);
|
return this;
|
}
|
|
public <T extends Enum> UrlParamsBuilder putToPost(String name, T value) {
|
if (value != null) {
|
postBodyMap.put(name, value.toString());
|
}
|
return this;
|
}
|
|
public UrlParamsBuilder putToPost(String name, Date value, String format) {
|
postBodyMap.put(name, value, format);
|
return this;
|
}
|
|
public UrlParamsBuilder putToPost(String name, Integer value) {
|
postBodyMap.put(name, value);
|
return this;
|
}
|
|
public <T> UrlParamsBuilder putToPost(String name, List list) {
|
postBodyMap.stringListMap.put(name, list);
|
return this;
|
}
|
|
public UrlParamsBuilder putToPost(String name, Long value) {
|
postBodyMap.put(name, value);
|
return this;
|
}
|
|
public UrlParamsBuilder putToPost(String name, BigDecimal value) {
|
postBodyMap.put(name, value);
|
return this;
|
}
|
|
public String buildUrl() {
|
Map<String, String> map = new HashMap<>(paramsMap.map);
|
StringBuilder head = new StringBuilder("?");
|
return AppendUrl(map, head);
|
|
}
|
|
public String buildSignature() {
|
Map<String, String> map = new TreeMap<>(paramsMap.map);
|
StringBuilder head = new StringBuilder();
|
return AppendUrl(map, head);
|
|
}
|
|
private String AppendUrl(Map<String, String> map, StringBuilder stringBuilder) {
|
for (Map.Entry<String, String> entry : map.entrySet()) {
|
if (!("").equals(stringBuilder.toString())) {
|
stringBuilder.append("&");
|
}
|
stringBuilder.append(entry.getKey());
|
stringBuilder.append("=");
|
stringBuilder.append(urlEncode(entry.getValue()));
|
}
|
return stringBuilder.toString();
|
}
|
|
public RequestBody buildPostBody() {
|
if (postBodyMap.stringListMap.isEmpty()) {
|
if (postBodyMap.map.isEmpty()) {
|
return RequestBody.create(JSON_TYPE, "");
|
} else {
|
return RequestBody.create(JSON_TYPE, JSON.toJSONString(postBodyMap.map));
|
}
|
} else {
|
return RequestBody.create(JSON_TYPE, JSON.toJSONString(postBodyMap.stringListMap));
|
|
|
}
|
}
|
|
public boolean hasPostParam() {
|
return !postBodyMap.map.isEmpty() || postMode;
|
}
|
|
|
public String buildUrlToJsonString() {
|
return JSON.toJSONString(paramsMap.map);
|
}
|
|
/**
|
* 使用标准URL Encode编码。注意和JDK默认的不同，空格被编码为%20而不是+。
|
*
|
* @param s String字符串
|
* @return URL编码后的字符串
|
*/
|
private static String urlEncode(String s) {
|
try {
|
return URLEncoder.encode(s, "UTF-8").replaceAll("\\+", "%20");
|
} catch (UnsupportedEncodingException e) {
|
throw new SDKException(SDKException.RUNTIME_ERROR,
|
"[URL] UTF-8 encoding not supported!");
|
}
|
}
|
}
|
