From d2a214cd28d2f0dd2ac392a9c6131d8faecbd663 Mon Sep 17 00:00:00 2001
From: sam <sam@keniutech.cn>
Date: Tue, 7 Oct 2025 22:33:19 +0800
Subject: [PATCH] fix company mapping initialization with flag to prevent
 redundant calls

---
 app/ingest/entity_recognition.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/app/ingest/entity_recognition.py b/app/ingest/entity_recognition.py
index 6e23021..d3ec8aa 100644
--- a/app/ingest/entity_recognition.py
+++ b/app/ingest/entity_recognition.py
@@ -9,6 +9,8 @@ from app.utils.logging import get_logger
 
 LOGGER = get_logger(__name__)
 
+_COMPANY_MAPPING_INITIALIZED = False
+
 # 股票代码正则表达式
 A_SH_CODE_PATTERN = re.compile(r"\b(\d{6})(\.(?:SH|SZ))?\b", re.IGNORECASE)
 HK_CODE_PATTERN = re.compile(r"\b(\d{4})\.HK\b", re.IGNORECASE)
@@ -136,6 +138,11 @@ def initialize_company_mapping(db_connection) -> None:
     Args:
         db_connection: SQLite数据库连接
     """
+    global _COMPANY_MAPPING_INITIALIZED
+    if _COMPANY_MAPPING_INITIALIZED:
+        return
+    _COMPANY_MAPPING_INITIALIZED = True
+
     cursor = db_connection.cursor()
     try:
         cursor.execute(