当前位置：首页 > news >正文

mT5分类增强版中文-base部署教程：systemd服务封装+自动重启+健康检查配置

news 2026/3/26 19:07:28

mT5分类增强版中文-base部署教程：systemd服务封装+自动重启+健康检查配置

1. 引言：为什么需要专业部署方案

当你成功运行了mT5分类增强版中文模型后，可能会遇到这样的问题：服务运行一段时间后莫名其妙停止，需要手动重新启动；服务器重启后服务不会自动恢复；或者想知道服务是否健康运行。这些都是生产环境部署中常见的问题。

本文将带你从简单的命令行启动，升级到专业的systemd服务部署方案。你将学会如何将mT5文本增强服务封装成系统服务，配置自动重启机制，并设置健康检查，确保服务7×24小时稳定运行。无论你是个人开发者还是运维工程师，这套方案都能让你的AI服务更加可靠。

2. 环境准备与基础部署

2.1 系统要求检查

在开始之前，请确保你的系统满足以下要求：

Ubuntu 18.04+ 或 CentOS 7+ 操作系统
Python 3.8+ 环境
至少8GB内存（推荐16GB）
GPU支持（可选，但推荐用于更好的性能）

2.2 基础服务启动

首先让我们回顾基础启动方式，这是后续配置的基础：

# 进入项目目录 cd /root/nlp_mt5_zero-shot-augment_chinese-base # 使用虚拟环境启动Web服务 /root/nlp_mt5_zero-shot-augment_chinese-base/dpp-env/bin/python /root/nlp_mt5_zero-shot-augment_chinese-base/webui.py

服务启动后，默认会在7860端口提供Web界面和API服务。你可以通过浏览器访问http://你的服务器IP:7860来使用Web界面，或者通过API进行调用。

3. systemd服务封装配置

3.1 创建systemd服务文件

我们将创建一个systemd服务文件来管理mT5服务。首先创建服务配置文件：

sudo nano /etc/systemd/system/mt5-text-augment.service

在文件中添加以下内容：

[Unit] Description=mT5 Text Augmentation Service After=network.target [Service] Type=simple User=root WorkingDirectory=/root/nlp_mt5_zero-shot-augment_chinese-base Environment=PATH=/root/nlp_mt5_zero-shot-augment_chinese-base/dpp-env/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin ExecStart=/root/nlp_mt5_zero-shot-augment_chinese-base/dpp-env/bin/python /root/nlp_mt5_zero-shot-augment_chinese-base/webui.py Restart=always RestartSec=10 StandardOutput=file:/var/log/mt5-service.log StandardError=file:/var/log/mt5-error.log [Install] WantedBy=multi-user.target

3.2 服务管理命令

配置完成后，使用以下命令管理服务：

# 重新加载systemd配置 sudo systemctl daemon-reload # 启动服务 sudo systemctl start mt5-text-augment # 设置开机自启 sudo systemctl enable mt5-text-augment # 查看服务状态 sudo systemctl status mt5-text-augment # 停止服务 sudo systemctl stop mt5-text-augment # 查看服务日志 sudo journalctl -u mt5-text-augment -f

4. 自动重启机制配置

4.1 配置自动重启策略

在上面的service文件中，我们已经配置了基本的重启机制：

Restart=always：服务在任何情况下退出都会重启
RestartSec=10：退出后等待10秒再重启

但对于生产环境，我们可能需要更精细的控制。可以创建重启监控脚本：

nano /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.sh

添加以下内容：

#!/bin/bash SERVICE="mt5-text-augment" PORT=7860 MAX_RESTARTS=10 RESTART_COUNT_FILE="/tmp/mt5_restart_count" # 检查端口是否监听 check_port() { netstat -ltn | grep ":${PORT}" | grep "LISTEN" > /dev/null 2>&1 return $? } # 检查服务是否响应 check_health() { curl -s -o /dev/null -w "%{http_code}" http://localhost:${PORT}/ | grep "200" > /dev/null 2>&1 return $? } # 主检查逻辑 if ! check_port || ! check_health; then # 读取重启计数 if [ -f "$RESTART_COUNT_FILE" ]; then count=$(cat "$RESTART_COUNT_FILE") else count=0 fi if [ "$count" -lt "$MAX_RESTARTS" ]; then echo "$((count + 1))" > "$RESTART_COUNT_FILE" systemctl restart $SERVICE echo "$(date): Service restarted (count: $((count + 1)))" >> /var/log/mt5_monitor.log else echo "$(date): Maximum restart attempts reached" >> /var/log/mt5_monitor.log fi else # 重置重启计数 echo "0" > "$RESTART_COUNT_FILE" fi

4.2 设置定时监控任务

将健康检查脚本设置为定时任务：

# 给脚本执行权限 chmod +x /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.sh # 添加定时任务 (crontab -l 2>/dev/null; echo "*/5 * * * * /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.sh") | crontab -

5. 健康检查方案实现

5.1 API健康检查端点

为了更好地监控服务状态，我们可以添加一个专门的健康检查端点。创建健康检查脚本：

nano /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.py

添加以下内容：

#!/usr/bin/env python3 import requests import json import sys from datetime import datetime def check_service_health(): health_url = "http://localhost:7860/" try: # 检查服务是否响应 response = requests.get(health_url, timeout=10) if response.status_code == 200: # 检查API功能是否正常 test_text = "测试健康检查" api_url = "http://localhost:7860/augment" api_data = { "text": test_text, "num_return_sequences": 1 } api_response = requests.post(api_url, json=api_data, timeout=30) if api_response.status_code == 200: return { "status": "healthy", "timestamp": datetime.now().isoformat(), "response_time": response.elapsed.total_seconds() } else: return { "status": "api_error", "timestamp": datetime.now().isoformat(), "error": f"API returned status {api_response.status_code}" } else: return { "status": "unhealthy", "timestamp": datetime.now().isoformat(), "error": f"Service returned status {response.status_code}" } except requests.exceptions.RequestException as e: return { "status": "unreachable", "timestamp": datetime.now().isoformat(), "error": str(e) } if __name__ == "__main__": health_status = check_service_health() print(json.dumps(health_status, ensure_ascii=False, indent=2)) # 如果状态不健康，返回非零退出码 if health_status["status"] != "healthy": sys.exit(1)

5.2 综合监控仪表板

创建简单的监控脚本，定期检查并记录服务状态：

nano /root/nlp_mt5_zero-shot-augment_chinese-base/monitor_dashboard.py

#!/usr/bin/env python3 import requests import json import time from datetime import datetime class MT5Monitor: def __init__(self): self.base_url = "http://localhost:7860" self.status_file = "/var/log/mt5_status.json" def check_all_endpoints(self): endpoints = { "web_ui": "/", "augment_api": "/augment", "batch_api": "/augment_batch" } results = {} for name, endpoint in endpoints.items(): try: start_time = time.time() if name == "augment_api": response = requests.post( f"{self.base_url}{endpoint}", json={"text": "测试文本", "num_return_sequences": 1}, timeout=30 ) else: response = requests.get(f"{self.base_url}{endpoint}", timeout=10) results[name] = { "status": response.status_code, "response_time": time.time() - start_time, "timestamp": datetime.now().isoformat() } except Exception as e: results[name] = { "status": "error", "error": str(e), "timestamp": datetime.now().isoformat() } return results def save_status(self, status_data): # 读取历史状态 try: with open(self.status_file, 'r') as f: history = json.load(f) except: history = [] # 添加新状态（保留最近100条记录） history.append(status_data) if len(history) > 100: history = history[-100:] with open(self.status_file, 'w') as f: json.dump(history, f, ensure_ascii=False, indent=2) def run_monitoring(self): print("Starting mT5 service monitoring...") while True: status = self.check_all_endpoints() self.save_status(status) # 打印当前状态 print(f"\n[{datetime.now()}] Service Status:") for endpoint, data in status.items(): status_str = "✓" if data.get('status') == 200 else "✗" print(f" {endpoint}: {status_str} ({data.get('response_time', 0):.3f}s)") time.sleep(300) # 每5分钟检查一次 if __name__ == "__main__": monitor = MT5Monitor() monitor.run_monitoring()

6. 完整部署与验证

6.1 完整部署流程

现在让我们执行完整的部署流程：

# 1. 创建systemd服务文件 sudo nano /etc/systemd/system/mt5-text-augment.service # 2. 重新加载systemd配置 sudo systemctl daemon-reload # 3. 启动服务 sudo systemctl start mt5-text-augment # 4. 设置开机自启 sudo systemctl enable mt5-text-augment # 5. 部署健康检查脚本 chmod +x /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.py chmod +x /root/nlp_mt5_zero-shot-augment_chinese-base/monitor_dashboard.py # 6. 设置监控定时任务 (crontab -l 2>/dev/null; echo "*/2 * * * * /root/nlp_mt5_zero-shot-augment_chinese-base/dpp-env/bin/python /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.py >> /var/log/mt5_health.log 2>&1") | crontab -

6.2 服务验证测试

验证服务是否正常部署：

# 检查服务状态 sudo systemctl status mt5-text-augment # 检查服务端口 netstat -ltn | grep 7860 # 测试API接口 curl -X POST http://localhost:7860/augment \ -H "Content-Type: application/json" \ -d '{"text": "测试服务部署", "num_return_sequences": 2}' # 查看服务日志 sudo journalctl -u mt5-text-augment -n 20 -f # 测试健康检查 /root/nlp_mt5_zero-shot-augment_chinese-base/dpp-env/bin/python /root/nlp_mt5_zero-shot-augment_chinese-base/health_check.py