AlphaFuturesPro/backend/src/services/datasource/PythonServiceManager.ts

import { spawn } from 'child_process';
import * as path from 'path';
import { config } from '../../config';

class PythonServiceManager {
  private pythonProcess: any = null;
  private isRunning: boolean = false;

  async start(): Promise<boolean> {
    if (this.isRunning) {
      console.log('Python服务已经在运行');
      return true;
    }

    try {
      // 构建Python服务的路径
      const pythonServicePath = path.join(__dirname, '../../../python_service/main.py');
      
      // 从配置中读取端口，默认3007
      const port = config.dataSource?.tqsdk?.pythonPort || 3007;
      
      console.log(`启动Python TQAPI服务，端口: ${port}...`);
      
      // 启动Python服务，传递端口参数
      this.pythonProcess = spawn('python', [pythonServicePath, '--port', port.toString()], {
        cwd: path.join(__dirname, '../../../python_service'),
        stdio: 'inherit',
        shell: true
      });

      this.pythonProcess.on('error', (error: any) => {
        console.error('启动Python服务失败:', error);
        this.isRunning = false;
      });

      this.pythonProcess.on('exit', (code: number, signal: string) => {
        console.log(`Python服务退出，代码: ${code}, 信号: ${signal}`);
        this.isRunning = false;
      });

      // 等待2秒，确保Python服务有时间启动
      await new Promise(resolve => setTimeout(resolve, 2000));
      
      this.isRunning = true;
      console.log('Python TQAPI服务启动成功');
      return true;
    } catch (error) {
      console.error('启动Python服务时出错:', error);
      this.isRunning = false;
      return false;
    }
  }

  stop(): void {
    if (this.pythonProcess) {
      try {
        console.log('停止Python TQAPI服务...');
        // 发送终止信号
        this.pythonProcess.kill();
        // 等待进程退出
        this.pythonProcess.on('exit', (code: number, signal: string) => {
          console.log(`Python服务退出，代码: ${code}, 信号: ${signal}`);
        });
        this.pythonProcess = null;
        this.isRunning = false;
        console.log('Python TQAPI服务已停止');
      } catch (error) {
        console.error('停止Python服务时出错:', error);
        this.pythonProcess = null;
        this.isRunning = false;
      }
    } else {
      console.log('Python服务未运行，无需停止');
    }
  }

  getStatus(): boolean {
    return this.isRunning;
  }
}

// 导出单例实例
export const pythonServiceManager = new PythonServiceManager();