自动化运维：编写基本的自动化脚本

Echo_Wish

发布于 2024-09-30 08:07:28

470

发布于 2024-09-30 08:07:28

文章被收录于专栏：数据结构和算法

在现代IT运维中，自动化脚本的使用已经成为提高效率和减少人为错误的关键手段。本文将详细介绍如何编写基本的自动化脚本，帮助运维人员更好地管理和维护系统。我们将以Shell脚本和Python脚本为例，展示如何实现常见的运维任务。

一、自动化脚本的优势

自动化脚本可以显著提高运维工作的效率，减少人为错误，确保任务的稳定执行。通过编写脚本，运维人员可以自动化完成系统监控、备份、日志管理等任务，从而将更多时间投入到更具创造性的工作中。

二、Shell脚本示例

Shell脚本是自动化日常任务的有力工具。以下是几个常见的Shell脚本示例：

检查磁盘使用情况

这个脚本检查系统的磁盘使用情况，并发送邮件通知磁盘使用超过阈值的情况。

#!/bin/bash
THRESHOLD=80
EMAIL="admin@example.com"
df -H | grep -vE '^Filesystem|tmpfs|cdrom' | awk '{ print $5 " " $1 }' | while read output; do
  usep=$(echo $output | awk '{ print $1 }' | cut -d'%' -f1)
  partition=$(echo $output | awk '{ print $2 }')
  if [ $usep -ge $THRESHOLD ]; then
    echo "The partition \"$partition\" has used $usep% at $(date)" | mail -s "Disk Space Alert: $partition" $EMAIL
  fi
done

自动备份MySQL数据库

这个脚本每天自动备份MySQL数据库，并保留最近7天的备份。

#!/bin/bash
BACKUP_DIR="/backup/mysql"
MYSQL_USER="root"
MYSQL_PASSWORD="password"
DATABASE_NAME="mydatabase"

# 创建备份目录
mkdir -p $BACKUP_DIR

# 创建一个新的备份
mysqldump -u $MYSQL_USER -p$MYSQL_PASSWORD $DATABASE_NAME > $BACKUP_DIR/$DATABASE_NAME-$(date +\%F).sql

# 移除超过七天备份文件
find $BACKUP_DIR -type f -mtime +7 -exec rm {} \;

检查并重启宕掉的服务

这个脚本检查指定服务是否运行，如果宕掉则重启该服务并发送通知邮件。

#!/bin/bash
SERVICE="nginx"
EMAIL="admin@example.com"

if ! systemctl is-active --quiet $SERVICE; then
  echo "$SERVICE is down. Attempting to restart..." | mail -s "$SERVICE is down" $EMAIL
  systemctl restart $SERVICE
  if systemctl is-active --quiet $SERVICE; then
    echo "$SERVICE was successfully restarted" | mail -s "$SERVICE restarted" $EMAIL
  else
    echo "Failed to restart $SERVICE" | mail -s "$SERVICE restart failed" $EMAIL
  fi
fi

三、Python脚本示例

Python脚本在自动化运维中也非常流行，尤其适用于复杂任务。以下是一个使用Python实现系统资源监控的示例：

系统资源监控

这个脚本每分钟记录系统的CPU和内存使用情况到日志文件中。

import psutil
import time

LOG_FILE = "/var/log/system_monitor.log"

def log_system_usage():
    with open(LOG_FILE, "a") as log:
        while True:
            cpu_usage = psutil.cpu_percent(interval=1)
            memory_info = psutil.virtual_memory()
            log.write(f"{time.ctime()}: CPU: {cpu_usage}% MEM: {memory_info.percent}%\n")
            time.sleep(60)

if __name__ == "__main__":
    log_system_usage()

自动化任务调度

使用Python的schedule库，可以轻松实现任务调度。以下是一个每小时执行一次备份任务的示例：

import schedule
import time
import os

def backup():
    os.system("mysqldump -u root -p'password' mydatabase > /backup/mysql/mydatabase-$(date +\%F).sql")

schedule.every().hour.do(backup)

while True:
    schedule.run_pending()
    time.sleep(1)