add: 数据扩增代码修改

taais-modules/taais-biz/src/main/java/com/taais/biz/service/impl/DataServiceImpl.java

@@ -33,6 +33,8 @@ import com.taais.common.json.utils.JsonUtils;
 import com.taais.common.orm.core.page.PageQuery;
 import com.taais.common.orm.core.service.impl.BaseServiceImpl;
 import com.taais.common.redis.utils.RedisUtils;
+import com.taais.system.domain.vo.SysDictDataVo;
+import com.taais.system.service.ISysDictDataService;
 import jakarta.annotation.Resource;
 import net.lingala.zip4j.model.FileHeader;
 import org.redisson.api.RedissonClient;
@@ -73,7 +75,6 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
     private static final String RAR = ".rar";
     private static final String TXT = ".txt";
     private static final String[] VALID_EXTENSIONS = {".jpg", ".jpeg", ".png", ".bmp"};
-    private static final String PYTHON_DATA_AMPLIFY_API = "http://127.0.0.1:11001/augment";
     private static final String RESULT_CODE = "status";
     private static final String RESULT_STATUS = "200";
     private static final String AMPLIFY = "/AMPLIFY/";
@@ -83,6 +84,9 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
     @Resource
     private IDataAmplificationTaskService dataAmplificationTaskService;
 
+    @Resource
+    private ISysDictDataService dictDataService;
+
     @Override
     public QueryWrapper query() {
         return super.query().from(DATA);
@@ -449,6 +453,8 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
     private CommonResult<Boolean> doAmplify(DataAmplifyDto dataAmplifyDto) {
         this.updateEmp(Long.valueOf(dataAmplifyDto.getId()));
         String[] split = dataAmplifyDto.getBatchNum().split(",");
+        SysDictDataVo sysDictDataVo = dictDataService.selectDictDataByTypeAndLabel("python_api_address", "python_expand_data");
+
         for (String batchNum : split) {
             QueryWrapper query = query();
             query.eq(Data::getBatchNum, batchNum);
@@ -497,7 +503,7 @@ public class DataServiceImpl extends BaseServiceImpl<DataMapper, Data> implement
                     System.out.println("logPath===>" + logPath);
                     bodyJson.put("logPath", logPath);
                     //实际请求接口，接口未提供，暂且注释
-                    String response = HttpRequest.post(PYTHON_DATA_AMPLIFY_API)
+                    String response = HttpRequest.post(sysDictDataVo.getDictValue())
                         .body(JsonUtils.toJsonString(bodyJson))
                         .execute().body();
 //                String response = "{\"status\":200,\"msg\":\"扩增成功\"}";