feat(system):优化后端
1.新增后端测试 2.增加了后端的加密 3.增加了i18n(国际化)
This commit is contained in:
@@ -12,24 +12,83 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import os
|
||||
import sys
|
||||
|
||||
from loguru import logger
|
||||
from rich.logging import RichHandler
|
||||
from loguru._logger import Logger
|
||||
|
||||
from kilostar.utils.request_context import get_request_id, get_trace_id
|
||||
|
||||
|
||||
def _is_json_mode() -> bool:
|
||||
"""根据环境变量决定是否启用 JSON 结构化日志。
|
||||
|
||||
支持开关:``KILOSTAR_LOG_FORMAT=json`` 或 ``KILOSTAR_LOG_JSON=1/true``。
|
||||
"""
|
||||
fmt = os.environ.get("KILOSTAR_LOG_FORMAT", "").lower()
|
||||
if fmt == "json":
|
||||
return True
|
||||
flag = os.environ.get("KILOSTAR_LOG_JSON", "").lower()
|
||||
return flag in {"1", "true", "yes", "on"}
|
||||
|
||||
|
||||
def _ctx_patcher(record):
|
||||
"""日志切面:每条日志写出前,把 contextvars 里的 request_id / trace_id 注入。
|
||||
|
||||
显式 ``bind(trace_id=...)`` 的 logger 优先(业务代码可以覆盖切面值);
|
||||
没有 bind 时回退到 contextvars,没有 contextvars 时为空串。
|
||||
"""
|
||||
extra = record["extra"]
|
||||
if not extra.get("trace_id"):
|
||||
extra["trace_id"] = get_trace_id()
|
||||
if not extra.get("request_id"):
|
||||
extra["request_id"] = get_request_id()
|
||||
|
||||
|
||||
def setup_logger() -> Logger:
|
||||
"""初始化全局 loguru logger,输出格式为 ``actor:(...) | trace_id:(...) : message``。"""
|
||||
"""初始化全局 loguru logger。
|
||||
|
||||
- 默认(开发模式):``RichHandler`` 彩色输出,格式 ``actor:(...) | request_id:(...) | trace_id:(...) : message``
|
||||
- JSON 模式(``KILOSTAR_LOG_FORMAT=json``):写到 stdout,每行一条 JSON,便于 ELK/Loki 采集
|
||||
|
||||
request_id / trace_id 来自 ``kilostar.utils.request_context``,由 FastAPI middleware
|
||||
或工作流入口绑定到 contextvars,本模块通过 ``patcher`` 透明注入。
|
||||
"""
|
||||
logger.remove()
|
||||
|
||||
log_level = os.environ.get("KILOSTAR_LOG_LEVEL", "DEBUG").upper()
|
||||
|
||||
if _is_json_mode():
|
||||
logger.configure(
|
||||
extra={"actor_name": "System", "trace_id": "", "request_id": ""},
|
||||
patcher=_ctx_patcher,
|
||||
)
|
||||
logger.add(
|
||||
sys.stdout,
|
||||
serialize=True,
|
||||
level=log_level,
|
||||
enqueue=True,
|
||||
)
|
||||
return logger
|
||||
|
||||
def format_record(record):
|
||||
# Format string for rich handler
|
||||
actor = record["extra"].get("actor_name", "System")
|
||||
trace_id = record["extra"].get("trace_id", "")
|
||||
request_id = record["extra"].get("request_id", "")
|
||||
ids = []
|
||||
if request_id:
|
||||
ids.append(f"request_id:({request_id})")
|
||||
if trace_id:
|
||||
ids.append(f"trace_id:({trace_id})")
|
||||
ids_str = " | " + " | ".join(ids) if ids else ""
|
||||
return f"actor:({actor}){ids_str} : {record['message']}"
|
||||
|
||||
trace_str = f" | trace_id:({trace_id})" if trace_id else ""
|
||||
return f"actor:({actor}){trace_str} : {record['message']}"
|
||||
|
||||
logger.configure(extra={"actor_name": "System", "trace_id": ""})
|
||||
logger.configure(
|
||||
extra={"actor_name": "System", "trace_id": "", "request_id": ""},
|
||||
patcher=_ctx_patcher,
|
||||
)
|
||||
|
||||
logger.add(
|
||||
RichHandler(
|
||||
@@ -40,8 +99,8 @@ def setup_logger() -> Logger:
|
||||
show_path=False,
|
||||
),
|
||||
format=format_record,
|
||||
level="DEBUG",
|
||||
enqueue=True, # 异步记录
|
||||
level=log_level,
|
||||
enqueue=True,
|
||||
)
|
||||
|
||||
return logger
|
||||
@@ -51,5 +110,9 @@ global_logger = setup_logger()
|
||||
|
||||
|
||||
def get_logger(actor_name: str, trace_id: str = "") -> Logger:
|
||||
"""获取一个绑定了 actor_name 与可选 trace_id 的 logger,便于日志按 Actor/请求归类。"""
|
||||
"""获取一个绑定了 actor_name 与可选 trace_id 的 logger,便于日志按 Actor/请求归类。
|
||||
|
||||
若 ``trace_id`` 留空,会回退到 ``contextvars`` 中的当前值(由 middleware 或
|
||||
工作流入口设置)。显式传值则会覆盖切面注入。
|
||||
"""
|
||||
return global_logger.bind(actor_name=actor_name, trace_id=trace_id)
|
||||
|
||||
Reference in New Issue
Block a user