feat :: 상태확인 API

team-xquare · Feb 20, 2024 · f70b6bf · f70b6bf
1 parent 5ef13b5
commit f70b6bf
Show file tree

Hide file tree

Showing 6 changed files with 60 additions and 12 deletions.
diff --git a/src/main/kotlin/com/example/mergebackend/domain/deploy/presentation/DeployController.kt b/src/main/kotlin/com/example/mergebackend/domain/deploy/presentation/DeployController.kt
@@ -81,4 +81,10 @@ class DeployController(
         @RequestParam("env_type")
         envType: EnvType
     ): DeployLogResponse = deployService.getLogs(deployId, envType)
+
+    @GetMapping("/status")
+    fun getPodStatus(
+        @RequestParam("deploy_id")
+        deployId: UUID
+    ) = deployService.checkPodStatus(deployId)
 }
diff --git a/src/main/kotlin/com/example/mergebackend/domain/deploy/service/DeployService.kt b/src/main/kotlin/com/example/mergebackend/domain/deploy/service/DeployService.kt
@@ -16,4 +16,5 @@ interface DeployService {
     fun getDeployList(projectId: UUID): DeployListResponse
     fun updateUrl(containerName: String, serviceType: ServiceType, prefix: String?, domain: Map<String,String>?)
     fun getLogs(deployId: UUID, envType: EnvType): DeployLogResponse
+    fun checkPodStatus(deployId: UUID): Map<String, String>
 }
diff --git a/src/main/kotlin/com/example/mergebackend/domain/deploy/service/impl/DeployServiceImpl.kt b/src/main/kotlin/com/example/mergebackend/domain/deploy/service/impl/DeployServiceImpl.kt
@@ -22,14 +22,12 @@ import com.example.mergebackend.infra.feign.deploy.dto.FeignCreateDeployRequest
 import com.example.mergebackend.infra.feign.tsdata.TsDataClient
 import com.example.mergebackend.infra.feign.tsdata.dto.GetLogRequest
 import com.example.mergebackend.infra.feign.tsdata.dto.QueryDto
-import kotlinx.serialization.encodeToString
-import kotlinx.serialization.json.Json
+import com.example.mergebackend.infra.kubernetes.KubernetesClientUtil
 import org.springframework.data.repository.findByIdOrNull
 import org.springframework.stereotype.Service
 import org.springframework.transaction.annotation.Transactional
 import java.time.Instant
 import java.util.*
-import javax.management.Query
 
 @Transactional
 @Service
@@ -38,7 +36,8 @@ private class DeployServiceImpl(
     private val projectRepository: ProjectRepository,
     private val deployClient: DeployClient,
     private val userFacade: UserFacade,
-    private val tsDataClient: TsDataClient
+    private val tsDataClient: TsDataClient,
+    private val kubernetesClientUtil: KubernetesClientUtil
 ): DeployService {
     override fun createDeploy(createDeployRequest: CreateDeployRequest): CreateDeployResponse {
         val project = projectRepository.findByIdOrNull(createDeployRequest.projectId) ?: throw ProjectNotFoundException
@@ -148,4 +147,19 @@ private class DeployServiceImpl(
         return DeployLogResponse(response.results.a.frames[0].data.values[2])
 
     }
+
+    override fun checkPodStatus(deployId: UUID): Map<String, String> {
+        val deploy = deployRepository.findByIdOrNull(deployId) ?: throw DeployNotFoundException
+
+        val namespacePrefix = "${deploy.project.teamNameEn}-"
+        val deployPrefix = "${deploy.containerName}-${deploy.serviceType.toString().lowercase()}-"
+
+        val stagStatus = kubernetesClientUtil.checkContainerStatus(namespace = "${namespacePrefix}stag", deploymentName = "${deployPrefix}stag")
+        val prodStatus = kubernetesClientUtil.checkContainerStatus(namespace = "${namespacePrefix}prod", deploymentName = "${deployPrefix}prod")
+
+        return mapOf(
+            "stag" to stagStatus,
+            "prod" to prodStatus
+        )
+    }
 }
diff --git a/src/main/kotlin/com/example/mergebackend/global/config/kubernetes/KubernetesClientConfig.kt b/src/main/kotlin/com/example/mergebackend/global/config/kubernetes/KubernetesClientConfig.kt
@@ -1,19 +1,14 @@
 package com.example.mergebackend.global.config.kubernetes
 
-import com.example.mergebackend.global.config.kubernetes.dto.KubernetesToken
 import com.example.mergebackend.global.env.kubernetes.KubernetesProperty
 import com.example.mergebackend.global.env.kubernetes.XquareAwsProperty
-import com.example.mergebackend.infra.kubernetes.exception.KubernetesException
 import io.kubernetes.client.openapi.Configuration
+import io.kubernetes.client.openapi.apis.CoreV1Api
 import io.kubernetes.client.openapi.apis.CustomObjectsApi
 import io.kubernetes.client.util.ClientBuilder
 import io.kubernetes.client.util.KubeConfig
-import kotlinx.serialization.decodeFromString
-import kotlinx.serialization.json.Json
 import org.springframework.context.annotation.Bean
 import software.amazon.awssdk.regions.Region
-import java.io.BufferedReader
-import java.io.InputStreamReader
 import java.io.StringReader
 import java.nio.charset.Charset
 import java.util.*
@@ -55,4 +50,7 @@ class KubernetesClientConfig(
 
     @Bean
     fun customObjectsApi() = CustomObjectsApi()
+
+    @Bean
+    fun coreV1API() = CoreV1Api()
 }
diff --git a/src/main/kotlin/com/example/mergebackend/infra/kubernetes/KubernetesClientUtil.kt b/src/main/kotlin/com/example/mergebackend/infra/kubernetes/KubernetesClientUtil.kt
@@ -2,4 +2,5 @@ package com.example.mergebackend.infra.kubernetes
 
 interface KubernetesClientUtil {
     fun deleteSecret(namespace: String, crName: String)
+    fun checkContainerStatus(deploymentName: String, namespace: String): String
 }
diff --git a/src/main/kotlin/com/example/mergebackend/infra/kubernetes/KubernetesClientUtilImpl.kt b/src/main/kotlin/com/example/mergebackend/infra/kubernetes/KubernetesClientUtilImpl.kt
@@ -1,14 +1,16 @@
 package com.example.mergebackend.infra.kubernetes
 
 import com.example.mergebackend.infra.kubernetes.exception.KubernetesException
+import io.kubernetes.client.openapi.ApiException
+import io.kubernetes.client.openapi.apis.CoreV1Api
 import io.kubernetes.client.openapi.apis.CustomObjectsApi
 import io.kubernetes.client.openapi.models.V1DeleteOptions
 import org.springframework.stereotype.Service
-import java.lang.Exception
 
 @Service
 class KubernetesClientUtilImpl(
-    private val customObjectsApi: CustomObjectsApi
+    private val customObjectsApi: CustomObjectsApi,
+    private val coreV1Api: CoreV1Api
 ): KubernetesClientUtil {
     override fun deleteSecret(namespace: String, crName: String) {
         val v1DeleteOption = V1DeleteOptions()
@@ -30,4 +32,30 @@ class KubernetesClientUtilImpl(
             throw KubernetesException
         }
     }
+
+    override fun checkContainerStatus(deploymentName: String, namespace: String): String {
+        try {
+            val podList = coreV1Api.listNamespacedPod(
+                namespace,
+                null,
+                null,
+                null,
+                null,
+                null,
+                null,
+                null,
+                null,
+                null,
+                null
+            )
+            podList.items.forEach { pod ->
+                if (pod.metadata?.name?.startsWith(deploymentName) == true) {
+                    return pod.status?.phase ?: "배포되지 않았습니다."
+                }
+            }
+        } catch (e: ApiException) {
+            e.printStackTrace()
+        }
+        return "d"
+    }
 }