PoseTransformation-初版

src/main/java/com/ai/da/python/PythonService.java

@@ -3799,4 +3799,62 @@ public class PythonService {
         throw new BusinessException("design.interface.exception");
     }
 
+    public Boolean poseTransformation(String productImage, int poseId, String taskId) {
+        OkHttpClient client = new OkHttpClient().newBuilder()
+                .connectTimeout(30, TimeUnit.SECONDS)
+                .pingInterval(5, TimeUnit.SECONDS)//websocket轮训间隔(单位:秒)
+                .readTimeout(60, TimeUnit.SECONDS)//读取超时(单位:秒)
+                .writeTimeout(60, TimeUnit.SECONDS)//写入超时(单位:秒)
+                .build();
+        MediaType mediaType = MediaType.parse("application/json");
+        Map<String, String> content = Maps.newHashMap();
+        content.put("image_url", productImage);
+        content.put("tasks_id", taskId);
+        content.put("pose_id", String.valueOf(poseId));
+        RequestBody body = RequestBody.create(mediaType, JSON.toJSONString(content));
+
+        log.info("poseTransformation 请求地址： {}", accessPythonIp + ":" + accessPythonPort + "/api/pose_transform");
+        Request request = new Request.Builder()
+                .url(accessPythonIp + ":" + accessPythonPort + "/api/pose_transform")
+                .method("POST", body)
+                .addHeader("Content-Type", "application/json")
+                .build();
+        Response response = null;
+        String bodyString;
+        try {
+            log.info("poseTransformation请求入参content###{}", JSON.toJSONString(content));
+            response = client.newCall(request).execute();
+        } catch (IOException ioException) {
+            log.error("PythonService##poseTransformation异常###{}", ExceptionUtil.getThrowableList(ioException));
+            throw new BusinessException(ioException.getMessage());
+        }
+
+        // 判断是否生成失败
+        if (Objects.isNull(response.body())) {
+            log.error("PythonService##poseTransformation异常###{}", "response or body is empty!");
+            throw new BusinessException("PythonService##poseTransformation异常###: response or body is empty!");
+        } else if (response.code() != HttpURLConnection.HTTP_OK) {
+            log.error("PythonService##poseTransformation异常###{}", "Response error!Response code ## " + response.code() + " ##");
+            throw new BusinessException("PythonService##poseTransformation异常### Response error!Response code ## " + response.code() + " ##");
+        } else {
+            try {
+                bodyString = response.body().string();
+            } catch (IOException e) {
+                log.error(e.getMessage());
+                throw new BusinessException(e.getMessage());
+            }
+        }
+        JSONObject jsonObject = JSON.parseObject(bodyString);
+        Boolean result = JSON.parseObject(JSON.toJSONString(response)).getBoolean("successful");
+
+        if (result && jsonObject.get("code").equals(200)) {
+            log.info("poseTransformation##responseObject###{}", jsonObject);
+            return Boolean.TRUE;
+        } else {
+            log.info("poseTransformation失败###{}", jsonObject);
+            log.info("poseTransformation Exception! Code : {}", jsonObject.get("code"));
+            return Boolean.FALSE;
+        }
+    }
+
 }