Custom nail robot alarm

The whole code logic

Source code

assembly.xml

<assembly
        xmlns="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.2"
        xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:schemaLocation="http://maven.apache.org/plugins/maven-assembly-plugin/assembly/1.1.2
   http://maven.apache.org/xsd/assembly-1.1.2.xsd">
    <id>${project.version}</id>

    <formats>
        <format>tar.gz</format>
    </formats>

    <dependencySets>
        <dependencySet>
            <outputDirectory>/lib</outputDirectory>
            <outputFileNameMapping>${artifact.artifactId}.${artifact.extension}</outputFileNameMapping>
            <useProjectArtifact>false</useProjectArtifact>
        </dependencySet>
    </dependencySets>

    <files>
        <file>
            <source>src/main/resources/plugin.properties</source>
            <destName>conf/plugin.properties</destName>
        </file>
        <file>
            <source>${project.build.directory}/${project.artifactId}-${project.version}.jar</source>
            <destName>lib/${project.artifactId}-${project.version}.jar</destName>
        </file>
    </files>
</assembly>

DingAlerter

package com.qf.bigdata.azkaban.alert;

import azkaban.alert.Alerter;
import azkaban.executor.ExecutableFlow;
import azkaban.sla.SlaOption;
import azkaban.utils.Props;
import com.dingtalk.api.DefaultDingTalkClient;
import com.dingtalk.api.request.OapiRobotSendRequest;
import com.taobao.api.ApiException;
import org.apache.log4j.Logger;

import java.text.SimpleDateFormat;
import java.util.Map;

/**
 *  Send a message to nail 
 */
public class DingAlerter implements Alerter {

    private final static Logger LOGGER = Logger.getLogger(DingAlerter.class);
    private String dingServerUrl;
    private Props props;

    public DingAlerter(Props props) {
        this.props = props;
        dingServerUrl = "https://oapi.dingtalk.com/robot/send?access_token=";
    }

    /**
     *  Determine whether the flow is activated 
     */
    public boolean isActived(ExecutableFlow flow, String type) {
        // Get all the parameters in the workflow 
        Map<String, String> parameters = flow.getExecutionOptions().getFlowParameters();
        //  Put together a azkaban Events 
        String property = "ding.alert.on." + type;
        //  Set the default value , If according to ding.alert.on.success You can get the value , Take this value as the criterion , otherwise false
        String defaultActive = props.getString(property, "false");
        //  Get the default value 
        String actived = parameters.getOrDefault(property, defaultActive);
        return Boolean.parseBoolean(actived);
    }

    /**
     *  take long Of ts transformation string Time for 
     */
    private String formatTime(long time) {
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        return dateFormat.format(time);
    }

    /**
     *  Convert the interval to a string representation 
     */
    private String formatDuration(long startTime, long endTime) {
        if (startTime == -1) {
            return "-";
        }
        long durationMs;
        if (endTime == -1) {
            durationMs = System.currentTimeMillis() - startTime;
        } else {
            durationMs = endTime - startTime;
        }

        long seconds = durationMs / 1000;
        if (seconds < 60) {
            return seconds + " sec";
        }

        long minute = seconds / 60;
        seconds %= 60;
        if (minute < 60) {
            return minute + " min " + seconds + " sec";
        }

        long hour = minute / 60;
        minute %= 60;
        if (hour < 24) {
            return hour + "h " + minute +" min ";
        }

        long day = hour / 24;
        hour %= 24;
        return day + " days " + hour + " h";
    }

    /**
     *  from azkaban Get the token to send the nail message in the process of (token)
     */
    private String getPropertiesByFlowParameter(ExecutableFlow flow, String propKey) {
        //  Get the parameters of the whole process 
        Map<String, String> parameters = flow.getExecutionOptions().getFlowParameters();
        String value = props.getString(propKey);
        return parameters.getOrDefault(propKey, value);
    }

    /**
     *  Nail message 
     */
    private void send(ExecutableFlow flow, String title, int execId, String msg) throws Exception {
        //  Get token string 
        String dingdingToken = getPropertiesByFlowParameter(flow, "ding.token");
        //  Get the client sending the message 
        DefaultDingTalkClient client = new DefaultDingTalkClient(dingServerUrl + dingdingToken);
        //  Read azkaban Of host
        String linkHost = getPropertiesByFlowParameter(flow, "ding.link.azkaban.host");
        //  Read keywords 
        String dingAuthWord = getPropertiesByFlowParameter(flow, "ding.auth.word");
        //  Splice an address 
        String linkAddress = linkHost + "/executor?execid=" + execId + "#joblist";
        //  Create request 
        OapiRobotSendRequest request = new OapiRobotSendRequest();
        request.setMsgtype("link");
        OapiRobotSendRequest.Link link = new OapiRobotSendRequest.Link();
        link.setTitle(dingAuthWord + title);
        link.setPicUrl("");
        link.setMessageUrl("http://www.1000phone.com/");
        link.setText(msg);
        request.setLink(link);
        try {
            client.execute(request);
        }catch (ApiException e) {
            String errorMsg = "send dingding message fail " + e.getMessage();
            LOGGER.error(errorMsg);
            throw new ApiException(errorMsg, e);
        }
    }

    /**
     *  Nail message 
     */
    private void sendSla(String title,String msg) throws Exception {
        //  Get token string 
        //String dingdingToken = props.getString("ding.token");
        String dingdingToken="660de36fffae0f1e34bcd47926a70a44afeed1a8e5a15ed057ca7f02c220276a";
        //  Get the client sending the message 
        DefaultDingTalkClient client = new DefaultDingTalkClient(dingServerUrl + dingdingToken);
        //  Read azkaban Of host
        //String linkHost = props.getString("ding.link.azkaban.host");
        String linkHost="http://172.16.53.11:8081";
        //  Read keywords 
        //String dingAuthWord = props.getString("ding.auth.word");
        String dingAuthWord="1000phone";
        //  Create request 
        OapiRobotSendRequest request = new OapiRobotSendRequest();
        request.setMsgtype("link");
        OapiRobotSendRequest.Link link = new OapiRobotSendRequest.Link();
        link.setTitle(dingAuthWord + title);
        link.setPicUrl("");
        link.setMessageUrl("http://www.1000phone.com/");
        link.setText(msg);
        request.setLink(link);
        try {
            client.execute(request);
        }catch (ApiException e) {
            String errorMsg = "send sla dingding message fail " + e.getMessage();
            LOGGER.error(errorMsg);
            throw new ApiException(errorMsg, e);
        }
    }


    @Override
    public void alertOnSuccess(ExecutableFlow flow) throws Exception {
        if (isActived(flow, "success")) {
            int execId = flow.getExecutionId();
            StringBuilder sb = new StringBuilder("Flow 1000phone").append(flow.getFlowId()).append("has success:\n");
            String title = sb.toString();
            sb.append("\t- Start: ").append(formatTime(flow.getStartTime())).append("\n");
            sb.append("\t- End: ").append(formatTime(flow.getEndTime())).append("\n");
            sb.append("\t- Duration: ").append(formatDuration(flow.getStartTime(), flow.getEndTime()));
            send(flow, title, execId, sb.toString());
        }
    }

    @Override
    public void alertOnError(ExecutableFlow flow, String... strings) throws Exception {
        if (isActived(flow, "error")) {
            int execId = flow.getExecutionId();
            StringBuilder sb = new StringBuilder("Flow 1000phone").append(flow.getFlowId()).append("has fail:\n");
            String title = sb.toString();
            sb.append("\t- Start: ").append(formatTime(flow.getStartTime())).append("\n");
            sb.append("\t- End: ").append(formatTime(flow.getEndTime())).append("\n");
            sb.append("\t- Duration: ").append(formatDuration(flow.getStartTime(), flow.getEndTime()));
            send(flow, title, execId, sb.toString());
        }
    }

    @Override
    public void alertOnFirstError(ExecutableFlow flow) throws Exception {
        if (isActived(flow, "first.error")) {
            int execId = flow.getExecutionId();
            StringBuilder sb = new StringBuilder("Flow 1000phone").append(flow.getFlowId()).append("has first fail:\n");
            String title = sb.toString();
            sb.append("\t- Start: ").append(formatTime(flow.getStartTime())).append("\n");
            sb.append("\t- End: ").append(formatTime(flow.getEndTime())).append("\n");
            sb.append("\t- Duration: ").append(formatDuration(flow.getStartTime(), flow.getEndTime()));
            send(flow, title, execId, sb.toString());
        }
    }

    @Override
    public void alertOnSla(SlaOption slaOption, String s) throws Exception {
        sendSla("sla alerter", "sla xxxx:\n" + s);
    }
}

plugin.properties

alerter.name = dingding
alerter.class = com.qf.bigdata.azkaban.alert.DingAlerter

ding.token = 660de36fffae0f1e34bcd47926a70a44afeed1a8e5a15ed057ca7f02c220276a
ding.auth.word = 1000phone
ding.link.azkaban.host = http://172.16.53.11:8081
ding.alert.on.success=true
ding.alert.on.first.error=true
ding.alert.on.error=true

pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<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>com.qf.bigdata</groupId>
    <artifactId>azkaban-dingding</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <azkaban.version>2.5.0</azkaban.version>
        <junit.version>4.13</junit.version>
        <dingtalk-api.version>1.0.0</dingtalk-api.version>
        <slf4j.version>1.6.4</slf4j.version>
        <log4j.version>1.2.16</log4j.version>
    </properties>


    <repositories>
        <repository>
            <id>ali-maven</id>
            <url>http://maven.aliyun.com/nexus/content/groups/public</url>
            <releases>
                <enabled>true</enabled>
            </releases>
            <snapshots>
                <enabled>true</enabled>
                <updatePolicy>always</updatePolicy>
                <checksumPolicy>fail</checksumPolicy>
            </snapshots>
        </repository>
    </repositories>
    <pluginRepositories>
        <pluginRepository>
            <id>ali-maven</id>
            <url>http://maven.aliyun.com/nexus/content/groups/public</url>
        </pluginRepository>
    </pluginRepositories>


    <dependencies>
        <dependency>
            <groupId>com.linkedin.azkaban</groupId>
            <artifactId>azkaban</artifactId>
            <version>${azkaban.version}</version>
            <scope>provided</scope>
        </dependency>

        <dependency>
            <groupId>log4j</groupId>
            <artifactId>log4j</artifactId>
            <version>${log4j.version}</version>
        </dependency>
        <dependency>
            <groupId>org.slf4j</groupId>
            <artifactId>slf4j-log4j12</artifactId>
            <version>${slf4j.version}</version>
        </dependency>

        <dependency>
            <groupId>com.dingtalk</groupId>
            <artifactId>dingtalk-api</artifactId>
            <version>${dingtalk-api.version}</version>
            <scope>compile</scope>
        </dependency>

        <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>${junit.version}</version>
            <scope>test</scope>
        </dependency>

        <dependency>
            <groupId>commons-logging</groupId>
            <artifactId>commons-logging</artifactId>
            <version>1.2</version>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <artifactId>maven-assembly-plugin</artifactId>
                <version>2.6</version>
                <executions>
                    <execution>
                        <id>create-distribution</id>
                        <phase>package</phase>
                        <goals>
                            <goal>single</goal>
                        </goals>
                        <configuration>
                            <descriptors>I
                                <descriptor>src/main/assembly/assembly.xml</descriptor>
                            </descriptors>
                            <finalName>${project.artifactId}</finalName>
                        </configuration>
                    </execution>
                </executions>
            </plugin>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-release-plugin</artifactId>
                <version>2.5.3</version>
                <configuration>
                    <tagNameFormat>@{project.version}</tagNameFormat>
                </configuration>
            </plugin>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-surefire-plugin</artifactId>
                <version>2.5</version>
                <configuration>
                    <forkMode>once</forkMode>
                    <argLine>-Dfile.encoding=UTF-8</argLine>
                    <includes>
                        <include>**/*Test.java</include>
                    </includes>
                    <excludes>
                    </excludes>
                </configuration>
            </plugin>

            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-compiler-plugin</artifactId>
                <configuration>
                    <source>8</source>
                    <target>8</target>
                </configuration>
            </plugin>
        </plugins>
    </build>
</project>

And load your project into azkaban On

 And be sure to specify the warning type when executing 
 namely ：alert.type   dingding

 When the task is wrong or the task is completed, the nailing robot will automatically alarm ( Is to send a message to the group )

If you want to know more , You can go to the beep station to search the video and watch