Files
aigrammar/bin/stat.sh
2025-07-11 19:10:34 +08:00

76 lines
2.3 KiB
Bash
Executable File

#!/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