comparison mercurial/worker.py @ 26063:d29859cfcfc2

worker: use multiprocessing to find cpu count

The multiprocessing package was added in Python 2.6.
The implementation of worker.countcpus() is very similar to
multiprocessing.cpu_count(). Ditch our one-off code.

multiprocessing does result in a number of imports. However,
the lazy importer ensures that we don't import anything until
cpu_count() is called. Furthermore, if we are doing something
with multiple cores, chances are the time of that operation
will dwarf the import time, so module bloat isn't a concern
here.

26062:7154a4a08b96

# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

import errno
import os
import signal
import sys
import threading

from .i18n import _
from . import util

def countcpus():
    '''try to count the number of CPUs on the system'''

    # posix
    try:
        n = int(os.sysconf('SC_NPROCESSORS_ONLN'))
        if n > 0:
            return n
    except (AttributeError, ValueError):
        pass

    # windows
    try:
        n = int(os.environ['NUMBER_OF_PROCESSORS'])
        if n > 0:
            return n
    except (KeyError, ValueError):
        pass

    return 1

def _numworkers(ui):
    s = ui.config('worker', 'numcpus')
    if s:
        try:

26063:d29859cfcfc2

# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

import errno
import multiprocessing
import os
import signal
import sys
import threading

from .i18n import _
from . import util

def countcpus():
    '''try to count the number of CPUs on the system'''
    try:
        return multiprocessing.cpu_count()
    except NotImplementedError:
        return 1

def _numworkers(ui):
    s = ui.config('worker', 'numcpus')
    if s:
        try: