Files
hr-portal/backend/app/batch/scheduler.py
Porsche Chen 360533393f feat: HR Portal - Complete Multi-Tenant System with Redis Session Storage
Major Features:
-  Multi-tenant architecture (tenant isolation)
-  Employee CRUD with lifecycle management (onboarding/offboarding)
-  Department tree structure with email domain management
-  Company info management (single-record editing)
-  System functions CRUD (permission management)
-  Email account management (multi-account per employee)
-  Keycloak SSO integration (auth.lab.taipei)
-  Redis session storage (10.1.0.254:6379)
  - Solves Cookie 4KB limitation
  - Cross-system session sharing
  - Sliding expiration (8 hours)
  - Automatic token refresh

Technical Stack:
Backend:
- FastAPI + SQLAlchemy
- PostgreSQL 16 (10.1.0.20:5433)
- Keycloak Admin API integration
- Docker Mailserver integration (SSH)
- Alembic migrations

Frontend:
- Next.js 14 (App Router)
- NextAuth 4 with Keycloak Provider
- Redis session storage (ioredis)
- Tailwind CSS

Infrastructure:
- Redis 7 (10.1.0.254:6379) - Session + Cache
- Keycloak 26.1.0 (auth.lab.taipei)
- Docker Mailserver (10.1.0.254)

Architecture Highlights:
- Session管理由 Keycloak + Redis 統一控制
- 支援多系統 (HR/WebMail/Calendar/Drive/Office) 共享 session
- Token 自動刷新,異質服務整合
- 未來可無縫遷移到雲端

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-02-23 20:12:43 +08:00

104 lines
3.0 KiB
Python

"""
批次作業排程器 (5.4)
使用 schedule 套件管理所有批次排程
排程清單:
- 每日 00:00 - auto_terminate_employees (未來實作)
- 每日 02:00 - daily_quota_check
- 每日 03:00 - sync_keycloak_users
- 每月 1 日 01:00 - archive_audit_logs
啟動方式:
python -m app.batch.scheduler
"""
import logging
import signal
import sys
import time
from datetime import datetime
logger = logging.getLogger(__name__)
def _setup_logging():
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
datefmt="%Y-%m-%d %H:%M:%S",
)
def _run_daily_quota_check():
logger.info("觸發: 每日配額檢查批次")
try:
from app.batch.daily_quota_check import run_daily_quota_check
result = run_daily_quota_check()
logger.info(f"每日配額檢查批次完成: {result.get('status')}")
except Exception as e:
logger.error(f"每日配額檢查批次異常: {e}")
def _run_sync_keycloak_users():
logger.info("觸發: Keycloak 同步批次")
try:
from app.batch.sync_keycloak_users import run_sync_keycloak_users
result = run_sync_keycloak_users()
logger.info(f"Keycloak 同步批次完成: {result.get('status')}")
except Exception as e:
logger.error(f"Keycloak 同步批次異常: {e}")
def _run_archive_audit_logs():
"""只在每月 1 日執行"""
if datetime.now().day != 1:
return
logger.info("觸發: 審計日誌歸檔批次 (每月 1 日)")
try:
from app.batch.archive_audit_logs import run_archive_audit_logs
result = run_archive_audit_logs()
logger.info(f"審計日誌歸檔批次完成: {result.get('status')}")
except Exception as e:
logger.error(f"審計日誌歸檔批次異常: {e}")
def start_scheduler():
"""啟動排程器"""
try:
import schedule
except ImportError:
logger.error("缺少 schedule 套件,請執行: pip install schedule")
sys.exit(1)
logger.info("=== HR Portal 批次排程器啟動 ===")
# 每日 02:00 - 配額檢查
schedule.every().day.at("02:00").do(_run_daily_quota_check)
# 每日 03:00 - Keycloak 同步
schedule.every().day.at("03:00").do(_run_sync_keycloak_users)
# 每日 01:00 - 審計日誌歸檔 (函式內部判斷是否為每月 1 日)
schedule.every().day.at("01:00").do(_run_archive_audit_logs)
logger.info("排程設定完成:")
logger.info(" 02:00 - 每日配額檢查")
logger.info(" 03:00 - Keycloak 同步")
logger.info(" 01:00 - 審計日誌歸檔 (每月 1 日)")
# 處理 SIGTERM (Docker 停止信號)
def handle_sigterm(signum, frame):
logger.info("收到停止信號,排程器正在關閉...")
sys.exit(0)
signal.signal(signal.SIGTERM, handle_sigterm)
logger.info("排程器運行中,等待任務觸發...")
while True:
schedule.run_pending()
time.sleep(60) # 每分鐘檢查一次
if __name__ == "__main__":
_setup_logging()
start_scheduler()