Skip to content

Commit 02570a7

Browse files
committed
Merge branch 'feat/ark/fix_request_id' into 'master'
Feat/ark/fix request See merge request iaasng/volcengine-java-sdk!210
2 parents da903a6 + 518ee9b commit 02570a7

File tree

2 files changed

+15
-8
lines changed

2 files changed

+15
-8
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/interceptor/RequestIdInterceptor.java

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,13 @@ public RequestIdInterceptor() {}
1717

1818
@Override
1919
public Response intercept(Chain chain) throws IOException {
20-
Request request = chain.request()
21-
.newBuilder()
22-
.header(Const.CLIENT_REQUEST_HEADER, genRequestId())
23-
.build();
20+
Request.Builder requestBuilder = chain.request().newBuilder();
21+
22+
if (chain.request().header(Const.CLIENT_REQUEST_HEADER) == null || chain.request().header(Const.CLIENT_REQUEST_HEADER).length() == 0) {
23+
requestBuilder = requestBuilder.header(Const.CLIENT_REQUEST_HEADER, genRequestId());
24+
}
25+
26+
Request request = requestBuilder.build();
2427
return chain.proceed(request);
2528
}
2629

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/utils/ResponseBodyCallback.java

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,10 +68,14 @@ public void onResponse(Call<ResponseBody> call, Response<ResponseBody> response)
6868
if (line.startsWith("data:")) {
6969
String data = line.substring(5).trim();
7070

71-
ArkAPIError err = mapper.readValue(data, ArkAPIError.class);
72-
if (err.getError() != null) {
73-
throw new ArkHttpException(err, null, -1, requestId);
74-
}
71+
try {
72+
ArkAPIError err = mapper.readValue(data, ArkAPIError.class);
73+
if (err.getError() != null) {
74+
throw new ArkHttpException(err, null, -1, requestId);
75+
}
76+
} catch (ArkHttpException e) {
77+
throw e;
78+
} catch (Exception ignored) {}
7579

7680
sse = new SSE(data);
7781
} else if (line.equals("") && sse != null) {

0 commit comments

Comments
 (0)