# Copyright 2015 The Chromium OS Authors. All rights reserved.
|
# Use of this source code is governed by a BSD-style license that can be
|
# found in the LICENSE file.
|
|
import logging
|
import multiprocessing
|
|
import common
|
from autotest_lib.server import test
|
from autotest_lib.server import hosts
|
from autotest_lib.server import crashcollect
|
|
|
def log_collector_dut_worker(dut, job):
|
"""Worker function to collect logs from each DUT in the pool.
|
|
The method called by multiprocessing worker pool for collecting DUT
|
logs. This function is the function which is repeatedly scheduled for each
|
DUT through the multiprocessing worker. This has to be defined outside
|
the class because it needs to be pickleable.
|
|
@param dut: DUTObject representing the DUT.
|
@param job: Autotest job object.
|
"""
|
host = dut.host
|
# Set the job on the host object for log collection.
|
host.job = job
|
logging.info("Collecting logs from: %s", host.hostname)
|
crashcollect.get_crashinfo(host, 0)
|
|
|
class CliqueDUTLogCollector(object):
|
"""CliqueDUTLogCollector is responsible for collecting the relevant logs
|
from all the DUT's in the DUT pool after the test is executed.
|
"""
|
|
def collect_logs(self, dut_objects, job):
|
"""Collects logs from all tall the DUT's in the pool to a provided
|
folder.
|
|
@param dut_objects: An array of DUTObjects corresponding to all the
|
DUT's in the DUT pool.
|
@param job: Autotest job object.
|
"""
|
tasks = []
|
for dut in dut_objects:
|
# Schedule the log collection for this DUT to the log process
|
# pool.
|
task = multiprocessing.Process(
|
target=log_collector_dut_worker,
|
args=(dut, job))
|
tasks.append(task)
|
# Run the log collections in parallel.
|
for task in tasks:
|
task.start()
|
for task in tasks:
|
task.join()
|
