Blame view

scripts/stop_clip_api.sh 2.28 KB
40f1e391   tangwang   cnclip
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
  #!/bin/bash
  
  ###############################################################################
  # CN-CLIP REST API 停止脚本
  #
  # 用途:
  #   停止 REST API 服务
  #
  # 使用方法:
  #   ./scripts/stop_clip_api.sh
  #
  ###############################################################################
  
  set -e
  
  # 颜色定义
  RED='\033[0;31m'
  GREEN='\033[0;32m'
  YELLOW='\033[1;33m'
  BLUE='\033[0;34m'
  NC='\033[0m'
  
  # 项目路径
  PROJECT_ROOT="/data/tw/SearchEngine"
  PID_FILE="${PROJECT_ROOT}/logs/clip_rest_api.pid"
  
  echo -e "${BLUE}========================================${NC}"
  echo -e "${BLUE}停止 CN-CLIP REST API 服务${NC}"
  echo -e "${BLUE}========================================${NC}"
  echo ""
  
  # 检查 PID 文件
  if [ ! -f "${PID_FILE}" ]; then
      echo -e "${YELLOW}警告: 未找到 PID 文件${NC}"
      echo -e "${YELLOW}REST API 服务可能未运行${NC}"
  
      # 尝试查找并终止进程
      if ps aux | grep "clip_rest_api.py" | grep -v grep > /dev/null; then
          echo -e "${YELLOW}发现运行中的 API 进程${NC}"
          API_PIDS=$(ps aux | grep "clip_rest_api.py" | grep -v grep | awk '{print $2}')
          echo -e "${YELLOW}正在终止...${NC}"
          for PID in ${API_PIDS}; do
              kill ${PID} 2>/dev/null || true
          done
          sleep 1
          echo -e "${GREEN}✓ 进程已终止${NC}"
      fi
      exit 0
  fi
  
  # 读取 PID
  PID="$(cat "${PID_FILE}")"
  
  # 检查进程
  if ps -p "${PID}" > /dev/null 2>&1; then
      echo -e "${BLUE}服务信息:${NC}"
      echo "  PID: ${PID}"
      echo ""
  
      echo -e "${YELLOW}正在停止服务...${NC}"
      kill "${PID}" || true
      sleep 2
  
      # 检查是否还在运行
      if ps -p "${PID}" > /dev/null 2>&1; then
          echo -e "${YELLOW}进程仍在运行,强制终止...${NC}"
          kill -9 "${PID}" || true
          sleep 1
      fi
  
      # 最终检查
      if ps -p "${PID}" > /dev/null 2>&1; then
          echo -e "${RED}错误: 无法停止进程${NC}"
          exit 1
      else
          echo -e "${GREEN}========================================${NC}"
          echo -e "${GREEN}✓ REST API 服务已停止${NC}"
          echo -e "${GREEN}========================================${NC}"
      fi
  else
      echo -e "${YELLOW}警告: 进程 ${PID} 不存在${NC}"
  fi
  
  # 清理 PID 文件
  rm -f "${PID_FILE}"
  echo ""
  echo -e "${GREEN}PID 文件已删除${NC}"