com.ecfront.ez.framework.service.masterslave.Executor.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ezf-master-slave Show documentation
EZ框架Master-Slave服务
There is a newer version: 3.2.0
package com.ecfront.ez.framework.service.masterslave

import com.typesafe.scalalogging.slf4j.LazyLogging

import scala.annotation.tailrec

/**
  * 执行器
  *
  * @param dto 任务准备DTO
  */
case class Executor(dto: TaskPrepareDTO) extends LazyLogging {

  val category: String = dto.category
  var running: Boolean = false

  /**
    * 触发执行
    *
    * @param useNewThread 是否在新线程中执行
    */
  def run(useNewThread: Boolean): Unit = {
    logger.debug(s"Executor Running [${dto.instanceId}]")
    running = true
    if (useNewThread) {
      val self = this
      ExecutorPool.executeThreadPool.execute(new Runnable {
        override def run(): Unit = {
          Executor.execute(dto)
          ExecutorPool.finishExecute(self)
        }
      })
    } else {
      Executor.execute(dto)
      ExecutorPool.finishExecute(this)
    }
  }

}

object Executor extends Serializable with LazyLogging {

  private val DEFAULT_MAX_TRY_TIME: Int = 5

  private[masterslave] def maxTryTime: Int = DEFAULT_MAX_TRY_TIME

  private var processorContainer = Map[String, TaskBaseProcessor[_]]()

  private[masterslave] def registerProcessors(processors: List[TaskBaseProcessor[_]]): Unit = {
    processorContainer = processors.map {
      processor =>
        processor.category -> processor
    }.toMap
  }

  /**
    * 开始执行
    * @param dto 任务准备DTO
    */
  private[masterslave] def execute(dto: TaskPrepareDTO): Unit = {
    // 先通知开始任务
    Assigner.Worker.startTask(TaskStartDTO(dto.instanceId))
    val processorOpt = processorContainer.get(dto.category)
    if (processorOpt.isDefined) {
      doExecute(dto, processorOpt.get, 0)
    } else {
      logger.error("Processor NOT exist " + dto.category)
      Assigner.Worker.finishTask(TaskFinishDTO(
        dto.instanceId,
        isSuccess = false,
        hasChange = false,
        s"Not found process by category:${dto.category}",
        Map(),
        Map()
      ))
    }
  }

  @tailrec
  private def doExecute(dto: TaskPrepareDTO, processor: TaskBaseProcessor[_], tryTimes: Int): Unit = {
    try {
      if (tryTimes != 0) {
        logger.warn(s"Worker process error try [$tryTimes] times ")
      }
      val changeCheckResp = processor.hasChange(dto.taskInfo, dto.taskVar, dto.instanceParameters)
      if (changeCheckResp) {
        if (changeCheckResp.body) {
          logger.debug(s"Worker process execute [${dto.instanceId}]")
          val resp = processor.execute(dto.taskInfo, dto.taskVar, dto.instanceParameters, tryTimes != 0)
          if (resp) {
            Assigner.Worker.finishTask(TaskFinishDTO(
              dto.instanceId,
              isSuccess = true,
              hasChange = true,
              "",
              resp.body._1,
              resp.body._2
            ))
          } else {
            Assigner.Worker.finishTask(TaskFinishDTO(
              dto.instanceId,
              isSuccess = false,
              hasChange = false,
              resp.message,
              Map(),
              Map()
            ))
          }
        } else {
          Assigner.Worker.finishTask(TaskFinishDTO(
            dto.instanceId,
            isSuccess = true,
            hasChange = false,
            "",
            Map(),
            Map()
          ))
        }
      } else {
        // 没有变更
        Assigner.Worker.finishTask(TaskFinishDTO(
          dto.instanceId,
          isSuccess = false,
          hasChange = false,
          changeCheckResp.message,
          Map(),
          Map()
        ))
      }
    } catch {
      case e: Throwable =>
        if (tryTimes < maxTryTime) {
          logger.error("Worker process execute error , try again",e)
          doExecute(dto, processor, tryTimes + 1)
        } else {
          logger.error("Worker process execute error",e)
          Assigner.Worker.finishTask(TaskFinishDTO(
            dto.instanceId,
            isSuccess = false,
            hasChange = false,
            e.getMessage,
            Map(),
            Map()
          ))
        }
    }
  }

}