From patchwork Thu Feb 23 13:47:19 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: Thu, 23 Feb 2012 13:47:19 -0000 From: Dongxiao Xu X-Patchwork-Id: 21711 Message-Id: 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()