First version, does not close correctly
bzr revid: nicolas.vanhoren@openerp.com-20130606123733-j6d2lpcgrvxfa7yr
This commit is contained in:
parent
c8719220a7
commit
5d5cca0041
|
@ -257,10 +257,14 @@ def main(args):
|
|||
setup_pid_file()
|
||||
# Some module register themselves when they are loaded so we need the
|
||||
# services to be running before loading any registry.
|
||||
if not openerp.evented:
|
||||
if config['workers']:
|
||||
openerp.service.start_services_workers()
|
||||
else:
|
||||
openerp.service.start_services()
|
||||
else:
|
||||
config['xmlrpc_port'] = config['longpolling_port']
|
||||
openerp.service.start_services()
|
||||
|
||||
rc = 0
|
||||
if config['db_name']:
|
||||
|
|
|
@ -14,6 +14,8 @@ import signal
|
|||
import socket
|
||||
import sys
|
||||
import time
|
||||
import subprocess
|
||||
import os.path
|
||||
|
||||
import werkzeug.serving
|
||||
try:
|
||||
|
@ -35,7 +37,6 @@ class Multicorn(object):
|
|||
def __init__(self, app):
|
||||
# config
|
||||
self.address = (config['xmlrpc_interface'] or '0.0.0.0', config['xmlrpc_port'])
|
||||
self.long_polling_address = (config['xmlrpc_interface'] or '0.0.0.0', config['longpolling_port'])
|
||||
self.population = config['workers']
|
||||
self.timeout = config['limit_time_real']
|
||||
self.limit_request = config['limit_request']
|
||||
|
@ -46,10 +47,10 @@ class Multicorn(object):
|
|||
self.socket = None
|
||||
self.workers_http = {}
|
||||
self.workers_cron = {}
|
||||
self.workers_longpolling = {}
|
||||
self.workers = {}
|
||||
self.generation = 0
|
||||
self.queue = []
|
||||
self.long_polling_pid = None
|
||||
|
||||
def pipe_new(self):
|
||||
pipe = os.pipe()
|
||||
|
@ -89,6 +90,14 @@ class Multicorn(object):
|
|||
worker.run()
|
||||
sys.exit(0)
|
||||
|
||||
def long_polling_spawn(self):
|
||||
nargs = [] + sys.argv
|
||||
cmd = nargs[0]
|
||||
cmd = os.path.join(os.path.dirname(cmd), "long_polling")
|
||||
nargs[0] = cmd
|
||||
popen = subprocess.Popen(nargs)
|
||||
self.long_polling_pid = popen.pid
|
||||
|
||||
def worker_pop(self, pid):
|
||||
if pid in self.workers:
|
||||
_logger.debug("Worker (%s) unregistered",pid)
|
||||
|
@ -143,8 +152,8 @@ class Multicorn(object):
|
|||
self.worker_spawn(WorkerHTTP, self.workers_http)
|
||||
while len(self.workers_cron) < config['max_cron_threads']:
|
||||
self.worker_spawn(WorkerCron, self.workers_cron)
|
||||
while len(self.workers_longpolling) < 1:
|
||||
self.worker_spawn(WorkerLongPolling, self.workers_longpolling)
|
||||
if not self.long_polling_pid:
|
||||
self.long_polling_spawn()
|
||||
|
||||
def sleep(self):
|
||||
try:
|
||||
|
@ -183,14 +192,11 @@ class Multicorn(object):
|
|||
self.socket.setblocking(0)
|
||||
self.socket.bind(self.address)
|
||||
self.socket.listen(8*self.population)
|
||||
# long polling socket
|
||||
self.long_polling_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.long_polling_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
self.long_polling_socket.setblocking(0)
|
||||
self.long_polling_socket.bind(self.long_polling_address)
|
||||
self.long_polling_socket.listen(8)
|
||||
|
||||
def stop(self, graceful=True):
|
||||
if self.long_polling_pid is not None:
|
||||
self.worker_kill(self.long_polling_pid, signal.SIGTERM)
|
||||
self.long_polling_pid = None
|
||||
if graceful:
|
||||
_logger.info("Stopping gracefully")
|
||||
limit = time.time() + self.timeout
|
||||
|
@ -353,49 +359,8 @@ class WorkerHTTP(Worker):
|
|||
|
||||
def start(self):
|
||||
Worker.start(self)
|
||||
self.multi.long_polling_socket.close()
|
||||
self.server = WorkerBaseWSGIServer(self.multi.app)
|
||||
|
||||
class WorkerLongPolling(Worker):
|
||||
""" Long polling workers """
|
||||
def __init__(self, multi):
|
||||
super(WorkerLongPolling, self).__init__(multi)
|
||||
# Disable the watchdog feature for this kind of worker.
|
||||
self.watchdog_timeout = None
|
||||
|
||||
def watch_parent(self):
|
||||
import gevent
|
||||
while True:
|
||||
if self.ppid != os.getppid():
|
||||
_logger.info("WorkerLongPolling (%s) Parent changed", self.pid)
|
||||
os.kill(os.getpid(), signal.SIGTERM)
|
||||
return
|
||||
gevent.sleep(self.multi.beat)
|
||||
|
||||
def start(self):
|
||||
openerp.evented = True
|
||||
_logger.info('Using gevent mode')
|
||||
import gevent.monkey
|
||||
gevent.monkey.patch_all()
|
||||
import gevent_psycopg2
|
||||
gevent_psycopg2.monkey_patch()
|
||||
from openerp.modules.registry import RegistryManager
|
||||
from gevent.coros import RLock
|
||||
RegistryManager.registries_lock = RLock()
|
||||
|
||||
Worker.start(self)
|
||||
self.multi.socket.close()
|
||||
|
||||
import gevent
|
||||
watcher = gevent.spawn(self.watch_parent)
|
||||
|
||||
log = _logger.getChild(self.__class__.__name__)
|
||||
log.write = lambda msg: log.info(msg.strip())
|
||||
|
||||
from gevent.wsgi import WSGIServer
|
||||
self.server = WSGIServer(self.multi.long_polling_socket, self.multi.app, log=log)
|
||||
self.server.serve_forever()
|
||||
|
||||
class WorkerBaseWSGIServer(werkzeug.serving.BaseWSGIServer):
|
||||
""" werkzeug WSGI Server patched to allow using an external listen socket
|
||||
"""
|
||||
|
@ -469,7 +434,6 @@ class WorkerCron(Worker):
|
|||
def start(self):
|
||||
Worker.start(self)
|
||||
self.multi.socket.close()
|
||||
self.multi.long_polling_socket.close()
|
||||
openerp.service.start_internal()
|
||||
|
||||
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|
||||
|
|
Loading…
Reference in New Issue