com.xxl.job.plus.executor.service.impl.JobGroupServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xxl-job-auto-register Show documentation
xxl-job-auto-register
There is a newer version: 0.0.2
Show newest version
package com.xxl.job.plus.executor.service.impl;

import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.json.JSONArray;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.xxl.job.plus.executor.config.XxlJobAutoRegisterConfigProperties;
import com.xxl.job.plus.executor.model.XxlJobGroup;
import com.xxl.job.plus.executor.service.JobGroupService;
import com.xxl.job.plus.executor.service.JobLoginService;
import org.apache.logging.log4j.util.Strings;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * @author : Hydra
 * @date: 2022/9/19 17:34
 * @version: 1.0
 */
@Service
public class JobGroupServiceImpl implements JobGroupService {
//
//    @Value("${xxl.job.admin.addresses}")
//    private String adminAddresses;
//
//    @Value("${xxl.job.executor.appname}")
//    private String appName;
//
//    @Value("${xxl.job.executor.title}")
//    private String title;
//
//    /*
//     * 执行器地址类型：0=自动注册、1=手动录入
//     * */
//    @Value("${xxl.job.executor.addressType:0}")
//    private Integer addressType;
//
//    /*
//     * 执行器地址列表，多地址逗号分隔(手动录入)
//     * */
//    @Value("${xxl.job.executor.addressList:}")
//    private String addressList;

    @Autowired
    private JobLoginService jobLoginService;

    @Autowired
    private XxlJobAutoRegisterConfigProperties xxlJobAutoRegisterConfigProperties;

    @Override
    public List getJobGroup() {
        String url = xxlJobAutoRegisterConfigProperties.getAdmin().getAddresses() + "/jobgroup/pageList";
        HttpResponse response = HttpRequest.post(url)
                .form("appname", xxlJobAutoRegisterConfigProperties.getExecutor().getAppName())
                .form("title", xxlJobAutoRegisterConfigProperties.getExecutor().getTitle())
                .cookie(jobLoginService.getCookie())
                .execute();

        String body = response.body();
        JSONArray array = JSONUtil.parse(body).getByPath("data", JSONArray.class);
        List list = array.stream()
                .map(o -> JSONUtil.toBean((JSONObject) o, XxlJobGroup.class))
                .collect(Collectors.toList());

        return list;
    }

    @Override
    public boolean autoRegisterGroup() {
        String url = xxlJobAutoRegisterConfigProperties.getAdmin().getAddresses() + "/jobgroup/save";
        HttpRequest httpRequest = HttpRequest.post(url)
                .form("appname", xxlJobAutoRegisterConfigProperties.getExecutor().getAppName())
                .form("title", xxlJobAutoRegisterConfigProperties.getExecutor().getTitle());

        httpRequest.form("addressType", xxlJobAutoRegisterConfigProperties.getExecutor().getAddressType());
        if (xxlJobAutoRegisterConfigProperties.getExecutor().getAddressType().equals(1)) {
            if (Strings.isBlank(xxlJobAutoRegisterConfigProperties.getExecutor().getAddressList())) {
                throw new RuntimeException("手动录入模式下,执行器地址列表不能为空");
            }
            httpRequest.form("addressList", xxlJobAutoRegisterConfigProperties.getExecutor().getAddressList());
        }

        HttpResponse response = httpRequest.cookie(jobLoginService.getCookie())
                .execute();
        Object code = JSONUtil.parse(response.body()).getByPath("code");
        return code.equals(200);
    }

    @Override
    public boolean preciselyCheck() {
        List jobGroup = getJobGroup();
        Optional has = jobGroup.stream()
                .filter(xxlJobGroup -> xxlJobGroup.getAppname().equals(xxlJobAutoRegisterConfigProperties.getExecutor().getAppName())
                        && xxlJobGroup.getTitle().equals(xxlJobAutoRegisterConfigProperties.getExecutor().getTitle()))
                .findAny();
        return has.isPresent();
    }

    @Override
    public XxlJobGroup getJobGroup(String appName, String title) {
        String url = xxlJobAutoRegisterConfigProperties.getAdmin().getAddresses() + "/jobgroup/pageList";
        HttpResponse response = HttpRequest.post(url)
                .form("appname", appName)
                .form("title", title)
                .cookie(jobLoginService.getCookie())
                .execute();

        String body = response.body();
        JSONArray array = JSONUtil.parse(body).getByPath("data", JSONArray.class);
        List list = array.stream()
                .map(o -> JSONUtil.toBean((JSONObject) o, XxlJobGroup.class))
                .collect(Collectors.toList());

        return !list.isEmpty() ? list.get(0) : null;
    }

}