Тайм-аут и службы Windows для длительных процессов (Python) - PullRequest
2 голосов
/ 28 февраля 2012

У меня есть простой сервис Windows, который я создал с помощью Python. Моя проблема в том, что я не знаю, сколько времени займет завершение службы, это может занять 15 секунд или 4+ часа в зависимости от того, что нужно сделать с данными. 4+ часа - редкий случай, но у меня была ситуация, когда это происходит.

Ниже приведена общая схема, которой я следовал для служб Windows. Я удалил всю логику, но это не проблема, и оставил только фиктивную команду регистрации. Есть ли способ, чтобы служба продолжала или не обновлялась до тех пор, пока логическая часть не будет завершена, вместо использования тайм-аута?

import win32service
import win32serviceutil
import win32api
import win32con
import win32event
import win32evtlogutil
import os
import sys
import time
import logging
class aservice(win32serviceutil.ServiceFramework):
    _svc_name_ = "WeatherService"
    _svc_display_name_ = "Weather Service"
    _svc_description_ = "Downloads weather data from NOAA and creates maps"
    def __init__(self, args):
        win32serviceutil.ServiceFramework.__init__(self, args)
        self.hWaitStop = win32event.CreateEvent(None, 0, 0, None)           
    def SvcStop(self):
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        win32event.SetEvent(self.hWaitStop)                      
    def SvcDoRun(self):
        import servicemanager      
        servicemanager.LogMsg(servicemanager.EVENTLOG_INFORMATION_TYPE,
                              servicemanager.PYS_SERVICE_STARTED,(self._svc_name_, '')) 
        self.timeout = 640000    #640 seconds / 10 minutes (value is in milliseconds)
        #self.timeout = 120000     #120 seconds / 2 minutes
        # This is how long the service will wait to run / refresh itself (see script below)
        while 1:
            # Wait for service stop signal, if I timeout, loop again
            rc = win32event.WaitForSingleObject(self.hWaitStop, self.timeout)
            # Check to see if self.hWaitStop happened
            if rc == win32event.WAIT_OBJECT_0:
                # Stop signal encountered
                servicemanager.LogInfoMsg(self._svc_name_ + " - STOPPED!")  #For Event Log
                break
            else:
                #[actual service code between rests]
                try:
                    logging.basicConfig(filename=r"c:\temp\example.log",level=logging.DEBUG,
                                        format='%(asctime)s %(message)s')
                    logging.debug('This message should go to the log file')
                    logging.info('So should this')
                    logging.warning('And this, too')

                    #file_path = "C:\whereever\my_REAL_py_work_to_be_done.py"
                    #execfile(file_path)             #Execute the script
                    #inc_file_path2 = "C:\whereever\MORE_REAL_py_work_to_be_done.py"
                    #execfile(inc_file_path2)        #Execute the script
                except:
                    pass
                #[actual service code between rests]


def ctrlHandler(ctrlType):
    return True

if __name__ == '__main__':   
    win32api.SetConsoleCtrlHandler(ctrlHandler, True)   
    win32serviceutil.HandleCommandLine(aservice)

Ответы [ 3 ]

0 голосов
/ 28 февраля 2012

Использование сервиса в качестве контроллера с несколькими потоками. Один поток (основной) должен синхронизировать и ставить команды в очередь и использовать служебную структуру win32 для связи, регистрации в системе и т. Д. Другой поток (рабочий) должен ожидать команды из очереди и выполнять их. Если вы выполняете произвольный код как отдельные процессы, вы можете порождать их из рабочего потока и просто читать результаты после их завершения и очищать.

Таким образом, когда приходит остановка, ваш основной поток зарегистрирует его для работника в его очереди, который проснется и попытается дать сигнал другим процессам выйти, немного подождать и очистить или принудительно завершить их. .

Обновление:

Ниже приведен пример концепции того, как вы могли бы иметь сервис, который всегда отзывчив и работает столько времени, сколько необходимо. Каждый работник может бросить

...
import threading
...

class InterruptedException(Exception):
    pass

class WorkerThread(threading.Thread):
    def __init__(self, controller):
        self._controller = controller
        self._stop = threading.Event()
        super(WorkerThread, self).__init__()

    def stop(self):
        self._stop.set()

    def stopped(self):
        return self._stop.isSet()

    def run(self):
        try:
           # Insert the code you want to run as a service here
           # rather than do "execfile(.../.../blah)" simply do:
           # You can have your code throw InterruptedException if your code needs to exit
           # Also check often if self.stopped and then cleanly exit

           import your_file
           your_file.main()

           # if code in another module is not yours or cannot check often if it should stop then use multiprocessing which will spawn separate processes that you can terminate then from here when you need to stop and return
           # in that case simply block here on self._stop.wait()

        except InterruptedException as exc:
           # We are forcefully quitting 
           pass
        except Exception as e:
           # Oh oh, did not anticipate this, better report to Windows or log it
        finally:
           # Close/release any connections, handles, files etc.

           # OK, we can stop now
           win32event.SetEvent(self._controller)

def __init__(self, args):
    win32serviceutil.ServiceFramework.__init__(self, args)
    self.hWaitStop = win32event.CreateEvent(None, 0, 0, None)      
    self.hWaitDone = win32event.CreateEvent(None, 0, 0, None)

def SvcDoRun(self):
    import servicemanager      
    servicemanager.LogMsg(servicemanager.EVENTLOG_INFORMATION_TYPE,
                          servicemanager.PYS_SERVICE_STARTED,(self._svc_name_, '')) 

    worker = WorkerThread(self.hWaitDone)
    worker.start()

    while True:
        # Wait for service stop signal
        rc = win32event.WaitForMultipleObjects([self.hWaitStop, self.hWaitDone], win32event.INFINITE)

        # Check to see if self.hWaitStop happened as part of Windows Service Management
        if rc == 0:
            # Stop signal encountered
            servicemanager.LogInfoMsg(self._svc_name_ + " - STOPPED!")  #For Event Log
            break

        if rc == 1:
            # Wait until worker has fully finished
            worker.join()

            # Determine from worker state if we need to start again (because run finished)
            # Or do whatever
            if not worker.need_to_start_again():
                break

            worker.start()
0 голосов
/ 29 февраля 2012

В итоге я использовал этот шаблон: http://code.activestate.com/recipes/551780/

Работает хорошо, но не как шарм. У меня есть проблемы с многопроцессорностью, процесс не порождает экземпляры. Предложения на что?

Пожалуйста, продолжайте публиковать ваши ответы, я хотел бы видеть решения каждого, потому что с win32api может быть трудно работать.

Спасибо всем

0 голосов
/ 28 февраля 2012

Вы можете начать новый процесс, чтобы делать долгосрочные вещи. Если поступает сигнал остановки, вы убиваете дочерний процесс.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...