io.github.kongweiguang.ok.Res Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of OK_Java Show documentation
Java project for OK
The newest version!
package io.github.kongweiguang.ok;

import static java.util.Objects.isNull;
import static java.util.Objects.nonNull;

import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import io.github.kongweiguang.ok.core.Header;
import io.github.kongweiguang.ok.core.TypeRef;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Type;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.OpenOption;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.StringJoiner;
import kotlin.Pair;
import okhttp3.Headers;
import okhttp3.MediaType;
import okhttp3.Response;
import okhttp3.ResponseBody;

/**
 * http的响应
 *
 * @author kongweiguang
 */
public final class Res {

  private final Response res;
  private byte[] bt;

  private Res(final Response response) {
    this.res = response;
  }

  public static Res of(final Response response) {
    return new Res(response);
  }

  public Response res() {
    return res;
  }

  public int status() {
    return res().code();
  }

  public boolean isOk() {
    return res().isSuccessful();
  }

  public String header(String name) {
    return res().header(name);
  }

  public Map> headers() {
    final Headers headers = res().headers();

    final Map> fr = new LinkedHashMap<>(headers.size(), 1);

    for (final Pair hd : headers) {
      fr.computeIfAbsent(hd.getFirst(), k -> new ArrayList<>()).add(hd.getSecond());
    }

    return fr;
  }

  private MediaType mediaType() {
    final String header = header(Header.content_type.v());
    if (isNull(header)) {
      return null;
    }

    return MediaType.parse(header);
  }


  public String contentType() {
    final MediaType mediaType = mediaType();
    if (mediaType == null) {
      return null;
    }
    return mediaType.type() + "/" + mediaType.subtype();
  }

  public Charset charset() {
    final MediaType mediaType = mediaType();
    if (mediaType == null) {
      return null;
    }
    return mediaType.charset(null);
  }

  public String contentEncoding() {
    return header(Header.content_encoding.v());
  }

  public long contentLength() {
    final String cl = header(Header.content_length.v());
    if (isNull(cl)) {
      return -1;
    }

    long contentLength = Long.parseLong(cl);

    if (contentLength > 0 && (isChunked() || nonNull(contentEncoding()))) {
      //按照HTTP协议规范，在 Transfer-Encoding和Content-Encoding设置后 Content-Length 无效。
      contentLength = -1;
    }

    return contentLength;
  }

  public boolean isChunked() {
    final String transferEncoding = header(Header.transfer_encoding.v());
    return "Chunked".equalsIgnoreCase(transferEncoding);
  }

  public String getCookieStr() {
    return header(Header.set_cookie.v());
  }

  public long reqMillis() {
    return res().sentRequestAtMillis();
  }

  public byte[] bytes() {
    if (isNull(bt)) {
      try {
        final ResponseBody body = res.body();

        if (isNull(body)) {
          this.bt = new byte[0];
        } else {
          this.bt = body.bytes();
        }
      } catch (IOException e) {
        throw new RuntimeException(e);
      }
    }

    return bt;
  }

  public String str() {
    return str(StandardCharsets.UTF_8);
  }

  public String str(final Charset charset) {
    return new String(bytes(), charset);
  }

  public JSONObject jsonObj() {
    return JSON.parseObject(str());
  }

  public InputStream stream() {
    return new ByteArrayInputStream(bytes());
  }

  public  R obj(Class clazz) {
    return JSON.parseObject(bytes(), clazz);
  }

  public  R obj(Type type) {
    return JSON.parseObject(bytes(), type);
  }

  public Integer rInt() {
    return obj(Integer.class);
  }

  public Long rLong() {
    return obj(Long.class);
  }

  public Boolean rBool() {
    return obj(Boolean.class);
  }

  public  List list() {
    return obj(new TypeRef>() {
    }.type());
  }

  public  Map map() {
    return obj(new TypeRef>() {
    }.type());
  }

  public void file(String path, OpenOption... options) throws IOException {
    Files.write(Paths.get(path), bytes(), options);
  }


  @Override
  public String toString() {
    return new StringJoiner(", ", Res.class.getSimpleName() + "[", "]")
        .add("res=" + res)
        .toString();
  }

}