옵투나 공유메모리로로

kis_trader/backtest/optuna_momentum.py

@@ -33,7 +33,7 @@ from kis_trader.backtest.param_search_momentum import (
 from kis_trader.backtest.tail_param_search import _results_dir_for_write
 from kis_trader.engine import momentum_engine as me
 from kis_trader.engine.indicator_cache import attach_indicator_caches_to_params
-from kis_trader.utils.env import get_env_float
+from kis_trader.utils.env import get_env_bool, get_env_float
 
 logger = logging.getLogger("param_search_optuna")
 
@@ -63,6 +63,7 @@ class MomentumSearchContext:
     start_key: str
     end_key: str
     cache_holder: Dict[str, Any] = field(default_factory=dict)
+    shared_tick_store: Any = None  # ws_ticks 공유메모리 핸들 (종료 시 unlink)
 
 
 def prepare_momentum_search_context(
@@ -209,6 +210,25 @@ def prepare_momentum_search_context(
     finally:
         _snap_db.close()
 
+    # ── ws_ticks 공유메모리 (Optuna, opt-in) — dict→numpy 컬럼 shared_memory 로 RAM 절감 ──
+    # 끄려면 OPTUNA_PARAM_SEARCH_SHARED_TICKS=0. numpy/shm 미지원·빌드 실패 시 자동 폴백.
+    shared_tick_store = None
+    if get_env_bool("OPTUNA_PARAM_SEARCH_SHARED_TICKS", True) and ticks_by_code:
+        from kis_trader.backtest.shared_ticks import build_shared_ticks_view
+        _view, shared_tick_store = build_shared_ticks_view(ticks_by_code, enabled=True)
+        if shared_tick_store is not None:
+            import atexit as _atexit
+            _atexit.register(shared_tick_store.unlink)  # 크래시 시 /dev/shm 누수 방지
+            logger.info("📦 ws_ticks 공유메모리 ON (Optuna) — dict 사본 제거, RAM 절감")
+            ticks_by_code = _view
+            import gc as _gc
+            _gc.collect()
+            try:
+                import ctypes as _ctypes
+                _ctypes.CDLL("libc.so.6").malloc_trim(0)
+            except Exception:
+                pass
+
     cache_holder: Dict[str, Any] = {}
     attach_indicator_caches_to_params(cache_holder, codes_candles)
 
@@ -234,6 +254,7 @@ def prepare_momentum_search_context(
         start_key=start_key,
         end_key=end_key,
         cache_holder=cache_holder,
+        shared_tick_store=shared_tick_store,
     )
 
 
@@ -322,7 +343,17 @@ def run_momentum_optuna(
         study_name, n_trials, sort_by,
     )
     t0 = time.time()
-    study.optimize(objective, n_trials=n_trials, n_jobs=n_jobs, show_progress_bar=show_progress)
+    try:
+        study.optimize(objective, n_trials=n_trials, n_jobs=n_jobs, show_progress_bar=show_progress)
+    finally:
+        # 탐색 종료(또는 예외) 시 공유메모리 즉시 해제 (atexit 는 크래시 대비 이중 안전장치).
+        _store = getattr(ctx, "shared_tick_store", None)
+        if _store is not None:
+            try:
+                _store.unlink()
+            except Exception:
+                pass
+            ctx.shared_tick_store = None
     elapsed = time.time() - t0
 
     passing: List[Dict[str, Any]] = []