需求背景
生產環境定時監控凌晨跑批生成文件，並獲取業務匯總信息發送到運維釘釘群。
主要原因還是懶得半夜監控~
變更記錄
- Version 0.0.1 2020/06/08
- 發送釘釘消息，支持 text，markdown 兩種類型消息
選項
sh send-ding.sh [options] <value> ...
* -a <value> 釘釘機器人 Webhook 地址的 access_token
* -t <value> 消息類型：text，markdown
& -T <value> title，首屏會話透出的展示內容；消息類型（-t）為：markdown 時
* -c <value> 消息內容，content或者text
& -m <value> 被@人的手機號（在 content 里添加@人的手機號），多個參數用逗號隔開；如：138xxxx6666，182xxxx8888；與是否@所有人（-A）互斥，僅能選擇一種方式
& -A 是否@所有人，即 isAtAll 參數設置為 ture；與被@人手機號（-m）互斥，僅能選擇一種方式
-v, --version 版本信息
--help 幫助信息
* 表示必輸，& 表示條件必輸，其余為可選
示例
1. 發送 text 消息類型，並@指定人
sh send-ding.sh -a xxx -t text -c "我就是我, 是不一樣的煙火" -m "138xxxx6666,182xxxx8888"
2. 發送 markdown 消息類型，並@所有人
sh send-ding.sh -a xxx -t markdown -T "markdown 測試標題" -c "# 我就是我, 是不一樣的煙火" -A
使用場景
定時監控跑批結果文件生成，發送匯總信息
由於跑批任務大概在凌晨 2:15 分左右完成，故設置 2:20 開始檢測，每 30 分鍾（可調整）釘釘告警一次未獲取到，之后一直檢測，直到檢測到文件生成。
crontab 定時任務設置
$ crontab -e
20 2 * * * sh /ops-scripts/checkPreSettle.sh
定時監控文件並反饋匯總信息腳本 checkPreSettle.sh：
#!/bin/bash
# 定時檢查預對賬文件，並輸出匯總信息
# 昨天日期
DAY=$(date -d yesterday +%Y%m%d)
# 文件路徑
PRE_SETTLE_PATH="/data/${DAY}"
# 預對賬文件
PRE_SETTLE="${PRE_SETTLE_PATH}/${DAY}_pre_settle.csv"
# 預對賬 md5 文件
PRE_SETTLE_MD5="${PRE_SETTLE_PATH}/${DAY}_pre_settle.md5"
# 檢測次數，即 60*30/60秒=30分鍾
COUNT=60
# 間隔掃描時間
SLEEP_TIME=30
# HOSTNAME
HOSTNAME=${HOSTNAME}
# IP
IP=$(ifconfig -a|grep inet|grep -v 127.0.0.1|grep -v inet6|awk '{print $2}')
# send-ding.sh
SEND_DING_SH="/ops-scripts/send-ding.sh"
# 開始時間
START_TIME=$(date "+%Y-%m-%d %H:%M:%S")
# ACCESS_TOKEN
ACCESS_TOKEN=xxxxxxx
# 釘釘消息標題
TILE="預對賬匯總信息\n"
# 計數
i=1
# 循環檢查文件是否存在，SLEEP_TIME 秒檢查一次，總共檢查 COUNT*SLEEP_TIME 秒
while [ $i -le ${COUNT} ];do
if [ -f "${PRE_SETTLE}" ] && [ -f "${PRE_SETTLE}" ]; then
amd5=$(md5sum "${PRE_SETTLE}" | awk '{print $1}')
xmd5=$(cat "${PRE_SETTLE_MD5}")
if [ "X${amd5}" = "X${xmd5}" ]; then
# 獲取文件最后一行
result=$(sed -n '$p' "${PRE_SETTLE}")
# 發送釘釘
sh "${SEND_DING_SH}" -a "${ACCESS_TOKEN}" -t markdown -T "${TILE}" -c "# 【SUCCESS】預對賬匯總信息(條數，結算總金額)：${result}\n #### **HOSTNAME**：${HOSTNAME} \n#### **IP**：${IP} \n#### **文件路徑**：${PRE_SETTLE}\n #### **當前時間**：$(date "+%Y-%m-%d %H:%M:%S")\n" -A
exit 0
fi
fi
# 繼續計數
(( i++ ))
# 間隔 SLEEP_TIME 秒
sleep ${SLEEP_TIME}
done
# 檢測 SLEEP_TIME * COUNT 秒后，仍然沒有檢查到，則告警
sh "${SEND_DING_SH}" -a "${ACCESS_TOKEN}" -t markdown -T "${TILE}" -c "# 【FAIL】經過 ${COUNT}*${SLEEP_TIME} 秒后，仍然未檢查到預對賬文件：${PRE_SETTLE}\n #### **HOSTNAME**：${HOSTNAME} \n#### **IP**：${IP} \n #### **開始檢查時間**：${START_TIME}\n #### **結束檢查時間**：$(date "+%Y-%m-%d %H:%M:%S")\n ## **說明**：會持續檢查（間隔時間：${COUNT}*${SLEEP_TIME} 秒），直到檢查到預對賬文件生成！！！" -A
# 繼續調起
sh "$0"
腳本
#!/bin/bash
#================================================================
# HEADER
#================================================================
# Filename send-ding.sh
# Revision 0.0.1
# Date 2020/06/08
# Author jiangliheng
# Email jiang_liheng@163.com
# Website https://jiangliheng.github.io/
# Description 發送釘釘消息
# Copyright Copyright (c) jiangliheng
# License GNU General Public License
#
#================================================================
#
# Version 0.0.1 2020/06/08
# 發送釘釘消息，支持 text，markdown 兩種類型消息
#
#================================================================
#%名稱(NAME)
#% ${SCRIPT_NAME} - 發送釘釘消息
#%
#%概要(SYNOPSIS)
#% sh ${SCRIPT_NAME} [options] <value> ...
#%
#%描述(DESCRIPTION)
#% 發送釘釘消息
#%
#%選項(OPTIONS)
#% * -a <value> 釘釘機器人 Webhook 地址的 access_token
#% * -t <value> 消息類型：text，markdown
#% & -T <value> title，首屏會話透出的展示內容；消息類型（-t）為：markdown 時
#% * -c <value> 消息內容，content或者text
#% & -m <value> 被@人的手機號（在 content 里添加@人的手機號），多個參數用逗號隔開；如：138xxxx6666，182xxxx8888；與是否@所有人（-A）互斥，僅能選擇一種方式
#% & -A 是否@所有人，即 isAtAll 參數設置為 ture；與被@人手機號（-m）互斥，僅能選擇一種方式
#% -v, --version 版本信息
#% --help 幫助信息
#%
#% * 表示必輸，& 表示條件必輸，其余為可選
#%
#%示例(EXAMPLES)
#%
#% 1. 發送 text 消息類型，並@指定人
#% sh ${SCRIPT_NAME} -a xxx -t text -c "我就是我, 是不一樣的煙火" -m "138xxxx6666,182xxxx8888"
#%
#% 2. 發送 markdown 消息類型，並@所有人
#% sh ${SCRIPT_NAME} -a xxx -t markdown -T "markdown 測試標題" -c "# 我就是我, 是不一樣的煙火" -A
#%
#================================================================
# END_OF_HEADER
#================================================================
# header 總行數
SCRIPT_HEADSIZE=$(head -200 "${0}" |grep -n "^# END_OF_HEADER" | cut -f1 -d:)
# 腳本名稱
SCRIPT_NAME="$(basename "${0}")"
# 版本
VERSION="0.0.1"
# usage
function usage() {
head -"${SCRIPT_HEADSIZE:-99}" "${0}" \
| grep -e "^#%" \
| sed -e "s/^#%//g" -e "s/\${SCRIPT_NAME}/${SCRIPT_NAME}/g" -e "s/\${VERSION}/${VERSION}/g"
}
# 發送 ding 消息
function sendDingMessage() {
curl -s "${1}" -H 'Content-Type: application/json' -d "${2}"
}
# 檢查參數輸入合法性
function checkParameters() {
# -a，-t，-c 參數必輸校驗
if [ -z "${ACCESS_TOKEN}" ] || [ -z "${MSG_TYPE}" ] || [ -z "${CONTENT}" ]
then
printf "Parameter [-a,-t,-c] is required!\n"
exit 1
fi
# -t 為：markdown 時，檢驗參數 -T 必輸
if [ "X${MSG_TYPE}" = "Xmarkdown" ] && [ -z "${TITLE}" ]
then
printf "When [-t] is 'markdown', you must enter the parameter [-T]!\n"
exit 1
fi
# -A 和 -m 互斥，僅能選擇一種方式
if [ "X${IS_AT_ALL}" = "Xtrue" ] && [ -n "${MOBILES}" ]
then
printf "Only one of the parameters [-A] and [-m] can be entered!\n"
exit 1
fi
}
# markdown 消息內容
function markdownMessage() {
# 標題
title=${1}
# 消息內容
text=${2}
# @ 方式
at=${3}
# 判斷是@所有人，還是指定人
if [ -z "${at}" ]; then
atJson=""
elif [ "X${at}" = "Xtrue" ]; then
atJson='"at": {
"isAtAll": true }'
else
# 判斷是否多個手機號
result=$(echo "${at}" | grep ",")
# N 個手機號
if [ "X${result}" != "X" ]; then
# 轉換為手機號數組
mobileArray=(${at//,/ })
# 循環遍歷數組，組織 json 格式字符串
for mobile in "${mobileArray[@]}"
do
mobiles="${mobile}",${mobiles}
# @ 指定人
atMobiles="@${mobile}",${atMobiles}
done
# 1 個手機號
else
mobiles="${at}"
# @ 指定人
atMobiles="@${at}"
fi
# @ json內容
atJson='"at": {
"atMobiles": [
'${mobiles/%,/}'
]
}'
# 內容信息添加 @指定人
text="${text}\n${atMobiles/%,/}"
fi
message='{
"msgtype": "markdown",
"markdown": {
"title":"'${title}'",
"text": "'${text}'"},
'${atJson}'
}'
echo "${message}"
}
# text 消息內容
function textMessage() {
# 消息內容
text=${1}
# @ 方式
at=${2}
# 判斷是@所有人，還是指定人
if [ -z "${at}" ]; then
atJson=""
elif [ "X${at}" = "Xtrue" ]; then
atJson='"at": {
"isAtAll": true }'
else
# 判斷是否多個手機號
result=$(echo "${at}" | grep ",")
# N 個手機號
if [ "X${result}" != "X" ]; then
# 轉換為手機號數組
mobileArray=(${at//,/ })
# 循環遍歷數組，組織 json 格式字符串
for mobile in "${mobileArray[@]}"
do
mobiles="${mobile}",${mobiles}
# @ 指定人
atMobiles="@${mobile}",${atMobiles}
done
# 1 個手機號
else
mobiles="${at}"
# @ 指定人
atMobiles="@${at}"
fi
# @ json內容
atJson='"at": {
"atMobiles": [
'${mobiles/%,/}'
]
}'
# 內容信息添加 @指定人
text="${text}\n${atMobiles/%,/}"
fi
message='{
"msgtype": "text",
"text": {
"content": "'${text}'"},
'${atJson}'
}'
echo "${message}"
}
# 主方法
function main() {
# 檢查參數輸入合法性
checkParameters
# 判斷發送消息類型
case ${MSG_TYPE} in
markdown)
# 判斷 @ 方式
if [ -n "${MOBILES}" ]; then
DING_MESSAGE=$(markdownMessage "${TITLE}" "${CONTENT}" "${MOBILES}")
elif [ -n "${IS_AT_ALL}" ]; then
DING_MESSAGE=$(markdownMessage "${TITLE}" "${CONTENT}" "${IS_AT_ALL}")
else
DING_MESSAGE=$(markdownMessage "${TITLE}" "${CONTENT}")
fi
;;
text)
if [ -n "${MOBILES}" ]; then
DING_MESSAGE=$(textMessage "${CONTENT}" "${MOBILES}")
elif [ -n "${IS_AT_ALL}" ]; then
DING_MESSAGE=$(textMessage "${CONTENT}" "${IS_AT_ALL}")
else
DING_MESSAGE=$(textMessage "${CONTENT}")
fi
;;
*)
printf "Unsupported message type, currently only [text, markdown] are supported!"
exit 1
;;
esac
sendDingMessage "${DING_URL}" "${DING_MESSAGE}"
}
# 判斷參數個數
if [ $# -eq 0 ];
then
usage
exit 1
fi
# getopt 命令行參數
if ! ARGS=$(getopt -o vAa:t:T:c:m: --long help,version -n "${SCRIPT_NAME}" -- "$@")
then
# 無效選項，則退出
exit 1
fi
# 命令行參數格式化
eval set -- "${ARGS}"
while [ -n "$1" ]
do
case "$1" in
-a)
# Webhook access_token
ACCESS_TOKEN=$2
# 釘釘機器人 url 地址
DING_URL="https://oapi.dingtalk.com/robot/send?access_token=${ACCESS_TOKEN}"
shift 2
;;
-t)
MSG_TYPE=$2
shift 2
;;
-T)
TITLE=$2
shift 2
;;
-c)
CONTENT=$2
shift 2
;;
-m)
MOBILES=$2
shift 2
;;
-A)
IS_AT_ALL=true
shift 2
;;
-v|--version)
printf "%s version %s\n" "${SCRIPT_NAME}" "${VERSION}"
exit 1
;;
--help)
usage
exit 1
;;
--)
shift
break
;;
*)
printf "%s is not an option!" "$1"
exit 1
;;
esac
done
main
微信公眾號：daodaotest