优化代码

src/main/java/com/lqkj/link/module/zone/service/ZoneInfoService.java

@@ -176,9 +176,13 @@ public class ZoneInfoService {
                         String modelFileName = model.getName();
                         if (modelFileName.endsWith(".fbx")) {
                             String modelPath = aliOSSUtils.upload(model);
-                            list.add(new ModelInfo(null, null, modelFileName.substring(0, modelFileName.lastIndexOf(".")).toLowerCase(),
-                                    null, null, null, modelPath,
-                                    zoneInfo1.getThumbnail(), null, zoneInfo1.getZoneId(), 1));
+                            ModelInfo modelInfo = new ModelInfo();
+                            modelInfo.setModelName(modelFileName.substring(0, modelFileName.lastIndexOf(".")).toLowerCase());
+                            modelInfo.setOriginalPath(modelPath);
+                            modelInfo.setModelIcon(zoneInfo1.getThumbnail());
+                            modelInfo.setTemplateId(zoneInfo1.getZoneId());
+                            modelInfo.setTypeNumber(1);
+                            list.add(modelInfo);
                         }
                     }
                 }
@@ -192,9 +196,13 @@ public class ZoneInfoService {
                 zoneInfo1.setPakPath(pakPath);
                 zoneInfoRepository.save(zoneInfo1);
                 for (TemplateInfo templateInfo : templateInfoList) {
-                    list.add(new ModelInfo(null, null, templateInfo.getModelPath().substring(templateInfo.getModelPath().lastIndexOf(".") + 1),
-                            null, null, null, templateInfo.getModelPath(),
-                            zoneInfo1.getThumbnail(), null, zoneInfo1.getZoneId(), 2));
+                    ModelInfo modelInfo = new ModelInfo();
+                    modelInfo.setModelName(templateInfo.getModelPath().substring(templateInfo.getModelPath().lastIndexOf(".") + 1));
+                    modelInfo.setOriginalPath(templateInfo.getModelPath());
+                    modelInfo.setModelIcon(zoneInfo1.getThumbnail());
+                    modelInfo.setTemplateId(zoneInfo1.getZoneId());
+                    modelInfo.setTypeNumber(2);
+                    list.add(modelInfo);
                 }
             }
             List<ModelInfo> modelInfoList = modelInfoRepository.saveAll(list);