try to use uuid to find crd

bugitt · Dec 31, 2023 · 1100c79 · 1100c79
1 parent 0fbfec8
commit 1100c79
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cloudapi-web/src/main/kotlin/cn/edu/buaa/scs/service/Vm.kt b/cloudapi-web/src/main/kotlin/cn/edu/buaa/scs/service/Vm.kt
@@ -44,6 +44,12 @@ class VmService(val call: ApplicationCall) : IService {
             ?: throw NotFoundException("VirtualMachine($uuid) is not found")
     }
 
+    fun getVmByUUid(uuid: String): VirtualMachineCrd {
+        return vmKubeClient.inAnyNamespace().withField("status.uuid", uuid).list().items.filterNot { it.spec.deleted }
+            .firstOrNull()
+            ?: throw NotFoundException("VirtualMachine($uuid) is not found")
+    }
+
     fun deleteVm(id: String) {
         val vm = getVmByUUID(id)
         vm.spec = vm.spec.copy(deleted = true)
@@ -346,7 +352,7 @@ class VmService(val call: ApplicationCall) : IService {
 //        if (mysql.virtualMachines.exists { it.isTemplate.eq(true) and it.name.eq(name) }) {
 //            throw BadRequestException("template name already exists")
 //        }
-        val vmCrd = getVmByUUID(crdId)
+        val vmCrd = getVmByUUid(uuid)
         vmCrd.spec = vmCrd.spec.copy(template = true)
         vmKubeClient.resource(vmCrd).patch()
         val owner = call.user()