diff --git a/src/main/java/com/huaheng/pc/config/location/mapper/LocationMapper.java b/src/main/java/com/huaheng/pc/config/location/mapper/LocationMapper.java
index 7b2afaa..005480e 100644
--- a/src/main/java/com/huaheng/pc/config/location/mapper/LocationMapper.java
+++ b/src/main/java/com/huaheng/pc/config/location/mapper/LocationMapper.java
@@ -19,6 +19,8 @@ public interface LocationMapper extends BaseMapper<Location> {
 
     int addList(@Param("locations") List<Location> locations);
 
+    int updateList(@Param("locations") List<Location> locations);
+
     Location getAllLocation(@Param("warehouseCode") String warehouseCode, @Param("type") String type);
 
     //库位利用率
diff --git a/src/main/java/com/huaheng/pc/config/location/service/LocationServiceImpl.java b/src/main/java/com/huaheng/pc/config/location/service/LocationServiceImpl.java
index 2820c72..0aa885f 100644
--- a/src/main/java/com/huaheng/pc/config/location/service/LocationServiceImpl.java
+++ b/src/main/java/com/huaheng/pc/config/location/service/LocationServiceImpl.java
@@ -160,6 +160,37 @@ public class LocationServiceImpl extends ServiceImpl<LocationMapper, Location> i
         }else {
             locationMapper.addList(locations);
         }
+
+        List<Integer> integerList = new ArrayList<>();
+        for(Location location2 : locations) {
+            if(integerList.size() == 0) {
+                integerList.add(location2.getIRow());
+            } else {
+                for(int i = 0; i< integerList.size(); i++) {
+                    if(integerList.get(i).intValue() == location2.getIRow().intValue()) {
+                        break;
+                    } else {
+                        if(i == (integerList.size() - 1)) {
+                            integerList.add(location2.getIRow());
+                        }
+                    }
+                }
+            }
+        }
+        if(integerList.size() == 4) {
+            for (Location location3 : locations) {
+                for (int i = 0; i < integerList.size(); i++) {
+                    if (location3.getIRow().intValue() == integerList.get(i).intValue()) {
+                        if (i == 0 || i == (integerList.size() - 1)) {
+                            location3.setRowFlag(1);
+                        } else {
+                            location3.setRowFlag(0);
+                        }
+                    }
+                }
+            }
+            locationMapper.updateList(locations);
+        }
         return true;
     }