com.tencent.devops.worker.WorkRunner.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of worker-common Show documentation
Tencent blueking ci project
The newest version!
/*
 * Tencent is pleased to support the open source community by making BK-CI 蓝鲸持续集成平台 available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company.  All rights reserved.
 *
 * BK-CI 蓝鲸持续集成平台 is licensed under the MIT license.
 *
 * A copy of the MIT License is included in this file.
 *
 *
 * Terms of the MIT License:
 * ---------------------------------------------------
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of
 * the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
 * LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
 * NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package com.tencent.devops.worker

import com.fasterxml.jackson.module.kotlin.readValue
import com.tencent.devops.common.api.exception.RemoteServiceException
import com.tencent.devops.common.api.util.JsonUtil
import com.tencent.devops.dispatch.pojo.thirdpartyagent.ThirdPartyBuildInfo
import com.tencent.devops.worker.common.ErrorMsgLogUtil
import com.tencent.devops.worker.common.Runner
import com.tencent.devops.worker.common.SLAVE_AGENT_PREPARE_START_FILE
import com.tencent.devops.worker.common.SLAVE_AGENT_START_FILE
import com.tencent.devops.worker.common.WorkspaceInterface
import com.tencent.devops.worker.common.api.utils.ThirdPartyAgentBuildInfoUtils
import com.tencent.devops.worker.common.env.BuildType
import com.tencent.devops.worker.common.exception.PropertyNotExistException
import com.tencent.devops.worker.common.logger.LoggerService
import com.tencent.devops.worker.common.utils.WorkspaceUtils
import org.slf4j.LoggerFactory
import java.io.File
import java.util.Base64
import kotlin.system.exitProcess

object WorkRunner {
    private val logger = LoggerFactory.getLogger(WorkRunner::class.java)

    fun execute(args: Array) {
        try {
            val buildInfo = getBuildInfo(args)!!

            addErrorMsgWriteToFileHook()

            logger.info("[${buildInfo.buildId}]|Start worker for build| projectId=${buildInfo.projectId}")

            deleteFileHooks()

            ThirdPartyAgentBuildInfoUtils.setBuildInfo(buildInfo)

            Runner.run(object : WorkspaceInterface {
                val workspace = buildInfo.workspace
                override fun getWorkspaceAndLogDir(
                    variables: Map,
                    pipelineId: String
                ): Pair {
                    val workspaceDir = WorkspaceUtils.getWorkspaceDir(
                        workspace = workspace,
                        buildType = BuildType.AGENT,
                        pipelineId = pipelineId,
                        variables = variables
                    )
                    val logPathDir = WorkspaceUtils.getPipelineLogDir(pipelineId)
                    return Pair(workspaceDir, logPathDir)
                }
            }, false)
            exitProcess(0)
        } catch (e: PropertyNotExistException) {
            logger.warn("The property(${e.key}) is not exist")
            exitProcess(-1)
        } catch (re: RemoteServiceException) {
            ErrorMsgLogUtil.appendErrorMsg(re.responseContent ?: re.message!!)
            exitProcess(-1) // fix #8360
        } catch (ignore: Throwable) {
            logger.error("Encounter unknown exception", ignore)
            LoggerService.addErrorLine("Other unknown error has occurred: " + ignore.message)
            ErrorMsgLogUtil.appendErrorMsg(ignore.message!!)
            exitProcess(-1)
        }
    }

    private fun deleteFileHooks() {
        val startFile = getStartFile()
        if (!startFile.isNullOrBlank()) {
            val file = File(startFile)
            if (file.exists()) {
                logger.info("The file ${file.absolutePath} will be deleted when exit")
                file.deleteOnExit()
            } else {
                logger.info("The file $file is not exist")
            }
        } else {
            logger.info("The start file is not exist in start file")
        }
        getPrepareStartFile()?.takeIf { it.isNotBlank() }?.let { prepareStartFile ->
            val file = File(prepareStartFile)
            if (file.exists()) {
                logger.info("The file ${file.absolutePath} will be deleted when exit")
                file.deleteOnExit()
            } else {
                logger.info("The file $file is not exist")
            }
        }
    }

    private fun getBuildInfo(args: Array): ThirdPartyBuildInfo? {
        if (args.isEmpty()) {
            logger.error("Empty argument")
            exitProcess(1)
        }
        val buildInfoStr = String(Base64.getDecoder().decode(args[0]))
        try {
            logger.info("Start read the build info ($buildInfoStr)")
            return JsonUtil.getObjectMapper().readValue(buildInfoStr)
        } catch (ignore: Throwable) {
            logger.warn("Fail to read the build Info", ignore)
            exitProcess(1)
        }
    }

    private fun addErrorMsgWriteToFileHook() {
        try {
            Runtime.getRuntime().addShutdownHook(object : Thread() {
                override fun run() {
                    ErrorMsgLogUtil.flushErrorMsgToFile()
                }
            })
        } catch (t: Throwable) {
            logger.warn("Fail to add shutdown hook", t)
        }
    }

    private fun getStartFile(): String? {
        return System.getProperty(SLAVE_AGENT_START_FILE, null)
    }

    private fun getPrepareStartFile(): String? {
        return System.getProperty(SLAVE_AGENT_PREPARE_START_FILE, null)
    }
}