修改

src/main/java/com/ruoyi/project/zjsgfa/controller/SjDjjcController.java

@@ -9,8 +9,10 @@ import java.util.List;
 import java.util.UUID;
 import javax.servlet.http.HttpServletResponse;
 
+import com.ruoyi.common.constant.Constants;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.framework.config.RuoYiConfig;
+import com.ruoyi.framework.config.ServerConfig;
 import com.ruoyi.project.zjsgfa.domain.*;
 import com.ruoyi.project.zjsgfa.domain.Vo.DrillingFluidConstant;
 import com.ruoyi.project.zjsgfa.mapper.*;
@@ -97,6 +99,9 @@ public class SjDjjcController extends BaseController
     @Autowired
     private SjGztzMapper sjGztzMapper;
 
+    @Autowired
+    private ServerConfig serverConfig;
+
 
 
 
@@ -1232,40 +1237,77 @@ public class SjDjjcController extends BaseController
 
 
 
-//            List<XSSFPicture> pictures = getPictures(sheet);
-//
-//            // 遍历图片并处理
-//            for (XSSFPicture picture : pictures) {
-//                // 获取锚点信息（位置）
-//                XSSFClientAnchor anchor = picture.getClientAnchor();
-//                XSSFPictureData pictureData = picture.getPictureData();
-//                String ext = pictureData.suggestFileExtension();
-//                String filePath = RuoYiConfig.getUploadPath()+"/gztz/";
-//                //判断是否为井位图
-//                if(isPictureInRow(anchor, 0, 40)){
-//                    SjGztz sjGztz=new SjGztz();
-//                    sjGztz.setJh(jh);
-//                    sjGztz.setGzlx("井位图");
-//                    String fileName = UUID.randomUUID().toString();
-//                    String lj = filePath + fileName + "." + ext;
-//                    sjGztz.setLj(lj);
-//                    FileOutputStream fos = new FileOutputStream(new File(filePath+fileName+"." + ext));
-//                    fos.write(pictureData.getData());
-//                    fos.close();
-//                    sjGztzList.add(sjGztz);
-//                }else if(isPictureInRow(anchor, 41, 81)){
-//                    SjGztz sjGztz=new SjGztz();
-//                    sjGztz.setJh(jh);
-//                    sjGztz.setGzlx("地震刨面图");
-//                    String fileName = UUID.randomUUID().toString();
-//                    String lj = filePath + fileName + "." + ext;
-//                    sjGztz.setLj(lj);
-//                    FileOutputStream fos = new FileOutputStream(new File(filePath+fileName+"." + ext));
-//                    fos.write(pictureData.getData());
-//                    fos.close();
-//                    sjGztzList.add(sjGztz);
-//                }
-//            }
+            List<XSSFPicture> pictures = getPictures(sheet);
+
+            // 遍历图片并处理
+            for (XSSFPicture picture : pictures) {
+                // 获取锚点信息（位置）
+                XSSFClientAnchor anchor = picture.getClientAnchor();
+                XSSFPictureData pictureData = picture.getPictureData();
+                String ext = pictureData.suggestFileExtension();
+                String filePath = RuoYiConfig.getUploadPath()+"/gztz/";
+                //判断是否为井位图
+                if(isPictureInRow(anchor, 0, 0)){
+                    SjGztz sjGztz=new SjGztz();
+                    sjGztz.setJh(jh);
+                    sjGztz.setGzlx("井位图");
+                    String fileName = UUID.randomUUID().toString();
+                    String lj = filePath + fileName + "." + ext;
+
+                    File desc = new File(lj);
+
+                    if (!desc.exists())
+                    {
+                        if (!desc.getParentFile().exists())
+                        {
+                            desc.getParentFile().mkdirs();
+                        }
+                    }
+                    sjGztz.setLj( Constants.RESOURCE_PREFIX +"/upload/gztz/"+fileName + "." + ext);
+                    FileOutputStream fos = new FileOutputStream(new File(filePath+fileName+"." + ext));
+                    fos.write(pictureData.getData());
+                    fos.close();
+                    sjGztzList.add(sjGztz);
+                }else if(isPictureInRow(anchor, 1, 1)){
+                    SjGztz sjGztz=new SjGztz();
+                    sjGztz.setJh(jh);
+                    sjGztz.setGzlx("地震刨面图");
+                    String fileName = UUID.randomUUID().toString();
+                    String lj = filePath + fileName + "." + ext;
+                    File desc = new File(lj);
+                    if (!desc.exists())
+                    {
+                        if (!desc.getParentFile().exists())
+                        {
+                            desc.getParentFile().mkdirs();
+                        }
+                    }
+                    sjGztz.setLj( Constants.RESOURCE_PREFIX +"/upload/gztz/"+fileName + "." + ext);
+                    FileOutputStream fos = new FileOutputStream(new File(filePath+fileName+"." + ext));
+                    fos.write(pictureData.getData());
+                    fos.close();
+                    sjGztzList.add(sjGztz);
+                }else if(isPictureInRow(anchor, 2, 2)){
+                    SjGztz sjGztz=new SjGztz();
+                    sjGztz.setJh(jh);
+                    sjGztz.setGzlx("三压力刨面图");
+                    String fileName = UUID.randomUUID().toString();
+                    String lj = filePath + fileName + "." + ext;
+                    File desc = new File(lj);
+                    if (!desc.exists())
+                    {
+                        if (!desc.getParentFile().exists())
+                        {
+                            desc.getParentFile().mkdirs();
+                        }
+                    }
+                    sjGztz.setLj( Constants.RESOURCE_PREFIX +"/upload/gztz/"+fileName + "." + ext);
+                    FileOutputStream fos = new FileOutputStream(new File(filePath+fileName+"." + ext));
+                    fos.write(pictureData.getData());
+                    fos.close();
+                    sjGztzList.add(sjGztz);
+                }
+            }
             if(sjGztzList.size()>0){
                 sjGztzMapper.deleteSjGztzByJh(jh);
                 sjGztzMapper.insertSjGztzBatch(sjGztzList);