การออกแบบระบบ API Key Multi-tenant พร้อมการแยกสิทธิ์แบบปลอดภัย

ในระบบ SaaS สมัยใหม่ การจัดการ API Key หลายผู้เช่า (Multi-tenant) อย่างปลอดภัยเป็นความท้าทายสำคัญ บทความนี้จะพาคุณเรียนรู้วิธีออกแบบระบบที่แยกข้อมูลระหว่างลูกค้าแต่ละรายได้อย่างชัดเจน โดยใช้ HolySheep AI เป็นตัวอย่างการปฏิบัติจริง

สถานการณ์ข้อผิดพลาดจริง: 401 Unauthorized - Tenant Isolation Breach

นักพัฒนาหลายคนเจอปัญหานี้เมื่อระบบ multi-tenant ไม่แยกสิทธิ์ถูกต้อง:

Error: 401 Unauthorized
{
  "error": {
    "message": "API key does not have access to this resource",
    "type": "invalid_request_error",
    "code": "tenant_access_denied"
  }
}

ปัญหานี้เกิดจากการออกแบบ permission layer ที่ไม่ดี ในบทความนี้เราจะแก้ไขทุกปัญหาที่เกี่ยวข้อง

โครงสร้างพื้นฐาน Multi-tenant API Key

การออกแบบระบบ API Key ที่รองรับ multi-tenant ต้องคำนึงถึง 3 องค์ประกอบหลัก ดังนี้

• Tenant Isolation - แยกข้อมูลระหว่างผู้เช่าแต่ละราย
• Role-based Access Control (RBAC) - กำหนดสิทธิ์ตามบทบาท
• Rate Limiting ต่อ Tenant - จำกัดการใช้งานตามแผน

การสร้าง API Key Manager ด้วย Python

ตัวอย่างโค้ดนี้แสดงการสร้าง API Key Manager ที่รองรับ multi-tenant:

import hashlib
import secrets
from datetime import datetime, timedelta
from typing import Optional, Dict, List

class TenantKeyManager:
    def __init__(self):
        self.tenant_keys: Dict[str, Dict] = {}
        self.tenant_permissions: Dict[str, List[str]] = {}
    
    def generate_api_key(self, tenant_id: str, permissions: List[str]) -> str:
        """สร้าง API Key ใหม่สำหรับ tenant เฉพาะ"""
        key_prefix = f"sk_holysheep_{tenant_id[:8]}"
        random_part = secrets.token_urlsafe(32)
        api_key = f"{key_prefix}_{random_part}"
        
        key_hash = hashlib.sha256(api_key.encode()).hexdigest()
        
        self.tenant_keys[key_hash] = {
            "tenant_id": tenant_id,
            "permissions": permissions,
            "created_at": datetime.now(),
            "is_active": True
        }
        
        self.tenant_permissions[tenant_id] = permissions
        return api_key
    
    def validate_key(self, api_key: str, required_permission: str) -> bool:
        """ตรวจสอบ API Key และสิทธิ์"""
        key_hash = hashlib.sha256(api_key.encode()).hexdigest()
        
        if key_hash not in self.tenant_keys:
            return False
        
        key_info = self.tenant_keys[key_hash]
        
        if not key_info["is_active"]:
            return False
        
        if required_permission not in key_info["permissions"]:
            return False
        
        return True

การใช้งาน
manager = TenantKeyManager()
api_key = manager.generate_api_key(
    tenant_id="tenant_abc123",
    permissions=["chat:read", "chat:write", "embedding:read"]
)
print(f"Generated API Key: {api_key}")

การเรียกใช้ HolySheep AI API ด้วย Tenant-aware Client

ตัวอย่างการสร้าง client ที่รองรับ multi-tenant อย่างถูกต้อง:

import requests
from typing import Dict, Optional, Any

class HolySheepMultiTenantClient:
    BASE_URL = "https://api.holysheep.ai/v1"
    
    def __init__(self, api_key: str, tenant_id: str):
        self.api_key = api_key
        self.tenant_id = tenant_id
        self.session = requests.Session()
        self.session.headers.update({
            "Authorization": f"Bearer {api_key}",
            "X-Tenant-ID": tenant_id,
            "Content-Type": "application/json"
        })
    
    def chat_completions(
        self, 
        model: str, 
        messages: List[Dict],
        max_tokens: int = 1000
    ) -> Dict[str, Any]:
        """เรียกใช้ Chat Completions API พร้อม tenant isolation"""
        url = f"{self.BASE_URL}/chat/completions"
        
        payload = {
            "model": model,
            "messages": messages,
            "max_tokens": max_tokens
        }
        
        response = self.session.post(url, json=payload)
        
        if response.status_code == 401:
            raise PermissionError("API key ไม่มีสิทธิ์เข้าถึง tenant นี้")
        elif response.status_code == 429:
            raise RateLimitError("เกิน rate limit ของ tenant")
        
        response.raise_for_status()
        return response.json()

การใช้งาน
client = HolySheepMultiTenantClient(
    api_key="YOUR_HOLYSHEEP_API_KEY",
    tenant_id="customer_xyz789"
)

response = client.chat_completions(
    model="gpt-4.1",
    messages=[{"role": "user", "content": "สวัสดีครับ"}]
)
print(response)

ข้อผิดพลาดที่พบบ่อยและวิธีแก้ไข

1. 401 Unauthorized - Invalid Tenant Context

# สาเหตุ: API Key ไม่มีสิทธิ์เข้าถึง tenant ที่ระบุ
วิธีแก้ไข: ตรวจสอบว่า X-Tenant-ID header ตรงกับ tenant ของ API Key

client = HolySheepMultiTenantClient(
    api_key="YOUR_HOLYSHEEP_API_KEY",
    tenant_id="tenant_abc123"  # ต้องตรงกับ tenant ที่สร้าง key
)

2. 403 Forbidden - Insufficient Permissions

สาเหตุ: API Key ไม่มีสิทธิ์เพียงพอสำหรับ operation ที่เรียกใช้

วิธีแก้ไข:

• ตรวจสอบสิทธิ์ที่กำหนดให้กับ API Key
• สร้าง API Key ใหม่พร้อมสิทธิ์ที่เหมาะสม
• ตรวจสอบ permission scope ที่ API endpoint ต้องการ

3. 429 Rate Limit Exceeded

สาเหตุ: เกินโควต้าการใช้งานที่กำหนดไว้สำหรับ tenant

วิธีแก้ไข:

• ใช้ exponential backoff สำหรับ retry
• ตรวจสอบ rate limit headers จาก response
• พิจารณาอัปเกรดแผนการใช้งาน

import time

def call_with_retry(client, payload, max_retries=3):
    for attempt in range(max_retries):
        try:
            return client.chat_completions(**payload)
        except RateLimitError:
            wait_time = 2 ** attempt  # exponential backoff
            time.sleep(wait_time)
    raise Exception("เกินจำนวนครั้งที่กำหนด")

การใช้ Middleware สำหรับ Tenant Isolation

สำหรับ FastAPI หรือ Flask application สามารถใช้ middleware เพื่อตรวจสอบ tenant isolation:

from functools import wraps
from flask import request, jsonify

def tenant_isolation_middleware(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        api_key = request.headers.get("Authorization", "").replace("Bearer ", "")
        tenant_id = request.headers.get("X-Tenant-ID", "")
        
        # ตรวจสอบว่า API Key ตรงกับ tenant
        key_tenant = extract_tenant_from_key(api_key)
        
        if key_tenant != tenant_id:
            return jsonify({
                "error": "Tenant isolation violation",
                "message": "API key ไม่ตรงกับ tenant ที่ระบุ"
            }), 403
        
        return f(*args, **kwargs)
    return decorated_function

@app.route("/api/v1/resource")
@tenant_isolation_middleware
def get_resource():
    return jsonify({"data": "resource data"})

ข้อแนะนำในการตั้งค่า API Key บน HolySheep AI

เมื่อใช้ HolySheep AI สำหรับ multi-tenant application ควรตั้งค่าดังนี้

• แยก API Key ต่อ tenant - อย่าใช้ API Key เดียวสำหรับหลาย tenant
• กำหนด Rate Limit ต่อ Key - ป้องกันปัญหา tenant หนึ่งใช้งานมากเกินไป
• ใช้ Webhook สำหรับ Audit - ติดตามการใช้งานแต่ละ tenant
• ตรวจสอบ Token Usage - ดูรายงานการใช้งานแยกตาม tenant

สรุป

การออกแบบระบบ API Key multi