Python 中使用 win32serviceutil.ServiceFramework 设置 Windows 服务的多实例命名

2024-03-21 21:21:53

使用 win32serviceutil.ServiceFramework 设置 Windows 服务的多实例命名

在当今复杂的 IT 环境中，我们经常需要在 Windows 系统上运行多个相同服务的实例。然而，在 Windows 服务的传统实现中，每个服务只能有一个实例。这对于需要根据特定配置或数据集同时运行多个服务实例的应用程序来说是一个限制。

为了解决此问题，我们可以使用 Python 中的 win32serviceutil.ServiceFramework 库来设置 Windows 服务的多实例命名。通过这种方法，我们可以为每个服务实例分配一个不同的名称，使其能够并行运行并针对不同的数据集工作。

步骤详解

要使用 win32serviceutil.ServiceFramework 设置具有多实例命名的 Windows 服务，我们需要遵循以下步骤：

定义服务类： 创建一个子类化自 win32serviceutil.ServiceFramework 的类，并指定服务名称、显示名称和。
在构造函数中初始化服务： 在服务类的构造函数中，初始化服务所需的任何设置，例如服务路径、事件句柄和子进程对象。
重写 SvcDoRun 方法： 在这个方法中，定义服务在运行时执行的操作。通常情况下，这包括启动子进程并等待它完成。
重写 SvcStop 方法： 在这个方法中，定义服务在停止时执行的操作。通常情况下，这包括停止子进程并关闭服务。
处理命令行参数： 使用 argparse 模块解析命令行参数，以获取要安装的服务的名称。
动态设置服务名称： 根据解析的名称动态设置服务名称、显示名称和。
调用 HandleCommandLine： 使用 win32serviceutil.HandleCommandLine 处理命令行，并使用指定的类名启动服务。

示例代码

以下是实现上述步骤的示例代码：

import win32serviceutil
import win32service
import win32event
import os
import sys
import argparse
import subprocess

AUTOROUTER_SCRIPT = r'\simple_test.py'
PYTHON_EXE = 'python'

class ARService(win32serviceutil.ServiceFramework):

    _svc_name_ = None
    _svc_display_name_ = None
    _svc_description_ = None

    def __init__(self, args):
        win32serviceutil.ServiceFramework.__init__(self, args)

        self.ar_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), AUTOROUTER_SCRIPT)

        self.hWaitStop = win32event.CreateEvent(None, 0, 0, None)
        self.autorouter_process = None
        
    def SvcDoRun(self):
        command = [PYTHON_EXE, self.ar_path, self._svc_name]
        self.autorouter_process = subprocess.Popen(command)
        win32event.WaitForSingleObject(self.hWaitStop, win32event.INFINITE)

    def SvcStop(self):
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        command = [PYTHON_EXE, self.ar_path, self._svc_name, '--halt']
        halt_process = subprocess.Popen(command)
        halt_process.wait()
        self.autorouter_process.wait()
        self.autorouter_process = None
        win32event.SetEvent(self.hWaitStop)


if __name__ == '__main__':
    
    parser = argparse.ArgumentParser()
    parser.add_argument('name', help='Name daemon by schema or projid')
    
    args, unknown_args = parser.parse_known_args()

    service_name = 'AutoRouter' + args.name
    service_display_name = 'AutoRouter for ' + args.name
    service_description = 'AutoRouter for ' + args.name
    
    ARService._svc_name_ = service_name
    ARService._svc_display_name_ = service_display_name
    ARService._svc_description_ = service_description
    
    # Remove name argument from sys.argv before calling HandleCommandLine
    sys.argv[1:] = unknown_args
    
    win32serviceutil.HandleCommandLine(ARService)