net.fastposter.client.FastposterCloudClient Maven / Gradle / Ivy
package net.fastposter.client;
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import lombok.Builder;
import lombok.Data;
import lombok.SneakyThrows;
import net.fastposter.client.utils.Base64Utils;
import net.fastposter.client.utils.DigestUtils;
import net.fastposter.client.utils.JsonUtils;
import net.fastposter.client.utils.RandomUtils;
import okhttp3.*;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;
/**
* fastposter云服务客户端
*
* @author Alex小新 [email protected]
* @site
*/
@Builder
public class FastposterCloudClient {
private static final String CLIENT_TYPE = "java";
private static final String CLIENT_VERSION = "1.2.5";
private static final String USER_AGENT = "fastposter-cloud-client/" + CLIENT_VERSION + " (" + CLIENT_TYPE + ")";
private final SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
@Builder.Default
private OkHttpClient okHttpClient;
@Builder.Default
private String endpoint = "https://api.fastposter.net";
private String appKey;
private String appSecret;
private boolean debug;
private boolean trace;
private static final OkHttpClient DEFAULT_OK_HTTP_CLIENT = new OkHttpClient.Builder().connectTimeout(10, TimeUnit.SECONDS).build();
@Builder.Default
private AtomicLong atomicLong = new AtomicLong(0);
// static {
// try {
// // 连接初始化，加速接口首次访问
// DEFAULT_OK_HTTP_CLIENT.newCall(new Request.Builder().url("https://api.fastposter.net/v1/init").build()).execute();
// } catch (Exception e) {
// throw new RuntimeException("客户端初始化异常", e);
// }
// }
@SneakyThrows
private T buildPoster(String uuid, Map params, PosterType type, Double scale, Boolean b64, String userAgent, boolean onlySign) {
long time = System.currentTimeMillis();
long seq = atomicLong.incrementAndGet();
String data = JsonUtils.toJson(params);
// base64编码
String playload = Base64Utils.encodeToUrlSafeString(data.getBytes());
// 计算签名
Long timestamp = System.currentTimeMillis() / 1000;
String nonce = RandomUtils.randomAlphanumeric(16);
String sign = DigestUtils.md5DigestAsHex((playload + timestamp + nonce + appSecret).getBytes());
// 构建请求参数
GetPosterArgs.GetPosterArgsBuilder build = GetPosterArgs.builder()
.uuid(uuid)
.b64(b64)
.scale(scale)
.timestamp(timestamp)
.nonce(nonce)
.appKey(appKey)
.sign(sign)
.payload(playload);
if (type != null) {
build.type(type.value);
}
GetPosterArgs args = build.build();
String url = this.endpoint + "/v1/build/poster";
MediaType mediaType = MediaType.parse("application/json");
String json = JsonUtils.toJson(args);
if (onlySign) {
return (T) args;
}
if (trace) {
System.out.println(SDF.format(new Date()) + " " + seq + " build poster request=" + json);
}
// 支持用户自己设置userAgent
userAgent = userAgent != null && userAgent.length() > 1 ? userAgent : USER_AGENT;
RequestBody body = RequestBody.create(json, mediaType);
Request request = new Request.Builder().url(url).post(body)
.addHeader("User-Agent", userAgent)
.addHeader("Client-Type", CLIENT_TYPE)
.addHeader("Client-Version", CLIENT_VERSION)
.addHeader("Content-Type", "application/json")
.addHeader("cache-control", "no-cache")
.build();
Response response = okHttpClient().newCall(request).execute();
String traceId = response.header("fastposter-cloud-traceid");
if (trace) {
response.headers().forEach(a -> System.out.println(a.getFirst() + ":" + a.getSecond()));
}
if (response.code() != 200) {
// System.err.println("code=" + response.code());
// System.out.println(request);
throw new FastpsoterException(seq, response.code(), " build poster error! \n" + response.body().string(), traceId);
}
String contentType = response.header("content-type");
if (contentType != null && contentType.contains("application/json")) {
assert response.body() != null;
PosterResponse pr = JsonUtils.parseObject(response.body().bytes(), PosterResponse.class);
if (pr.code != 0) {
throw new FastpsoterException(seq, pr.code, pr.getMsg(), traceId);
}
return null;
}
assert response.body() != null;
byte[] bytes = response.body().bytes();
long size = bytes.length;
if (type == null) {
// 获取服务器返回的图片类型
String rcontentType = response.header("Content-Type");
if (rcontentType != null && !Objects.equals(contentType, "")) {
if (rcontentType.contains("jpeg")) {
type = PosterType.jpeg;
} else if (rcontentType.contains("pdf")) {
type = PosterType.pdf;
} else if (rcontentType.contains("png")) {
type = PosterType.png;
} else if (rcontentType.contains("jpg")) {
type = PosterType.jpg;
} else {
type = PosterType.jpg;
}
} else {
type = PosterType.jpg;
}
}
if (debug) {
time = System.currentTimeMillis() - time;
System.out.println(SDF.format(new Date()) + " " + seq + " build poster completed. traceId=" + traceId + " uuid=" + uuid + " type=" + type + " " + (type.value.length() <= 3 ? " " : "") + "size=" + size + " rt=" + time + "ms" + (b64 != null && b64 ? " b64=" + b64 : "") + (scale != null ? " scale=" + scale : ""));
}
return (T) new Poster(traceId, type, bytes, b64);
}
private OkHttpClient okHttpClient() {
if (this.okHttpClient == null) {
this.okHttpClient = DEFAULT_OK_HTTP_CLIENT;
}
return okHttpClient;
}
public PosterBuilder buildPoster(String uuid) {
return PosterBuilder.builder().client(this).uuid(uuid).build();
}
protected Poster build(PosterBuilder posterBuilder) {
return buildPoster(posterBuilder.uuid, posterBuilder.params, posterBuilder.type, posterBuilder.scale, posterBuilder.b64, posterBuilder.userAgent, false);
}
/**
* 生成签名
*
* @param posterBuilder
* @return
*/
protected GetPosterArgs getArgs(PosterBuilder posterBuilder) {
return buildPoster(posterBuilder.uuid, posterBuilder.params, posterBuilder.type, posterBuilder.scale, posterBuilder.b64, posterBuilder.userAgent, true);
}
@Data
@JsonIgnoreProperties(ignoreUnknown = true)
static class PosterResponse {
int code;
String msg;
Object data;
String traceId;
}
}