From patchwork Fri Jan 13 08:30:41 2012 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit Subject: [bitbake-devel, 7/9] command.py: add new API to get the cpu count on the server Date: Fri, 13 Jan 2012 08:30:41 -0000 From: Dongxiao Xu X-Patchwork-Id: 19263 Message-Id: <44369e95249adee9818d8a79b723eb27f08cbdbc.1326442655.git.dongxiao.xu@intel.com> To: bitbake-devel@lists.openembedded.org From: Shane Wang Add a new API in command.py to get the cpu count in order to set the appropriate default BB_NUMBER_THREADS and PARALLEL_MAKE variables. Signed-off-by: Shane Wang --- lib/bb/command.py | 6 ++++++ lib/bb/utils.py | 4 ++++ 2 files changed, 10 insertions(+), 0 deletions(-) diff --git a/lib/bb/command.py b/lib/bb/command.py index 05555c5..43875f7 100644 --- a/lib/bb/command.py +++ b/lib/bb/command.py @@ -173,6 +173,12 @@ class CommandsSync: """ command.cooker.reset() + def getCpuCount(self, command, params): + """ + Get the CPU count on the bitbake server + """ + return bb.utils.cpu_count() + class CommandsAsync: """ diff --git a/lib/bb/utils.py b/lib/bb/utils.py index a26635a..d7cefb2 100644 --- a/lib/bb/utils.py +++ b/lib/bb/utils.py @@ -25,6 +25,7 @@ import errno import logging import bb import bb.msg +import multiprocessing from commands import getstatusoutput from contextlib import contextmanager @@ -862,3 +863,6 @@ def contains(variable, checkvalues, truevalue, falsevalue, d): if checkvalues.issubset(val): return truevalue return falsevalue + +def cpu_count(): + return multiprocessing.cpu_count()