344 lines
12 KiB
Python
344 lines
12 KiB
Python
import logging
|
||
import socket
|
||
from typing import Dict, Any
|
||
|
||
import yaml
|
||
from pydantic import Field
|
||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||
from v2.nacos import ClientConfigBuilder, GRPCConfig, NacosConfigService, ConfigParam, NacosNamingService, RegisterInstanceParam, DeregisterInstanceParam
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# ====================== Nacos 配置 ======================
|
||
NACOS_SERVER_ADDRESSES = "18.167.251.121:28848"
|
||
NACOS_NAMESPACE = "zcr"
|
||
NACOS_USERNAME = "nacos"
|
||
NACOS_PASSWORD = "Aidlab123123!"
|
||
NACOS_GROUP = "LOCAL"
|
||
NACOS_DATA_ID = "aida.python"
|
||
SERVICE_NAME = "fastapi-service" # ←←← 必须修改！建议格式：项目名-环境，例如 ai-image-service-dev
|
||
|
||
|
||
class Settings(BaseSettings):
|
||
"""
|
||
应用配置类。Pydantic Settings 会自动从环境变量和 .env 文件中加载这些值。
|
||
"""
|
||
model_config = SettingsConfigDict(
|
||
env_file='.env',
|
||
env_file_encoding='utf-8',
|
||
# extra='ignore' # 忽略环境变量中多余的键
|
||
)
|
||
# --- 服务端口配置信息 ---
|
||
PORT: int = Field(default=8001, description="")
|
||
# --- 服务环境 配置信息 ---
|
||
SERVE_ENV: str = Field(default='', description="")
|
||
# --- 开发状态 配置信息 ---
|
||
DEBUG: bool = Field(default=False, description="")
|
||
# --- 千问api 配置信息 ---
|
||
QWEN_API_KEY: str = Field(default="", description="")
|
||
|
||
# --- ComfyUI 配置信息 ---
|
||
COMFYUI_SERVER_ADDRESS: str = Field(default='', description="")
|
||
|
||
# --- minio 配置信息 ---
|
||
MINIO_URL: str = Field(default='', description="")
|
||
MINIO_ACCESS: str = Field(default='', description="")
|
||
MINIO_SECRET: str = Field(default='', description="")
|
||
MINIO_SECURE: bool = Field(default=True, description="")
|
||
|
||
# --- redis 配置信息 ---
|
||
REDIS_HOST: str = Field(default='', description="")
|
||
REDIS_PORT: str = Field(default='', description="")
|
||
REDIS_DB: int = Field(default=0, description="")
|
||
|
||
# --- mysql 配置信息 ---
|
||
MYSQL_HOST: str = Field(default='', description="")
|
||
MYSQL_PORT: int = Field(default=3306, description="")
|
||
MYSQL_USER: str = Field(default='', description="")
|
||
MYSQL_PASSWORD: str = Field(default='', description="")
|
||
MYSQL_DB: str = Field(default='', description="")
|
||
MYSQL_CHARSET: str = Field(default='utf8mb4', description="")
|
||
|
||
# --- rabbit-mq 配置信息 ---
|
||
MQ_HOST: str = Field(default='', description="")
|
||
MQ_PORT: str = Field(default='', description="")
|
||
MQ_USERNAME: str = Field(default='', description="")
|
||
MQ_PASSWORD: str = Field(default='', description="")
|
||
MQ_VIRTUAL_HOST: str = Field(default='/', description="")
|
||
MQ_ENV: str = Field(default='', description="")
|
||
|
||
# --- milvus 配置信息 ---
|
||
MILVUS_URL: str = Field(default='', description="")
|
||
MILVUS_TOKEN: str = Field(default='', description="")
|
||
MILVUS_ALIAS: str = Field(default='', description="")
|
||
|
||
# --- ollama 配置信息 ---
|
||
CHROMADB_PATH: str = Field(default='', description="")
|
||
|
||
# --- ollama 配置信息 ---
|
||
OLLAMA_URL: str = Field(default='', description="")
|
||
|
||
# --- Design Callback Java 接口 ---
|
||
JAVA_STREAM_API_URL: str = Field(default='', description="")
|
||
|
||
# --- flux2 klein model url ---
|
||
FLUX2_GEN_IMG_MODEL_URL: str = Field(default='', description="")
|
||
|
||
# --- 服务器IP ---
|
||
A6000_SERVICE_HOST: str = Field(default='', description="")
|
||
B_4_X_4090_SERVICE_HOST: str = Field(default='', description="")
|
||
|
||
# --- 其他配置信息 以下均为Docker容器内配置---
|
||
LOGS_PATH: str = Field(default="/logs/", description="")
|
||
CATEGORY_PATH: str = Field(default="/app/service/attribute/config/descriptor/category/category_dis.csv", description="")
|
||
SEG_CACHE_PATH: str = Field(default="/seg_cache/", description="")
|
||
RECOMMEND_PATH_PREFIX: str = Field(default="/app/service/recommend/", description="")
|
||
SERVE_PORT: int = Field(default=2010, description="")
|
||
|
||
|
||
settings = Settings()
|
||
|
||
# ====================== Nacos 配置管理 ======================
|
||
|
||
client_config = (ClientConfigBuilder()
|
||
.server_address(NACOS_SERVER_ADDRESSES)
|
||
.username(NACOS_USERNAME)
|
||
.password(NACOS_PASSWORD)
|
||
.namespace_id(NACOS_NAMESPACE)
|
||
.log_level('INFO')
|
||
.grpc_config(GRPCConfig(grpc_timeout=5000))
|
||
.build())
|
||
|
||
# ====================== Nacos 配置管理 ======================
|
||
nacos_config_data: Dict[str, Any] = {}
|
||
nacos_config_client = None
|
||
async def load_nacos_config() -> None:
|
||
"""初始化 Nacos 配置并监听变化"""
|
||
global nacos_config_data, settings
|
||
|
||
try:
|
||
client = await NacosConfigService.create_config_service(client_config)
|
||
|
||
# 1. 第一次获取配置
|
||
content = await client.get_config(ConfigParam(
|
||
data_id=NACOS_DATA_ID,
|
||
group=NACOS_GROUP
|
||
))
|
||
|
||
if content:
|
||
loaded = yaml.safe_load(content) or {}
|
||
nacos_config_data = loaded
|
||
# 用 Nacos 配置覆盖 settings
|
||
for key, value in loaded.items():
|
||
if hasattr(settings, key):
|
||
setattr(settings, key, value)
|
||
logger.info(f"✅ Nacos 配置加载成功: {NACOS_DATA_ID} | 覆盖字段数量: {len(loaded)}")
|
||
else:
|
||
logger.warning("Nacos 返回配置为空，使用 .env + 默认值")
|
||
|
||
# 2. 注册动态监听器（配置变更自动刷新）
|
||
async def listener(tenant: str, data_id: str, group: str, content: str):
|
||
global nacos_config_data, settings
|
||
try:
|
||
new_config = yaml.safe_load(content) if content else {}
|
||
nacos_config_data = new_config
|
||
# 实时覆盖 settings
|
||
for key, value in new_config.items():
|
||
if hasattr(settings, key):
|
||
old_val = getattr(settings, key)
|
||
setattr(settings, key, value)
|
||
if old_val != value:
|
||
logger.info(f"🔄 配置更新 → {key}: {old_val} → {value}")
|
||
logger.info(f"【Nacos 动态更新】{NACOS_DATA_ID}")
|
||
except Exception as e:
|
||
logger.error(f"Nacos 配置解析失败: {e}")
|
||
|
||
await client.add_listener(NACOS_DATA_ID, NACOS_GROUP, listener)
|
||
logger.info("✅ Nacos 配置监听器已注册（支持热更新）")
|
||
|
||
await register_service_to_nacos()
|
||
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ Nacos 初始化失败: {e}，将仅使用 .env 配置")
|
||
|
||
|
||
async def register_service_to_nacos():
|
||
"""启动时把服务注册到 Nacos"""
|
||
global nacos_config_client
|
||
|
||
nacos_config_client = await NacosConfigService.create_config_service(client_config)
|
||
|
||
if not nacos_config_client: # 如果配置客户端都没连上，就不注册
|
||
logger.warning("Nacos 配置客户端未初始化，跳过服务注册")
|
||
return
|
||
|
||
try:
|
||
nacos_naming_client = await NacosNamingService.create_naming_service(client_config)
|
||
|
||
# 获取服务 IP（生产环境建议通过环境变量传入，避免 Docker/K8s 内获取错误）
|
||
host_ip = socket.gethostbyname(socket.gethostname())
|
||
if not host_ip or host_ip.startswith('127.'):
|
||
host_ip = "127.0.0.1" # 本地测试用
|
||
|
||
param = RegisterInstanceParam(
|
||
service_name="aida.python",
|
||
group_name=NACOS_GROUP,
|
||
ip=host_ip,
|
||
port=settings.PORT, # 使用你 settings 中的 PORT
|
||
cluster_name="DEFAULT",
|
||
weight=1.0,
|
||
metadata={
|
||
"version": "1.0.0",
|
||
"env": settings.SERVE_ENV,
|
||
"framework": "fastapi",
|
||
"debug": str(settings.DEBUG),
|
||
},
|
||
enabled=True,
|
||
healthy=True,
|
||
ephemeral=True, # 临时实例，推荐生产使用
|
||
)
|
||
|
||
await nacos_naming_client.register_instance(request=param)
|
||
logger.info(f"✅ 服务已成功注册到 Nacos！ → {SERVICE_NAME} | {host_ip}:{settings.PORT} | env={settings.SERVE_ENV}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 服务注册到 Nacos 失败: {e}")
|
||
|
||
|
||
async def deregister_service_from_nacos():
|
||
"""服务关闭时优雅注销（防止 Nacos 长时间显示不健康实例）"""
|
||
try:
|
||
nacos_naming_client = await NacosNamingService.create_naming_service(client_config)
|
||
|
||
host_ip = socket.gethostbyname(socket.gethostname()) or "127.0.0.1"
|
||
param = DeregisterInstanceParam(
|
||
service_name=SERVICE_NAME,
|
||
group_name=NACOS_GROUP,
|
||
ip=host_ip,
|
||
port=settings.PORT,
|
||
cluster_name='c1',
|
||
ephemeral=True,
|
||
)
|
||
await nacos_naming_client.deregister_instance(request=param)
|
||
logger.info(f"✅ 服务已从 Nacos 注销 → {SERVICE_NAME}")
|
||
except Exception as e:
|
||
logger.warning(f"服务注销时出现异常（通常可忽略）: {e}")
|
||
|
||
|
||
|
||
|
||
# 提供给 FastAPI 的依赖
|
||
def get_settings() -> Settings:
|
||
return settings
|
||
|
||
|
||
"""Design 服务"""
|
||
# 推荐服装类别映射
|
||
TABLE_CATEGORIES = {
|
||
"female_dress": "female/dress",
|
||
"female_outwear": "female/outwear",
|
||
"female_trousers": "female/trousers",
|
||
"female_skirt": "female/skirt",
|
||
"female_blouse": "female/blouse",
|
||
"male_tops": "male/tops",
|
||
"male_bottoms": "male/bottoms",
|
||
"male_outwear": "male/outwear"
|
||
}
|
||
# Design前后排优先级
|
||
PRIORITY_DICT = {
|
||
'earring_front': 99,
|
||
'bag_front': 98,
|
||
'hairstyle_front': 97,
|
||
'outwear_front': 20,
|
||
'tops_front': 19,
|
||
'dress_front': 18,
|
||
'blouse_front': 17,
|
||
'skirt_front': 16,
|
||
'trousers_front': 15,
|
||
'bottoms_front': 14,
|
||
'shoes_right': 1,
|
||
'shoes_left': 1,
|
||
'body': 0,
|
||
'bottoms_back': -14,
|
||
'trousers_back': -15,
|
||
'skirt_back': -16,
|
||
'blouse_back': -17,
|
||
'dress_back': -18,
|
||
'tops_back': -19,
|
||
'outwear_back': -20,
|
||
'hairstyle_back': -97,
|
||
'bag_back': -98,
|
||
'earring_back': -99,
|
||
}
|
||
# Design 关键点字段
|
||
KEYPOINT_RESULT_TABLE_FIELD_SET = ('neckline_left', 'neckline_right', 'shoulder_left', 'shoulder_right', 'armpit_left', 'armpit_right', 'cuff_left_in', 'cuff_left_out', 'cuff_right_in', 'cuff_right_out', 'waistband_left', 'waistband_right')
|
||
# milvus配置信息
|
||
MILVUS_TABLE_KEYPOINT = "keypoint_cache_2"
|
||
|
||
# ollama 地址
|
||
OLLAMA_URL = f"http://{settings.A6000_SERVICE_HOST}:11434/api/embeddings"
|
||
|
||
"""Triton Server Config"""
|
||
# Design
|
||
DESIGN_MODEL_URL = f'{settings.A6000_SERVICE_HOST}:10000'
|
||
DESIGN_MODEL_NAME = 'seg_knet'
|
||
# Seg Product
|
||
SEG_PRODUCT_MODEL_URL = f'{settings.B_4_X_4090_SERVICE_HOST}:30000'
|
||
# Generate Image
|
||
GI_MODEL_URL = f'{settings.A6000_SERVICE_HOST}:10061'
|
||
GI_MODEL_NAME = 'flux'
|
||
# Generate Single Logo
|
||
GSL_MODEL_URL = f'{settings.B_4_X_4090_SERVICE_HOST}:10041'
|
||
GSL_MODEL_NAME = 'stable_diffusion_xl_transparent'
|
||
# Generate Product （整套和单品）
|
||
GPI_MODEL_URL = f'{settings.B_4_X_4090_SERVICE_HOST}:10051'
|
||
GPI_MODEL_NAME_OVERALL = 'diffusion_ensemble_all'
|
||
GPI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_cnet'
|
||
|
||
# 以下停用中...*************
|
||
# 多视角生成
|
||
GMV_MODEL_URL = f'{settings.B_4_X_4090_SERVICE_HOST}:10081'
|
||
GMV_MODEL_NAME = 'multi_view'
|
||
# 超分
|
||
SR_MODEL_NAME = "super_resolution"
|
||
SR_TRITON_URL = f"{settings.A6000_SERVICE_HOST}:10031"
|
||
# 打光
|
||
GRI_MODEL_URL = f'{settings.A6000_SERVICE_HOST}:10051'
|
||
GRI_MODEL_NAME_OVERALL = 'diffusion_relight_ensemble'
|
||
GRI_MODEL_NAME_SINGLE = 'stable_diffusion_1_5_relight'
|
||
# agent 图片生成
|
||
FAST_GI_MODEL_URL = f'{settings.B_4_X_4090_SERVICE_HOST}:10011'
|
||
FAST_GI_MODEL_NAME = 'stable_diffusion_xl'
|
||
# 图转视频 triton版
|
||
PT_MODEL_URL = f'{settings.B_4_X_4090_SERVICE_HOST}:10061'
|
||
|
||
# *************
|
||
|
||
"""MQ 队列信息"""
|
||
# 生成图片 moodboard printboard sketchboard
|
||
GI_RABBITMQ_QUEUES = f"GenerateImage-{settings.SERVE_ENV}"
|
||
# 生成slogan
|
||
SLOGAN_RABBITMQ_QUEUES = f"Slogan-{settings.SERVE_ENV}"
|
||
# 转产品图
|
||
GPI_RABBITMQ_QUEUES = f"ToProductImage-{settings.SERVE_ENV}"
|
||
# 产品图转视频
|
||
PS_RABBITMQ_QUEUES = f"PoseTransform-{settings.SERVE_ENV}"
|
||
|
||
# 以下停用中...*************
|
||
# 产品图打光
|
||
GRI_RABBITMQ_QUEUES = f"Relight-{settings.SERVE_ENV}"
|
||
# 超分
|
||
SR_RABBITMQ_QUEUES = f"SuperResolution-{settings.SERVE_ENV}"
|
||
# 生成多视图
|
||
GMV_RABBITMQ_QUEUES = f"GenerateMultiView-{settings.SERVE_ENV}"
|
||
# 批量转产品图
|
||
BATCH_GPI_RABBITMQ_QUEUES = f"BatchToProductImage-{settings.SERVE_ENV}"
|
||
# 批量打光
|
||
BATCH_GRI_RABBITMQ_QUEUES = f"BatchRelight-{settings.SERVE_ENV}"
|
||
# 批量图片转视频
|
||
BATCH_PS_RABBITMQ_QUEUES = f"BatchPoseTransform-{settings.SERVE_ENV}"
|
||
# 批量design
|
||
BATCH_DESIGN_RABBITMQ_QUEUES = f"DesignBatch-{settings.SERVE_ENV}"
|
||
# *************
|