modify scripts
This commit is contained in:
oscarz
75
tools/grammar_stat.sh
Normal file
75
tools/grammar_stat.sh
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# Configuration
|
||||||
|
LOG_FILE="/usr/local/aigrammar/log/app.log"
|
||||||
|
PY_SCRIPT="/home/ubuntu/projects/devops/tools/send_to_wecom.py"
|
||||||
|
TODAY=$(date +%Y-%m-%d)
|
||||||
|
YESTERDAY=$(date -d "yesterday" +%Y-%m-%d)
|
||||||
|
FILTERED_LOG="${LOG_FILE%.log}_${YESTERDAY}.log"
|
||||||
|
|
||||||
|
# Check if log file exists
|
||||||
|
if [ ! -f "$LOG_FILE" ]; then
|
||||||
|
echo "Error: Log file not found - $LOG_FILE" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Check if Python script exists and is executable
|
||||||
|
if [ ! -f "$PY_SCRIPT" ]; then
|
||||||
|
echo "Error: Python script not found or not executable - $PY_SCRIPT" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Create filtered log file
|
||||||
|
echo "Filtering yesterday's logs..."
|
||||||
|
#grep "$TODAY" "$LOG_FILE" > "$FILTERED_LOG"
|
||||||
|
grep "$YESTERDAY" "$LOG_FILE" > "$FILTERED_LOG"
|
||||||
|
|
||||||
|
# Check if filtered log has content
|
||||||
|
if [ ! -s "$FILTERED_LOG" ]; then
|
||||||
|
echo "Warning: No logs found for $YESTERDAY"
|
||||||
|
content="Date: $YESTERDAY\nNo log records found"
|
||||||
|
python3 "$PY_SCRIPT" "$content"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Filtered log created: $FILTERED_LOG"
|
||||||
|
|
||||||
|
# Statistics
|
||||||
|
echo "Calculating statistics..."
|
||||||
|
request_total=$(grep -E "\"level\":\"info\"" "$FILTERED_LOG" | grep "\"func\":\"main.parseToken\"" | wc -l)
|
||||||
|
grammar_total=$(grep -v "\"level\":\"debug\"" "$FILTERED_LOG" | grep "\"func\":\"main.GrammarHandler\"" | wc -l)
|
||||||
|
translate_total=$(grep -v "\"level\":\"debug\"" "$FILTERED_LOG" | grep "\"func\":\"main.TranslateHandler\"" | wc -l)
|
||||||
|
words_total=$(grep -v "\"level\":\"debug\"" "$FILTERED_LOG" | grep "\"func\":\"main.WordsHandler\"" | wc -l)
|
||||||
|
free_limit_total=$(grep -E "\"level\":\"warn\"" "$FILTERED_LOG" | grep "\"func\":\"main.queryUserBenefits\"" | wc -l)
|
||||||
|
error_total=$(grep "\"level\":\"error\"" "$FILTERED_LOG" | wc -l)
|
||||||
|
|
||||||
|
# Generate content to send
|
||||||
|
content="Date: $YESTERDAY
|
||||||
|
Total Requests: $request_total
|
||||||
|
Grammar Check Requests: $grammar_total
|
||||||
|
Translation Requests: $translate_total
|
||||||
|
Words Requests: $words_total
|
||||||
|
Free Quota Exceeded: $free_limit_total
|
||||||
|
Total Errors: $error_total"
|
||||||
|
|
||||||
|
# Print statistics
|
||||||
|
echo "===== Statistics ====="
|
||||||
|
echo -e "$content"
|
||||||
|
|
||||||
|
# Send to WeCom
|
||||||
|
echo "Sending to WeChat Work..."
|
||||||
|
python3 "$PY_SCRIPT" "$content"
|
||||||
|
|
||||||
|
# Check result
|
||||||
|
if [ $? -eq 0 ]; then
|
||||||
|
echo "Successfully sent to WeChat Work"
|
||||||
|
else
|
||||||
|
echo "Error: Failed to send to WeChat Work" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# remove tmp logfile
|
||||||
|
rm -rf "$FILTERED_LOG"
|
||||||
|
|
||||||
|
echo "Script execution completed"
|
||||||
|
exit 0
|
||||||
165
tools/monitor.sh
Normal file
165
tools/monitor.sh
Normal file
@ -0,0 +1,165 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# 颜色定义
|
||||||
|
RED='\033[0;31m'
|
||||||
|
GREEN='\033[0;32m'
|
||||||
|
YELLOW='\033[0;33m'
|
||||||
|
NC='\033[0m' # 无颜色
|
||||||
|
|
||||||
|
# 定义命令映射和函数
|
||||||
|
declare -A COMMAND_MAP
|
||||||
|
declare -a COMMAND_ORDER # 记录命令注册顺序，用于帮助信息
|
||||||
|
|
||||||
|
# 注册命令函数并添加到映射
|
||||||
|
register_command() {
|
||||||
|
local cmd_name=$1
|
||||||
|
local cmd_desc=$2
|
||||||
|
COMMAND_MAP[$cmd_name]=$cmd_desc
|
||||||
|
COMMAND_ORDER+=($cmd_name)
|
||||||
|
}
|
||||||
|
|
||||||
|
# 执行远程命令的通用函数
|
||||||
|
execute_remote() {
|
||||||
|
local server=$1
|
||||||
|
local user=$2
|
||||||
|
local ssh_opts=$3
|
||||||
|
local cmd=$4
|
||||||
|
local desc=$5
|
||||||
|
|
||||||
|
echo -e "${YELLOW}执行命令: ${desc}${NC}"
|
||||||
|
echo "----------------------------------------"
|
||||||
|
echo -e "${YELLOW}服务器: ${user}@${server}${NC}"
|
||||||
|
|
||||||
|
# 执行远程命令
|
||||||
|
RESULT=$(ssh $ssh_opts $user@$server "$cmd")
|
||||||
|
EXIT_CODE=$?
|
||||||
|
|
||||||
|
# 输出结果
|
||||||
|
if [ $EXIT_CODE -eq 0 ]; then
|
||||||
|
echo -e "${GREEN}命令执行成功${NC}"
|
||||||
|
echo "$RESULT"
|
||||||
|
|
||||||
|
# 执行本地脚本并将远程命令结果作为输入
|
||||||
|
python3 ./send_to_wecom.py "$RESULT"
|
||||||
|
return 0
|
||||||
|
else
|
||||||
|
echo -e "${RED}命令执行失败，退出代码: $EXIT_CODE${NC}"
|
||||||
|
return $EXIT_CODE
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# 定义具体命令函数
|
||||||
|
cmd_javbus() {
|
||||||
|
local SERVER="170.106.191.35"
|
||||||
|
local USER="root"
|
||||||
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
||||||
|
|
||||||
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
||||||
|
"cd /root/projects/resources && python3 -m src.monitor.check_javbus" \
|
||||||
|
"检查服务状态"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 定义具体命令函数
|
||||||
|
cmd_check_status() {
|
||||||
|
local SERVER="170.106.191.35"
|
||||||
|
local USER="root"
|
||||||
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
||||||
|
|
||||||
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
||||||
|
"cd ~/resources/aabook/src/ && python3 ./check_status.py" \
|
||||||
|
"检查服务状态"
|
||||||
|
}
|
||||||
|
|
||||||
|
cmd_restart_service() {
|
||||||
|
local SERVER="170.106.191.35"
|
||||||
|
local USER="root"
|
||||||
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
||||||
|
|
||||||
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
||||||
|
"cd ~/resources/aabook/src/ && python3 ./restart_service.py" \
|
||||||
|
"重启服务"
|
||||||
|
}
|
||||||
|
|
||||||
|
cmd_fetch_data() {
|
||||||
|
local SERVER="170.106.191.36" # 示例：使用不同的服务器
|
||||||
|
local USER="admin" # 示例：使用不同的用户
|
||||||
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
||||||
|
|
||||||
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
||||||
|
"cd ~/resources/aabook/src/ && python3 ./fetch_data.py" \
|
||||||
|
"获取最新数据"
|
||||||
|
}
|
||||||
|
|
||||||
|
cmd_backup_db() {
|
||||||
|
local SERVER="170.106.191.35"
|
||||||
|
local USER="root"
|
||||||
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
||||||
|
|
||||||
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
||||||
|
"cd ~/resources/aabook/db/ && ./backup_db.sh" \
|
||||||
|
"备份数据库"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 注册命令
|
||||||
|
register_command "check" "检查服务状态"
|
||||||
|
register_command "restart" "重启服务"
|
||||||
|
register_command "fetch" "获取最新数据"
|
||||||
|
register_command "backup" "备份数据库"
|
||||||
|
register_command "javbus" "检查服务状态"
|
||||||
|
|
||||||
|
# 显示帮助信息
|
||||||
|
show_help() {
|
||||||
|
echo "用法: $0 [命令1] [命令2] [...]"
|
||||||
|
echo "可用命令:"
|
||||||
|
|
||||||
|
for cmd in "${COMMAND_ORDER[@]}"; do
|
||||||
|
printf " %-10s %s\n" "$cmd" "${COMMAND_MAP[$cmd]}"
|
||||||
|
done
|
||||||
|
|
||||||
|
echo
|
||||||
|
echo "示例:"
|
||||||
|
echo " $0 check fetch"
|
||||||
|
echo " $0 backup"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 主函数
|
||||||
|
main() {
|
||||||
|
# 检查是否提供了命令参数
|
||||||
|
if [ $# -eq 0 ]; then
|
||||||
|
show_help
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
local success=true
|
||||||
|
|
||||||
|
# 处理每个命令参数
|
||||||
|
for cmd in "$@"; do
|
||||||
|
# 检查命令是否存在
|
||||||
|
if [[ -z "${COMMAND_MAP[$cmd]}" ]]; then
|
||||||
|
echo -e "${RED}错误: 未知命令 '$cmd'${NC}"
|
||||||
|
show_help
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 执行对应的命令函数
|
||||||
|
cmd_${cmd}
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
success=false
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo
|
||||||
|
done
|
||||||
|
|
||||||
|
# 输出总体执行结果
|
||||||
|
if [ "$success" = true ]; then
|
||||||
|
echo -e "${GREEN}所有命令执行成功${NC}"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo -e "${RED}部分命令执行失败${NC}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 执行主函数
|
||||||
|
main "$@"
|
||||||
Reference in New Issue
Block a user