Azkaban的AJAX的RestFul接口调用事例

azkaban是一个开源的调度工具,可能需要自己修改源码或者自己调用azkaban后台提供的restFul接口进行定制化的开发,在azkaban的官方文档上有接口的详细文档,具体参考:http://azkaban.github.io/azkaban/docs/latest/#ajax-api

测试代码如下:

• 1.SSLUtil

绕过ssl认证的工具类

注意:
如果不绕过ssl认证可以自己导入证书,通过如下语句在jre的 jre\lib\security目录下执行:

keytool -import -keystore cacerts -storepass changeit -keypass changeit -alias my-cas -file c:\xxxx\xxx.cer

package com.ys.azkaban.utils;

import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.X509Certificate;

import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;
import javax.net.ssl.x509trustmanager;

public final class SSLUtil {

    private static final TrustManager[] UNQUESTIONING_TRUST_MANAGER = new TrustManager[] { new x509trustmanager() {
        public java.security.cert.X509Certificate[] getAcceptedissuers() {
            return null;
        }

        public void checkClientTrusted(X509Certificate[] certs,String authType) {
        }

        public void checkServerTrusted(X509Certificate[] certs,String authType) {
        }
    } };

    public static void turnOffSslChecking() throws NoSuchAlgorithmException,KeyManagementException {
        final SSLContext sc = SSLContext.getInstance("SSL");
        sc.init(null,UNQUESTIONING_TRUST_MANAGER,null);
        HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory());
    }

    public static void turnOnSslChecking() throws KeyManagementException,NoSuchAlgorithmException {
        SSLContext.getInstance("SSL").init(null,null,null);
    }

    private SSLUtil() {
        throw new UnsupportedOperationException("Do not instantiate libraries.");
    }
}

• 2.测试用例

package com.ys.azkaban;

import java.io.File;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.Map;

import org.junit.Before;
import org.junit.Test;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.httpentity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;

import com.ys.azkaban.utils.SSLUtil;

public class AzkabanTest {

    // azkaban api addr : http://azkaban.github.io/azkaban/docs/latest/#ajax-api

    private static String API = "https://lijie:8443";

    RestTemplate restTemplate = null;

    @Before
    public void beforeTest() {
        SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
        requestFactory.setConnectTimeout(2000);
        requestFactory.setReadTimeout(2000);
        restTemplate = new RestTemplate(requestFactory);
    }

    /** * 登录测试 登录调度系统 * * @throws Exception */
    @Test
    public void loginTest() throws Exception {

        SSLUtil.turnOffSslChecking();
        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type","application/x-www-form-urlencoded; charset=utf-8");
        hs.add("X-Requested-With","XMLHttpRequest");
        LinkedMultiValueMap<String,String> linkedMultiValueMap = new LinkedMultiValueMap<String,String>();
        linkedMultiValueMap.add("action","login");
        linkedMultiValueMap.add("username","azkaban");
        linkedMultiValueMap.add("password","azkaban");

        httpentity<MultiValueMap<String,String>> httpentity = new httpentity<>(linkedMultiValueMap,hs);
        String postForObject = restTemplate.postForObject(API,httpentity,String.class);
        System.out.println(postForObject);
    }

    /** * 创建任务测试 创建一个project * * @throws Exception */
    @Test
    public void createProTest() throws Exception {
        SSLUtil.turnOffSslChecking();
        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type",String>();
        linkedMultiValueMap.add("session.id","fa47065d-4f00-46ea-bc82-c766a6262886");
        linkedMultiValueMap.add("action","create");
        linkedMultiValueMap.add("name","ystest0101");
        linkedMultiValueMap.add("description","ystest01 description");

        httpentity<MultiValueMap<String,hs);
        String postForObject = restTemplate.postForObject(API + "/manager",String.class);
        System.out.println(postForObject);

    }

    /** * 删除任务测试 删除一个project * * @throws Exception */
    @Test
    public void deleteProTest() throws Exception {

        SSLUtil.turnOffSslChecking();

        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type","XMLHttpRequest");
        hs.add("Accept","text/plain;charset=utf-8");

        Map<String,String> map = new HashMap<>();

        map.put("id","64d6ce03-3ffa-44b5-9eaf-fad28d09b3e9");
        map.put("project","ystest0101");

        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/manager?session.id={id}&delete=true&project={project}",HttpMethod.GET,new httpentity<String>(hs),String.class,map);

        System.out.println(exchange.getBody());
    }

    /** * 上传zip 上传依赖文件 zip包 * * @throws Exception */
    @Test
    public void uploadZip() throws Exception {

        SSLUtil.turnOffSslChecking();
        FileSystemResource resource = new FileSystemResource(new File("C:\\Users\\Administrator\\Desktop\\ystest.zip"));
        LinkedMultiValueMap<String,Object> linkedMultiValueMap = new LinkedMultiValueMap<String,Object>();
        linkedMultiValueMap.add("session.id","64d6ce03-3ffa-44b5-9eaf-fad28d09b3e9");
        linkedMultiValueMap.add("ajax","upload");
        linkedMultiValueMap.add("project","ystest01");
        linkedMultiValueMap.add("file",resource);
        String postForObject = restTemplate.postForObject(API + "/manager",linkedMultiValueMap,String.class);
        System.out.println(postForObject);
    }

    /** * Fetch Flows of a Project 获取一个project的流ID * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchFlowAProTest() throws KeyManagementException,NoSuchAlgorithmException {

        SSLUtil.turnOffSslChecking();

        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type","lijie_demo");

        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/manager?session.id={id}&ajax=fetchprojectflows&project={project}",map);

        System.out.println(exchange.getBody());

    }

    /** * Fetch Jobs of a Flow 获取一个job的流结构 依赖关系 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchFlowTest() throws KeyManagementException,NoSuchAlgorithmException {
        SSLUtil.turnOffSslChecking();
        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type","lijie_demo");
        map.put("flow","end");

        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/manager?session.id={id}&ajax=fetchflowgraph&project={project}&flow={flow}",map);

        System.out.println(exchange.getBody());

    }

    /** * Fetch Executions of a Flow 获取执行的project 列表 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchEXEFlowTest() throws KeyManagementException,"ystest01");
        map.put("flow","end");
        map.put("start","1");
        map.put("length","30");

        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/manager?session.id={id}&ajax=fetchFlowExecutions&project={project}&flow={flow}&start={start}&length={length}",map);

        System.out.println(exchange.getBody());

    }

    /** * Fetch Running Executions of a Flow 获取正在执行的流id * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchRunningEXEFlowTest() throws KeyManagementException,String> map = new HashMap<>();
        map.put("id","46e144bf-0be8-419e-a200-c83a5461b894");
        map.put("project","end");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=getRunning&project={project}&flow={flow}",map);
        System.out.println(exchange.getBody());
    }

    /** * Execute a Flow 执行一个流 还有很多其他参数 具体参考api * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchEXEaFlowTest() throws KeyManagementException,"end");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=executeFlow&project={project}&flow={flow}",map);
        System.out.println(exchange.getBody());
    }

    /** * Cancel a Flow Execution 中断一个执行流 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void cancelEXEaFlowTest() throws KeyManagementException,"46e144bf-0be8-419e-a200-c83a5461b894");
        map.put("execid","10");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=cancelFlow&execid={execid}",map);
        System.out.println(exchange.getBody());
    }

    /** * Schedule a period-based Flow 创建调度任务 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void scheduleEXEaFlowTest() throws KeyManagementException,"fa47065d-4f00-46ea-bc82-c766a6262886");
        linkedMultiValueMap.add("ajax","scheduleFlow");
        linkedMultiValueMap.add("projectName","ystest01");
        linkedMultiValueMap.add("projectId","14");

        linkedMultiValueMap.add("flow","end");
        linkedMultiValueMap.add("scheduleTime","10,28,am,EDT");
        linkedMultiValueMap.add("scheduleDate","05/16/2017");
        linkedMultiValueMap.add("flowName","ystest01 description");

        // 是否循环
        linkedMultiValueMap.add("is_recurring","on");

        // 循环周期 天 年 月等
        // M Months
        // w Weeks
        // d Days
        // h Hours
        // m Minutes
        // s Seconds
        linkedMultiValueMap.add("period","d");

        httpentity<MultiValueMap<String,hs);
        String postForObject = restTemplate.postForObject(API + "/schedule",String.class);
        System.out.println(postForObject);
    }

    /** * Flexible scheduling using Cron 通过cron表达式调度执行 创建调度任务 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void scheduleByCronEXEaFlowTest() throws KeyManagementException,NoSuchAlgorithmException {
        // Todo 执行有问题
        SSLUtil.turnOffSslChecking();
        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type","ffad7355-4427-4770-9c14-3d19736fa73a");
        linkedMultiValueMap.add("ajax","scheduleCronFlow");
        linkedMultiValueMap.add("projectName","ystest01");
        linkedMultiValueMap.add("cronExpression","* * * * *");
        linkedMultiValueMap.add("flowName","end");

        httpentity<MultiValueMap<String,String.class);
        System.out.println(postForObject);
    }

    /** * Fetch a Schedule 获取一个调度器job的信息 根据project的id 和 flowId * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchScheduleInfoTest() throws KeyManagementException,"46e144bf-0be8-419e-a200-c83a5461b894");
        map.put("projectId","14");
        map.put("flowId","end");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/schedule?session.id={id}&ajax=fetchSchedule&projectId={projectId}&flowId={flowId}",map);
        System.out.println(exchange.getBody());
    }

    /** * Unschedule a Flow 取消一个流的调度 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void unscheduleFlowTest() throws KeyManagementException,"ffad7355-4427-4770-9c14-3d19736fa73a");
        linkedMultiValueMap.add("action","removeSched");
        linkedMultiValueMap.add("scheduleId","4");

        httpentity<MultiValueMap<String,String.class);
        System.out.println(postForObject);
    }

    /** * Set a SLA 设置调度任务 执行的时候 或者执行成功失败等等的规则匹配 发邮件或者... * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void setScheduleFlowSLATest() throws KeyManagementException,"setSla");
        linkedMultiValueMap.add("scheduleId","6");
        linkedMultiValueMap.add("slaEmails","779261177@qq.com");
        linkedMultiValueMap.add("settings[0]","begin,SUCCESS,5:00,true,false");
        linkedMultiValueMap.add("settings[1]","exe,false");
        linkedMultiValueMap.add("settings[2]","end,false");
        // linkedMultiValueMap.add("settings[3]",
        // "xxx,false");

        httpentity<MultiValueMap<String,String.class);
        System.out.println(postForObject);
    }

    /** * Fetch a SLA 获取调度的规则配置 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchScheduleSLAInfoTest() throws KeyManagementException,NoSuchAlgorithmException {

        SSLUtil.turnOffSslChecking();
        HttpHeaders hs = new HttpHeaders();
        hs.add("Content-Type","c4adf192-dcf4-4e05-bd08-f6989dc544a7");
        map.put("scheduleId","6");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/schedule?session.id={id}&ajax=slaInfo&scheduleId={scheduleId}",map);
        System.out.println(exchange.getBody());
    }

    /** * Pause a Flow Execution 暂停一个执行流 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void pauseScheduleTest() throws KeyManagementException,"c4adf192-dcf4-4e05-bd08-f6989dc544a7");
        map.put("execid","12");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=pauseFlow&execid={execid}",map);
        System.out.println(exchange.getBody());
    }

    /** * Resume a Flow Execution 重新执行一个执行流 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void resumeScheduleTest() throws KeyManagementException,"11");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=resumeFlow&execid={execid}",map);
        System.out.println(exchange.getBody());
    }

    /** * Fetch a Flow Execution 获取一个执行流的详细信息 这个流的每个节点的信息 成功或者失败等等 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchFlowInfoTest() throws KeyManagementException,"11");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=fetchexecflow&execid={execid}",map);
        System.out.println(exchange.getBody());
    }

    /** * Fetch Execution Job Logs 获取一个执行流的日志 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchFlowLogTest() throws KeyManagementException,"11");
        map.put("jobId","exe");
        map.put("offset","0");
        map.put("length","100");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=fetchExecJobLogs&execid={execid}&jobId={jobId}&offset={offset}&length={length}",map);
        System.out.println(exchange.getBody());
    }

    /** * Fetch Flow Execution Updates 获取执行流的信息状态 * * @throws KeyManagementException * @throws NoSuchAlgorithmException */
    @Test
    public void fetchFlowUpdateTest() throws KeyManagementException,"11");
        map.put("lastUpdateTime","-1");
        ResponseEntity<String> exchange = restTemplate.exchange(
                API + "/executor?session.id={id}&ajax=fetchexecflowupdate&execid={execid}&lastUpdateTime={lastUpdateTime}",map);
        System.out.println(exchange.getBody());
    }

}

• 3.pom

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <groupId>ys-azkaban</groupId>
    <artifactId>ys-azkaban</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <packaging>war</packaging>
    <properties>
        <spring.version>4.1.3.RELEASE</spring.version>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.apache.httpcomponents</groupId>
            <artifactId>httpclient</artifactId>
            <version>4.3.6</version>
        </dependency>
        <dependency>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-log4j12</artifactId>
            <version>1.6.4</version>
        </dependency>
        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.10</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-context</artifactId>
            <version>${spring.version}</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-beans</artifactId>
            <version>${spring.version}</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-webmvc</artifactId>
            <version>${spring.version}</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-jdbc</artifactId>
            <version>${spring.version}</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-aspects</artifactId>
            <version>${spring.version}</version>
        </dependency>
    </dependencies>
    <build>
        <plugins>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-compiler-plugin</artifactId>
                <configuration>
                    <source>1.8</source>
                    <target>1.8</target>
                </configuration>
            </plugin>
        </plugins>
    </build>
</project>