tjx
昨天 a53e6872c70ba7c6c870627007ece21df9bcedbc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package com.hk.NumericalCollection.service;
 
import com.fasterxml.jackson.databind.ObjectMapper;
import com.hk.NumericalCollection.config.DataAcquisitionConfiguration;
import com.hk.NumericalCollection.dto.ApiDataResult;
import com.hk.NumericalCollection.dto.ApiRequestBody;
import com.hk.NumericalCollection.dto.ApiResponse;
import okhttp3.*;
import org.springframework.stereotype.Service;
 
import java.io.IOException;
import java.util.concurrent.TimeUnit;
 
@Service
public class ApiService {
    private final OkHttpClient client;
    private final ObjectMapper objectMapper;
 
    public ApiService() {
        this.client = new OkHttpClient.Builder()
                .connectTimeout(90, TimeUnit.SECONDS) // Set connection timeout
                .readTimeout(90, TimeUnit.SECONDS)    // Set read timeout
                .build();
        this.objectMapper = new ObjectMapper();
    }
 
    public ApiResponse<?> getSig(String publicKey, String appId) throws IOException {
        MediaType mediaType = MediaType.parse("text/plain");
        RequestBody body = RequestBody.create(mediaType, "{\"publicKey\":\"" + publicKey + "\"}");
 
        Request request = new Request.Builder()
                .url(String.format(DataAcquisitionConfiguration.ASK_SIGN_URL, appId))
                .post(body)
                .addHeader("Content-Type", "text/plain")
                .addHeader("Accept", "*/*")
                .addHeader("Connection", "keep-alive")
                .build();
 
        try (Response response = client.newCall(request).execute()) {
            if (!response.isSuccessful()) throw new IOException("Unexpected code " + response);
            return objectMapper.readValue(response.body().string(), ApiResponse.class);
        }
    }
 
    public <T> ApiResponse<T> sendListRequest(ApiRequestBody requestBody, Class<T> responseType) throws IOException {
        MediaType mediaType = MediaType.parse("application/json");
        String jsonBody = objectMapper.writeValueAsString(requestBody);
        RequestBody body = RequestBody.create(mediaType, jsonBody);
 
        Request request = new Request.Builder()
                .url(String.format(DataAcquisitionConfiguration.ASK_DATA_URL, requestBody.getAppId()))
                .method("POST", body)
                .addHeader("Content-Type", "application/json")
                .addHeader("Accept", "*/*")
                .addHeader("Connection", "keep-alive")
                .build();
 
        try (Response response = client.newCall(request).execute()) {
            if (response.isSuccessful()) {
                return objectMapper.readValue(
                        response.body().string(),
                        objectMapper.getTypeFactory().constructParametricType(ApiResponse.class, responseType)
                );
            } else {
                return handleErrorResponse(response);
            }
        }
    }
 
    private <T> ApiResponse<T> handleErrorResponse(Response response) throws IOException {
        ApiResponse<T> errorResponse = new ApiResponse<>();
        errorResponse.setCode(String.valueOf(response.code()));
        errorResponse.setError(response.message());
        // Optionally set other fields like list or total
        return errorResponse;
    }
 
    public <T> ApiDataResult<T> sendDataRequest(ApiRequestBody requestBody, Class<T> responseType) throws IOException {
        MediaType mediaType = MediaType.parse("application/json");
        String jsonBody = objectMapper.writeValueAsString(requestBody);
        RequestBody body = RequestBody.create(mediaType, jsonBody);
 
        Request request = new Request.Builder()
                .url(String.format(DataAcquisitionConfiguration.ASK_DATA_URL, requestBody.getAppId()))
                .method("POST", body)
                .addHeader("Content-Type", "application/json")
                .addHeader("Accept", "*/*")
                .addHeader("Connection", "keep-alive")
                .build();
 
        try (Response response = client.newCall(request).execute()) {
            if (response.isSuccessful()) {
                return objectMapper.readValue(response.body().string(), objectMapper.getTypeFactory().constructParametricType(ApiDataResult.class, responseType));
            } else {
                return handleErrorDataResponse(response);
            }
        }
    }
 
    private <T> ApiDataResult<T> handleErrorDataResponse(Response response) throws IOException {
        ApiDataResult<T> errorResponse = new ApiDataResult<>();
        errorResponse.setCode(String.valueOf(response.code()));
        errorResponse.setError(response.message());
        // Optionally set other fields like list or total
        return errorResponse;
    }
}