malei05 3 weeks ago
parent
commit
c2bdf4a363

+ 5 - 3
fn-opensearch/src/main/java/net/yaoyi/pipeline/opensearch/OpenSearchHandler.java

@@ -19,14 +19,16 @@ import java.util.function.Function;
 import java.util.stream.Collectors;
 
 @Slf4j
-public class OpenSearchHandler implements PojoRequestHandler<TaskImageMsg, TaskImageResultMsg> {
+public class OpenSearchHandler implements PojoRequestHandler<String, TaskImageResultMsg> {
 
     private static final int TOP_K = 10;
     private static final float SCORE_THRESHOLD = 0.7f;
 
     @Override
-    public TaskImageResultMsg handleRequest(TaskImageMsg taskImageMsg, Context context) {
-        log.info("收到请求消息:{}", JacksonUtils.toJson(taskImageMsg));
+    public TaskImageResultMsg handleRequest(String jsonString, Context context) {
+        log.info("收到请求消息:{}", jsonString);
+
+        TaskImageMsg taskImageMsg = JacksonUtils.parseObject(jsonString, TaskImageMsg.class);
         Long taskId = taskImageMsg.getTaskId();
         String namespace = taskImageMsg.getNamespace();