1467766f3d
- OAuth 2.0 서비스 구현 * Authorization Code, Client Credentials, Refresh Token 플로우 지원 * 애플리케이션 등록 및 관리 기능 * 토큰 introspection 및 revocation * SSO 설정 지원 (Google, GitHub, SAML) * 실용적인 스코프 시스템 (user, app, org, api 관리) - 사용자 프로필 기능 확장 * 프로필 사진 및 썸네일 필드 추가 * bio, location, website 등 추가 프로필 정보 * 이메일 인증 및 계정 활성화 상태 관리 * UserPublicResponse 모델 추가 - OAuth 스코프 관리 * picture 스코프 추가 (프로필 사진 접근 제어) * 카테고리별 스코프 정리 (기본 인증, 사용자 데이터, 앱 관리, 조직, API) * 스코프별 승인 필요 여부 설정 - 인프라 개선 * Users 서비스 포트 매핑 추가 (8001) * OAuth 서비스 Docker 구성 (포트 8003) * Kafka 이벤트 통합 (USER_CREATED, USER_UPDATED, USER_DELETED) 🤖 Generated with Claude Code Co-Authored-By: Claude <noreply@anthropic.com>
101 lines
3.5 KiB
Python
101 lines
3.5 KiB
Python
import json
|
|
import asyncio
|
|
from typing import Optional, Dict, Any
|
|
from aiokafka import AIOKafkaProducer
|
|
from aiokafka.errors import KafkaError
|
|
import logging
|
|
|
|
from .events import Event
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class KafkaProducer:
|
|
def __init__(self, bootstrap_servers: str = "kafka:9092"):
|
|
self.bootstrap_servers = bootstrap_servers
|
|
self._producer: Optional[AIOKafkaProducer] = None
|
|
|
|
async def start(self):
|
|
"""Kafka Producer 시작"""
|
|
try:
|
|
self._producer = AIOKafkaProducer(
|
|
bootstrap_servers=self.bootstrap_servers,
|
|
value_serializer=lambda v: json.dumps(v).encode(),
|
|
compression_type="gzip",
|
|
acks='all',
|
|
retry_backoff_ms=100
|
|
)
|
|
await self._producer.start()
|
|
logger.info(f"Kafka Producer started: {self.bootstrap_servers}")
|
|
except Exception as e:
|
|
logger.error(f"Failed to start Kafka Producer: {e}")
|
|
raise
|
|
|
|
async def stop(self):
|
|
"""Kafka Producer 종료"""
|
|
if self._producer:
|
|
await self._producer.stop()
|
|
logger.info("Kafka Producer stopped")
|
|
|
|
async def send_event(self, topic: str, event: Event) -> bool:
|
|
"""이벤트 전송"""
|
|
if not self._producer:
|
|
logger.error("Producer not started")
|
|
return False
|
|
|
|
try:
|
|
event_dict = event.dict()
|
|
event_dict['timestamp'] = event.timestamp.isoformat()
|
|
|
|
await self._producer.send_and_wait(
|
|
topic,
|
|
value=event_dict,
|
|
key=event.correlation_id.encode() if event.correlation_id else None
|
|
)
|
|
|
|
logger.info(f"Event sent to {topic}: {event.event_type}")
|
|
return True
|
|
|
|
except KafkaError as e:
|
|
logger.error(f"Failed to send event to {topic}: {e}")
|
|
return False
|
|
except Exception as e:
|
|
logger.error(f"Unexpected error sending event: {e}")
|
|
return False
|
|
|
|
async def send_batch(self, topic: str, events: list[Event]) -> int:
|
|
"""여러 이벤트를 배치로 전송"""
|
|
if not self._producer:
|
|
logger.error("Producer not started")
|
|
return 0
|
|
|
|
sent_count = 0
|
|
batch = self._producer.create_batch()
|
|
|
|
for event in events:
|
|
event_dict = event.dict()
|
|
event_dict['timestamp'] = event.timestamp.isoformat()
|
|
|
|
metadata = batch.append(
|
|
key=event.correlation_id.encode() if event.correlation_id else None,
|
|
value=json.dumps(event_dict).encode(),
|
|
timestamp=None
|
|
)
|
|
|
|
if metadata is None:
|
|
# 배치가 가득 찼으면 전송하고 새 배치 생성
|
|
await self._producer.send_batch(batch, topic)
|
|
sent_count += len(batch)
|
|
batch = self._producer.create_batch()
|
|
batch.append(
|
|
key=event.correlation_id.encode() if event.correlation_id else None,
|
|
value=json.dumps(event_dict).encode(),
|
|
timestamp=None
|
|
)
|
|
|
|
# 남은 배치 전송
|
|
if batch:
|
|
await self._producer.send_batch(batch, topic)
|
|
sent_count += len(batch)
|
|
|
|
logger.info(f"Sent {sent_count} events to {topic}")
|
|
return sent_count |