升级固件

src/main/java/com/cisd/tms/modules/upgrade/executor/UpgradeTaskRunner.java

@@ -98,16 +98,8 @@ public class UpgradeTaskRunner {
                 return;
             }
 
-            // 只有 execute 和 verify 都成功后，才认为升级成功并回写设备软件版本表。
-            DeviceSoftwareVersionEntity versionEntity = new DeviceSoftwareVersionEntity();
-            String componentCode = componentCode(task.getTaskType());
-            versionEntity.setComponentCode(componentCode);
-            versionEntity.setComponentName(componentName(componentCode));
-            versionEntity.setCurrentVersion(task.getTargetVersion());
-            versionEntity.setSourceType("UPGRADE");
-            versionEntity.setDetectedAt(LocalDateTime.now());
-            versionEntity.setRemarks("updated by offline upgrade task " + task.getTaskId());
-            deviceSoftwareVersionRepository.saveOrUpdate(versionEntity);
+            // 只有软件类升级才回写设备软件版本表；固件版本第一版暂不在该表维护。
+            updateSoftwareVersionIfRequired(task);
 
             upgradeTaskRepository.updateStatus(
                     task.getTaskId(),
@@ -221,6 +213,21 @@ public class UpgradeTaskRunner {
         };
     }
 
+    private void updateSoftwareVersionIfRequired(UpgradeTaskEntity task) {
+        String componentCode = componentCode(task.getTaskType());
+        if (!"TMS".equals(componentCode) && !"RECEIVER".equals(componentCode)) {
+            return;
+        }
+        DeviceSoftwareVersionEntity versionEntity = new DeviceSoftwareVersionEntity();
+        versionEntity.setComponentCode(componentCode);
+        versionEntity.setComponentName(componentName(componentCode));
+        versionEntity.setCurrentVersion(task.getTargetVersion());
+        versionEntity.setSourceType("UPGRADE");
+        versionEntity.setDetectedAt(LocalDateTime.now());
+        versionEntity.setRemarks("updated by offline upgrade task " + task.getTaskId());
+        deviceSoftwareVersionRepository.saveOrUpdate(versionEntity);
+    }
+
     private String componentCode(String taskType) {
         return trim(taskType);
     }

src/test/java/com/cisd/tms/modules/upgrade/executor/UpgradeTaskRunnerTest.java

@@ -82,6 +82,30 @@ class UpgradeTaskRunnerTest {
         Assertions.assertEquals("标准收发器", versionRepository.store.get("RECEIVER").getComponentName());
     }
 
+    @Test
+    void shouldNotUpdateSoftwareVersionWhenFirmwareUpgradeSucceeds() throws Exception {
+        Path tempDir = Files.createTempDirectory("upgrade-runner-test");
+        FileRecordEntity fileRecord = buildPackage(tempDir, false, "#!/bin/sh\necho execute\nexit 0\n", false);
+        InMemoryFileRecordRepository fileRepository = new InMemoryFileRecordRepository();
+        fileRepository.save(fileRecord);
+        InMemoryUpgradeTaskRepository taskRepository = new InMemoryUpgradeTaskRepository();
+        InMemoryDeviceSoftwareVersionRepository versionRepository = new InMemoryDeviceSoftwareVersionRepository();
+
+        UpgradeTaskRunner runner = new UpgradeTaskRunner(
+                taskRepository,
+                fileRepository,
+                versionRepository,
+                stagingService(tempDir.resolve("staging"), tempDir.resolve("logs")),
+                properties(tempDir.resolve("staging"), tempDir.resolve("logs")),
+                objectMapper
+        );
+
+        runner.run(task("UPG-001-F", fileRecord.getFileId(), "FIRMWARE", "V3.0.0"));
+
+        Assertions.assertEquals("SUCCESS", taskRepository.status);
+        Assertions.assertTrue(versionRepository.store.isEmpty());
+    }
+
     @Test
     void shouldMarkTaskFailedWhenExecuteScriptFails() throws Exception {
         Path tempDir = Files.createTempDirectory("upgrade-runner-test");