rabbitmq

job_crawler/app/core/config.py

@@ -27,10 +27,13 @@ class ApiConfig(BaseModel):
     tasks: List[TaskConfig] = []
 
 
-class KafkaConfig(BaseModel):
-    bootstrap_servers: str = "localhost:9092"
-    topic: str = "job_data"
-    consumer_group: str = "job_consumer_group"
+class RabbitMQConfig(BaseModel):
+    host: str = "localhost"
+    port: int = 5672
+    username: str = "guest"
+    password: str = "guest"
+    queue: str = "job_data"
+    message_ttl: int = 604800000  # 7天(毫秒)
 
 
 class CrawlerConfig(BaseModel):
@@ -49,7 +52,7 @@ class Settings(BaseModel):
     """应用配置"""
     app: AppConfig = AppConfig()
     api: ApiConfig = ApiConfig()
-    kafka: KafkaConfig = KafkaConfig()
+    rabbitmq: RabbitMQConfig = RabbitMQConfig()
     crawler: CrawlerConfig = CrawlerConfig()
     database: DatabaseConfig = DatabaseConfig()
     
@@ -71,7 +74,7 @@ class Settings(BaseModel):
         return cls(
             app=AppConfig(**data.get('app', {})),
             api=api_config,
-            kafka=KafkaConfig(**data.get('kafka', {})),
+            rabbitmq=RabbitMQConfig(**data.get('rabbitmq', {})),
             crawler=CrawlerConfig(**data.get('crawler', {})),
             database=DatabaseConfig(**data.get('database', {}))
         )