org.smartboot.http.client.HttpPost Maven / Gradle / Ivy
/*******************************************************************************
* Copyright (c) 2017-2021, org.smartboot. All rights reserved.
* project name: smart-http
* file name: HttpPost.java
* Date: 2021-02-04
* Author: sandao ([email protected])
******************************************************************************/
package org.smartboot.http.client;
import org.smartboot.http.common.enums.HeaderNameEnum;
import org.smartboot.http.common.enums.HeaderValueEnum;
import org.smartboot.http.common.enums.HttpMethodEnum;
import org.smartboot.socket.transport.AioSession;
import java.net.URLEncoder;
import java.util.AbstractQueue;
import java.util.Iterator;
import java.util.Map;
import java.util.function.Consumer;
/**
* @author 三刀（[email protected]）
* @version V1.0 , 2021/2/4
*/
public final class HttpPost extends HttpRest {
HttpPost(AioSession session, AbstractQueue queue) {
super(session, queue);
request.setMethod(HttpMethodEnum.POST.getMethod());
}
@Override
public HttpRest setMethod(String method) throws UnsupportedOperationException {
throw new UnsupportedOperationException();
}
@Override
public PostBody body() {
return new PostBody(super.body(), this) {
@Override
public HttpPost formUrlencoded(Map params) {
if (params == null || params.isEmpty()) {
HttpPost.this.done();
return HttpPost.this;
}
try {
willSendRequest();
//编码Post表单
Iterator> iterator = params.entrySet().iterator();
Map.Entry entry = iterator.next();
StringBuilder sb = new StringBuilder();
sb.append(URLEncoder.encode(entry.getKey(), "utf8")).append("=").append(URLEncoder.encode(entry.getValue(), "utf8"));
while (iterator.hasNext()) {
entry = iterator.next();
sb.append("&").append(URLEncoder.encode(entry.getKey(), "utf8")).append("=").append(URLEncoder.encode(entry.getValue(), "utf8"));
}
byte[] bytes = sb.toString().getBytes();
// 设置 Header
request.setContentLength(bytes.length);
request.addHeader(HeaderNameEnum.CONTENT_TYPE.getName(), HeaderValueEnum.X_WWW_FORM_URLENCODED.getName());
//输出数据
request.write(bytes);
request.getOutputStream().flush();
} catch (Exception e) {
e.printStackTrace();
completableFuture.completeExceptionally(e);
}
return HttpPost.this;
}
};
}
@Override
public HttpPost onSuccess(Consumer consumer) {
super.onSuccess(consumer);
return this;
}
@Override
public HttpPost onFailure(Consumer consumer) {
super.onFailure(consumer);
return this;
}
@Override
public Header header() {
return new HeaderWrapper<>(this, super.header());
}
}