perf: code shorten

labring · Dec 17, 2024 · 0e11709 · 0e11709
1 parent 454d4e0
commit 0e11709
Showing 1 changed file with 3 additions and 24 deletions.
diff --git a/frontend/providers/devbox/app/api/platform/getRuntime/route.ts b/frontend/providers/devbox/app/api/platform/getRuntime/route.ts
@@ -49,17 +49,10 @@ export async function GET(req: NextRequest) {
     const languageList = runtimeClasses?.items.filter((item: any) => item.spec.kind === 'Language')
     const dealtLanguageList = languageList.map((item: any) => {
       const aRuntime = runtimes.find((runtime: any) => runtime.spec.classRef === item.metadata.name)
-      if (aRuntime?.spec.category?.includes('gpu')) {
-        return {
-          id: item.metadata.name,
-          label: item.spec.title,
-          gpu: true
-        }
-      }
       return {
         id: item.metadata.name,
         label: item.spec.title,
-        gpu: false
+        gpu: !!aRuntime?.spec.category?.includes('gpu')
       }
     })
     languageTypeList.push(...dealtLanguageList)
@@ -69,35 +62,21 @@ export async function GET(req: NextRequest) {
     )
     const dealtFrameworkList = frameworkList.map((item: any) => {
       const aRuntime = runtimes.find((runtime: any) => runtime.spec.classRef === item.metadata.name)
-      if (aRuntime?.spec.category?.includes('gpu')) {
-        return {
-          id: item.metadata.name,
-          label: item.spec.title,
-          gpu: true
-        }
-      }
       return {
         id: item.metadata.name,
         label: item.spec.title,
-        gpu: false
+        gpu: !!aRuntime?.spec.category?.includes('gpu')
       }
     })
     frameworkTypeList.push(...dealtFrameworkList)
 
     const osList = runtimeClasses?.items.filter((item: any) => item.spec.kind === 'OS')
     const dealtOsList = osList.map((item: any) => {
       const aRuntime = runtimes.find((runtime: any) => runtime.spec.classRef === item.metadata.name)
-      if (aRuntime?.spec.category?.includes('gpu')) {
-        return {
-          id: item.metadata.name,
-          label: item.spec.title,
-          gpu: true
-        }
-      }
       return {
         id: item.metadata.name,
         label: item.spec.title,
-        gpu: false
+        gpu: !!aRuntime?.spec.category?.includes('gpu')
       }
     })
     osTypeList.push(...dealtOsList)