content: package com.platform.admin.controller;
  
  import com.platform.admin.mapper.JobGroupMapper;
  import com.platform.admin.mapper.JobInfoMapper;
  import com.platform.admin.mapper.JobRegistryMapper;
  import com.platform.core.biz.model.ReturnT;
  import com.platform.core.enums.RegistryConfig;
  import com.platform.admin.core.util.I18nUtil;
  import com.platform.admin.entity.JobGroup;
  import com.platform.admin.entity.JobRegistry;
  import io.swagger.annotations.Api;
  import io.swagger.annotations.ApiOperation;
  import io.swagger.annotations.ApiParam;
  import org.springframework.web.bind.annotation.*;
  
  import javax.annotation.Resource;
  import java.util.*;
  
  /**
   *
   * @author AllDataDC
   * @date 2023/3/26 11:14
   * 执行器管理接口
   **/
  @RestController
  @RequestMapping("/jobGroup")
  @Api(tags = "执行器管理接口")
  public class JobGroupController {
  
      @Resource
      public JobInfoMapper jobInfoMapper;
      @Resource
      public JobGroupMapper jobGroupMapper;
      @Resource
      private JobRegistryMapper jobRegistryMapper;
  
      @GetMapping("/list")
      @ApiOperation("执行器列表")
      public ReturnT<List<JobGroup>> getExecutorList() {
          return new ReturnT<>(jobGroupMapper.findAll());
      }
  
      @PostMapping("/save")
      @ApiOperation("新建执行器")
      public ReturnT<String> save(@RequestBody JobGroup jobGroup) {
  
          // valid
          if (jobGroup.getAppName() == null || jobGroup.getAppName().trim().length() == 0) {
              return new ReturnT<String>(500, (I18nUtil.getString("system_please_input") + "AppName"));
          }
          if (jobGroup.getAppName().length() < 4 || jobGroup.getAppName().length() > 64) {
              return new ReturnT<String>(500, I18nUtil.getString("jobgroup_field_appName_length"));
          }
          if (jobGroup.getTitle() == null || jobGroup.getTitle().trim().length() == 0) {
              return new ReturnT<String>(500, (I18nUtil.getString("system_please_input") + I18nUtil.getString("jobgroup_field_title")));
          }
          if (jobGroup.getAddressType() != 0) {
              if (jobGroup.getAddressList() == null || jobGroup.getAddressList().trim().length() == 0) {
                  return new ReturnT<String>(500, I18nUtil.getString("jobgroup_field_addressType_limit"));
              }
              String[] addresses = jobGroup.getAddressList().split(",");
              for (String item : addresses) {
                  if (item == null || item.trim().length() == 0) {
                      return new ReturnT<String>(500, I18nUtil.getString("jobgroup_field_registryList_invalid"));
                  }
              }
          }
  
          int ret = jobGroupMapper.save(jobGroup);
          return (ret > 0) ? ReturnT.SUCCESS : ReturnT.FAIL;
      }
  
      @PostMapping("/update")
      @ApiOperation("更新执行器")
      public ReturnT<String> update(@RequestBody JobGroup jobGroup) {
          // valid
          if (jobGroup.getAppName() == null || jobGroup.getAppName().trim().length() == 0) {
              return new ReturnT<String>(500, (I18nUtil.getString("system_please_input") + "AppName"));
          }
          if (jobGroup.getAppName().length() < 4 || jobGroup.getAppName().length() > 64) {
              return new ReturnT<String>(500, I18nUtil.getString("jobgroup_field_appName_length"));
          }
          if (jobGroup.getTitle() == null || jobGroup.getTitle().trim().length() == 0) {
              return new ReturnT<String>(500, (I18nUtil.getString("system_please_input") + I18nUtil.getString("jobgroup_field_title")));
          }
          if (jobGroup.getAddressType() == 0) {
              // 0=自动注册
              List<String> registryList = findRegistryByAppName(jobGroup.getAppName());
              String addressListStr = null;
              if (registryList != null && !registryList.isEmpty()) {
                  Collections.sort(registryList);
                  addressListStr = "";
                  for (String item : registryList) {
                      addressListStr += item + ",";
                  }
                  addressListStr = addressListStr.substring(0, addressListStr.length() - 1);
              }
              jobGroup.setAddressList(addressListStr);
          } else {
              // 1=手动录入
              if (jobGroup.getAddressList() == null || jobGroup.getAddressList().trim().length() == 0) {
                  return new ReturnT<String>(500, I18nUtil.getString("jobgroup_field_addressType_limit"));
              }
              String[] addresses = jobGroup.getAddressList().split(",");
              for (String item : addresses) {
                  if (item == null || item.trim().length() == 0) {
                      return new ReturnT<String>(500, I18nUtil.getString("jobgroup_field_registryList_invalid"));
                  }
              }
          }
  
          int ret = jobGroupMapper.update(jobGroup);
          return (ret > 0) ? ReturnT.SUCCESS : ReturnT.FAIL;
      }
  
      private List<String> findRegistryByAppName(String appNameParam) {
          HashMap<String, List<String>> appAddressMap = new HashMap<>();
          List<JobRegistry> list = jobRegistryMapper.findAll(RegistryConfig.DEAD_TIMEOUT, new Date());
          if (list != null) {
              for (JobRegistry item : list) {
                  if (RegistryConfig.RegistType.EXECUTOR.name().equals(item.getRegistryGroup())) {
                      String appName = item.getRegistryKey();
                      List<String> registryList = appAddressMap.get(appName);
                      if (registryList == null) {
                          registryList = new ArrayList<>();
                      }
  
                      if (!registryList.contains(item.getRegistryValue())) {
                          registryList.add(item.getRegistryValue());
                      }
                      appAddressMap.put(appName, registryList);
                  }
              }
          }
          return appAddressMap.get(appNameParam);
      }
  
      @PostMapping("/remove")
      @ApiOperation("移除执行器")
      public ReturnT<String> remove(int id) {
  
          // valid
          int count = jobInfoMapper.pageListCount(0, 10, id, -1, null, null, 0,null);
          if (count > 0) {
              return new ReturnT<>(500, I18nUtil.getString("jobgroup_del_limit_0"));
          }
  
          List<JobGroup> allList = jobGroupMapper.findAll();
          if (allList.size() == 1) {
              return new ReturnT<>(500, I18nUtil.getString("jobgroup_del_limit_1"));
          }
  
          int ret = jobGroupMapper.remove(id);
          return (ret > 0) ? ReturnT.SUCCESS : ReturnT.FAIL;
      }
  
      @RequestMapping(value = "/loadById", method = RequestMethod.POST)
      @ApiOperation("根据id获取执行器")
      public ReturnT<JobGroup> loadById(int id) {
          JobGroup jobGroup = jobGroupMapper.load(id);
          return jobGroup != null ? new ReturnT<>(jobGroup) : new ReturnT<>(ReturnT.FAIL_CODE, null);
      }
  
      @GetMapping("/query")
      @ApiOperation("查询执行器")
      public ReturnT<List<JobGroup>> get(@ApiParam(value = "执行器AppName")
                                         @RequestParam(value = "appName", required = false) String appName,
                                         @ApiParam(value = "执行器名称")
                                         @RequestParam(value = "title", required = false) String title,
                                         @ApiParam(value = "执行器地址列表")
                                         @RequestParam(value = "addressList", required = false) String addressList) {
          return new ReturnT<>(jobGroupMapper.find(appName, title, addressList));
      }
  
  }
  