run.sh

#!/bin/bash

# Redis Monitoring Stack 관리 스크립트
# 사용법: ./run.sh [start|stop|restart|status|logs|backup|clean]

set -e

# 색상 코드
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
log_success() { echo -e "${GREEN}[SUCCESS]${NC} $1"; }
log_warning() { echo -e "${YELLOW}[WARNING]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }

# 사용법 출력
usage() {
    echo "사용법: $0 [command]"
    echo
    echo "Commands:"
    echo "  start     - 서비스 시작"
    echo "  stop      - 서비스 중지"
    echo "  restart   - 서비스 재시작"
    echo "  status    - 서비스 상태 확인"
    echo "  logs      - 로그 확인"
    echo "  backup    - 데이터 백업"
    echo "  clean     - 정리 (컨테이너, 볼륨, 이미지)"
    echo "  health    - 헬스체크"
    echo "  monitor   - 실시간 모니터링"
    echo
    exit 1
}

# 서비스 시작
start_services() {
    log_info "서비스 시작 중..."
    docker compose up -d
    log_success "서비스가 시작되었습니다."
    sleep 5
    show_status
}

# 서비스 중지
stop_services() {
    log_info "서비스 중지 중..."
    docker compose down
    log_success "서비스가 중지되었습니다."
}

# 서비스 재시작
restart_services() {
    log_info "서비스 재시작 중..."
    docker compose restart
    log_success "서비스가 재시작되었습니다."
    sleep 5
    show_status
}

# 상태 확인
show_status() {
    log_info "서비스 상태:"
    echo "=================================================="
    docker compose ps
    echo
    
    # 포트 확인
    log_info "포트 사용 상태:"
    echo "Redis Main (6379):"
    netstat -ln | grep :6379 || echo "  포트 미사용"
    echo "Redis Sub (6380):"
    netstat -ln | grep :6380 || echo "  포트 미사용"
    echo "Prometheus (9090):"
    netstat -ln | grep :9090 || echo "  포트 미사용"
    echo
}

# 로그 확인
show_logs() {
    if [ $# -eq 2 ]; then
        log_info "$2 서비스 로그:"
        docker compose logs -f "$2"
    else
        log_info "전체 서비스 로그:"
        docker compose logs -f
    fi
}

# 백업
backup_data() {
    log_info "데이터 백업 시작..."
    if [ -f ./backup.sh ]; then
        ./backup.sh
    else
        log_error "backup.sh 파일이 없습니다. deploy.sh를 먼저 실행해주세요."
        exit 1
    fi
}

# 정리
clean_all() {
    log_warning "모든 컨테이너, 볼륨, 이미지를 정리합니다."
    read -p "계속하시겠습니까? (y/N): " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        log_info "정리 시작..."
        docker compose down -v --remove-orphans
        docker system prune -f
        log_success "정리 완료"
    else
        log_info "정리가 취소되었습니다."
    fi
}

# 헬스체크
health_check() {
    log_info "헬스체크 실행 중..."
    echo "=================================================="
    
    # Redis Main 체크
    echo -n "Redis Main: "
    if docker exec redis-main redis-cli ping &> /dev/null; then
        echo -e "${GREEN}OK${NC}"
    else
        echo -e "${RED}FAIL${NC}"
    fi
    
    # Redis Sub 체크
    echo -n "Redis Sub: "
    if docker exec redis-sub redis-cli ping &> /dev/null; then
        echo -e "${GREEN}OK${NC}"
    else
        echo -e "${RED}FAIL${NC}"
    fi
    
    # Prometheus 체크
    echo -n "Prometheus: "
    if curl -s http://localhost:9090/-/healthy &> /dev/null; then
        echo -e "${GREEN}OK${NC}"
    else
        echo -e "${RED}FAIL${NC}"
    fi
    
    # Redis Exporter Main 체크
    echo -n "Redis Exporter Main: "
    if curl -s http://localhost:9121/metrics &> /dev/null; then
        echo -e "${GREEN}OK${NC}"
    else
        echo -e "${RED}FAIL${NC}"
    fi
    
    # Redis Exporter Sub 체크
    echo -n "Redis Exporter Sub: "
    if curl -s http://localhost:9122/metrics &> /dev/null; then
        echo -e "${GREEN}OK${NC}"
    else
        echo -e "${RED}FAIL${NC}"
    fi
    
    echo "=================================================="
}

# 실시간 모니터링
monitor_services() {
    log_info "실시간 모니터링 시작 (Ctrl+C로 종료)"
    echo "=================================================="
    
    while true; do
        clear
        echo -e "${BLUE}Redis Monitoring Stack - $(date)${NC}"
        echo "=================================================="
        
        # 컨테이너 상태
        docker compose ps
        echo
        
        # 메모리 사용량
        echo "메모리 사용량:"
        docker stats --no-stream --format "table {{.Name}}\t{{.CPUPerc}}\t{{.MemUsage}}\t{{.MemPerc}}" | head -6
        echo
        
        # Redis 정보
        echo "Redis 정보:"
        echo -n "Main - 연결된 클라이언트: "
        docker exec redis-main redis-cli info clients | grep connected_clients | cut -d: -f2 | tr -d '\r' || echo "N/A"
        echo -n "Sub - 연결된 클라이언트: "
        docker exec redis-sub redis-cli info clients | grep connected_clients | cut -d: -f2 | tr -d '\r' || echo "N/A"
        echo
        
        sleep 5
    done
}

# 메인 로직
case "${1:-}" in
    start)
        start_services
        ;;
    stop)
        stop_services
        ;;
    restart)
        restart_services
        ;;
    status)
        show_status
        ;;
    logs)
        show_logs "$@"
        ;;
    backup)
        backup_data
        ;;
    clean)
        clean_all
        ;;
    health)
        health_check
        ;;
    monitor)
        monitor_services
        ;;
    *)
        usage
        ;;
esac