com.github.joekerouac.common.tools.net.http.response.IHttpResponse Maven / Gradle / Ivy
The newest version!
// Generated by delombok at Fri Mar 14 11:41:38 CST 2025
/*
* Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE
* file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file
* to You under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
* specific language governing permissions and limitations under the License.
*/
package com.github.joekerouac.common.tools.net.http.response;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.Message;
import com.github.joekerouac.common.tools.constant.Const;
import com.github.joekerouac.common.tools.io.IOUtils;
import com.github.joekerouac.common.tools.io.InMemoryFile;
import com.github.joekerouac.common.tools.net.http.exception.NetException;
import com.github.joekerouac.common.tools.string.StringUtils;
/**
* HTTP响应，如果不使用必须关闭，如果调用过getresult方法可以不关闭
*
* PS:非线程安全，并发调用有可能会导致未知错误
*
* @since 1.0.0
* @author JoeKerouac
* @date 2022-10-14 14:37:00
*/
public class IHttpResponse {
@java.lang.SuppressWarnings("all")
@lombok.Generated
private static final com.github.joekerouac.common.tools.log.Logger LOGGER = com.github.joekerouac.common.tools.log.LoggerFactory.getLogger(IHttpResponse.class.getName());
/**
* 400错误码
*/
private static final int ERR400 = 400;
/**
* 请求状态
*/
private final int status;
/**
* 响应数据
*/
private final InMemoryFile body;
/**
* 请求中的异常，如果没有异常那么该值为空
*/
private final NetException exception;
/**
* 响应header
*/
private final List headers;
/**
* 响应编码
*/
private final String charset;
/**
* 数据长度
*/
private final int len;
/**
* 数据，第一次获取时初始化
*/
private volatile byte[] data;
public IHttpResponse(Message message) {
HttpResponse httpResponse = message.getHead();
this.status = httpResponse.getCode();
this.headers = Arrays.stream(httpResponse.getHeaders()).map(IHeader::new).collect(Collectors.toList());
InMemoryFile file = message.getBody();
this.len = file == null ? 0 : file.getLen();
this.body = file;
this.charset = Optional.ofNullable(file).map(InMemoryFile::getCharset).orElse(Const.DEFAULT_CHARSET).name();
if (status >= ERR400) {
byte[] data;
try {
data = file == null || file.getDataAsInputStream() == null ? new byte[0] : IOUtils.read(file.getDataAsInputStream(), file.getLen(), true);
} catch (IOException e) {
throw new RuntimeException(e);
}
String msg = new String(data, Charset.forName(this.charset));
this.exception = new NetException(msg);
} else {
this.exception = null;
}
}
/**
* 获取请求头
*
* @param name
* 请求头的名字
* @return 对应的值
*/
public List getHeader(String name) {
if (name == null) {
return Collections.emptyList();
}
return headers.stream().filter(header -> name.equals(header.getName())).collect(Collectors.toList());
}
/**
* 获取所有响应头
*
* @return 响应头
*/
public List getAllHeaders() {
return headers;
}
/**
* 获取网络异常
*
* @return 网络异常
*/
public RuntimeException getException() {
return exception;
}
/**
* 将结果转换为字符串（调用此方法后input流将会关闭）
*
* @return 请求结果的字符串
*/
public String getResult() {
return getResult(null);
}
/**
* 将结果转换为指定字符集字符串（调用此方法后input流将会关闭）
*
* @param charset
* 结果字符集
* @return 请求结果的字符串
*/
public String getResult(String charset) {
return getResult(charset, true);
}
/**
* 将结果转换为指定字符集字符串，调用此方法后input流将会关闭，无论是否发生异常（有可能因为字符集异常）
*
* @param charset
* 结果字符集
* @param force
* 是否强制使用指定字符集而忽略服务器响应字符集，true表示强制使用指定字符集，忽略服务器响应字符集
* @return 请求结果的字符串
*/
public String getResult(String charset, boolean force) {
String usedCharset = charset;
if (!force) {
usedCharset = getCharset();
}
if (StringUtils.isBlank(usedCharset)) {
usedCharset = getCharset();
}
return new String(getResultAsBinary(), Charset.forName(usedCharset));
}
/**
* 以流的形式获取响应
*
* @return 响应流
* @throws IOException
* IOException
*/
public InputStream getResultAsStream() throws IOException {
if (exception != null) {
throw exception;
}
if (data == null) {
return body.getDataAsInputStream();
} else {
return new ByteArrayInputStream(data);
}
}
/**
* 以byte数组的形式获取响应数据
*
* @return 响应数据的byte数组
*/
public synchronized byte[] getResultAsBinary() {
if (exception != null) {
throw exception;
}
if (data == null) {
try {
data = IOUtils.read(body.getDataAsInputStream(), len, true);
} catch (IOException e) {
throw new RuntimeException(e);
}
}
return this.data;
}
/**
* 获取响应编码字符集
*
* @return 服务器响应编码字符集
*/
public String getCharset() {
return charset;
}
/**
* 获取响应HTTP状态
*
* @return http状态码
*/
public int getStatus() {
return this.status;
}
/**
* 响应header
*/
@java.lang.SuppressWarnings("all")
@lombok.Generated
public List getHeaders() {
return this.headers;
}
/**
* 数据长度
*/
@java.lang.SuppressWarnings("all")
@lombok.Generated
public int getLen() {
return this.len;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy