feat: websocket page添加提供扩展 TencentBlueKing#4470

fitzcao · Jun 24, 2021 · e1f0c10 · e1f0c10
1 parent bf9c4bd
commit e1f0c10
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 4 deletions.
diff --git a/...ss/src/main/kotlin/com/tencent/devops/process/websocket/PipelineWebSocketConfiguration.kt b/...ss/src/main/kotlin/com/tencent/devops/process/websocket/PipelineWebSocketConfiguration.kt
@@ -33,6 +33,7 @@ import com.tencent.devops.common.websocket.dispatch.WebSocketDispatcher
 import com.tencent.devops.process.websocket.listener.PipelineWebSocketListener
 import com.tencent.devops.process.websocket.page.DefaultDetailPageBuild
 import com.tencent.devops.process.websocket.page.DefaultHistoryPageBuild
+import com.tencent.devops.process.websocket.page.DefaultStatusPageBuild
 import com.tencent.devops.process.websocket.page.DetailPageBuild
 import org.springframework.amqp.core.Binding
 import org.springframework.amqp.core.BindingBuilder
@@ -114,4 +115,8 @@ class PipelineWebSocketConfiguration {
     @ConditionalOnMissingBean(name = ["detailPage"])
     fun detailPage() = DefaultDetailPageBuild()
 
+    @Bean
+    @ConditionalOnMissingBean(name = ["statusPage"])
+    fun statusPage() = DefaultStatusPageBuild()
+
 }
diff --git a/...ocess/src/main/kotlin/com/tencent/devops/process/websocket/page/DefaultStatusPageBuild.kt b/...ocess/src/main/kotlin/com/tencent/devops/process/websocket/page/DefaultStatusPageBuild.kt
@@ -0,0 +1,9 @@
+package com.tencent.devops.process.websocket.page
+
+import com.tencent.devops.common.websocket.pojo.BuildPageInfo
+
+class DefaultStatusPageBuild: StatusPageBuild() {
+    override fun extStatusPage(buildPageInfo: BuildPageInfo): String? {
+        return null
+    }
+}
diff --git a/.../biz-process/src/main/kotlin/com/tencent/devops/process/websocket/page/StatusPageBuild.kt b/.../biz-process/src/main/kotlin/com/tencent/devops/process/websocket/page/StatusPageBuild.kt
@@ -30,6 +30,14 @@ package com.tencent.devops.process.websocket.page
 import com.tencent.devops.common.websocket.page.IPath
 import com.tencent.devops.common.websocket.pojo.BuildPageInfo
 
-class StatusPageBuild : IPath {
-    override fun buildPage(buildPageInfo: BuildPageInfo): String = "/console/pipeline/${buildPageInfo.projectId}/list"
+abstract class StatusPageBuild : IPath {
+    override fun buildPage(buildPageInfo: BuildPageInfo): String {
+        val defaultPage = "/console/pipeline/${buildPageInfo.projectId}/list"
+        if (!extStatusPage(buildPageInfo).isNullOrEmpty()) {
+            return extStatusPage(buildPageInfo)!!
+        }
+        return defaultPage
+    }
+
+    abstract fun extStatusPage(buildPageInfo: BuildPageInfo): String?
 }
diff --git a/.../src/main/kotlin/com/tencent/devops/process/websocket/service/PipelineWebsocketService.kt b/.../src/main/kotlin/com/tencent/devops/process/websocket/service/PipelineWebsocketService.kt
@@ -48,7 +48,8 @@ class PipelineWebsocketService @Autowired constructor(
     val redisOperation: RedisOperation,
     val objectMapper: ObjectMapper,
     val historyPageBuild: HistoryPageBuild,
-    val detailPageBuild: DetailPageBuild
+    val detailPageBuild: DetailPageBuild,
+    val statusPageBuild: StatusPageBuild
 ) {
     fun buildDetailMessage(
         buildId: String,
@@ -128,7 +129,7 @@ class PipelineWebsocketService @Autowired constructor(
         pipelineId: String,
         userId: String
     ): StatusWebsocketPush {
-        val page = StatusPageBuild().buildPage(
+        val page = statusPageBuild.buildPage(
             buildPageInfo = BuildPageInfo(
                 buildId = buildId,
                 pipelineId = pipelineId,