Agent.php

Go to the documentation of this file.

<?php
/*
 Author: Daniele Fognini
 SPDX-FileCopyrightText: © 2014-2015 Siemens AG
 
 SPDX-License-Identifier: GPL-2.0-only
*/
 
namespace Fossology\Lib\Agent;
 
use Fossology\Lib\Dao\AgentDao;
use Fossology\Lib\Db\DbManager;
use Symfony\Component\DependencyInjection\ContainerBuilder;
 
require_once(dirname(dirname(__FILE__))."/common-cli.php");
 
define("ALARM_SECS", 30);
 
abstract class Agent
{
  private $agentName;
  private $agentVersion;
  private $agentRev;
  private $agentDesc;
  private $agentArs;
  private $agentId;
 
  protected $userId;
  protected $groupId;
  protected $jobId;
 
  protected $agentSpecifOptions = "";
  protected $agentSpecifLongOptions = array();
 
  protected $args = array();
 
  protected $dbManager;
 
  protected $agentDao;
 
  protected $container;
 
  protected $schedulerMode;
 
  function __construct($agentName, $version, $revision)
  {
    $this->agentName = $agentName;
    $this->agentVersion = $version;
    $this->agentDesc = $agentName. " agent";
    $this->agentRev = $version.".".$revision;
    $this->agentArs = strtolower( $agentName ) . "_ars";
    $this->schedulerMode = false;
 
    $GLOBALS['processed'] = 0;
    $GLOBALS['alive'] = false;
 
    /* initialize the environment */
    cli_Init();
 
    global $container;
    $this->container = $container;
    $this->dbManager = $container->get('db.manager');
    $this->agentDao = $container->get('dao.agent');
 
    $this->agentId = $this->agentDao->getCurrentAgentId($this->agentName, $this->agentDesc, $this->agentRev);
  }
 
  function scheduler_connect()
  {
    $schedulerHandledOpts = "c:";
    $schedulerHandledLongOpts = array("userID:","groupID:","jobId:","scheduler_start",'config:');
 
    $longOpts = array_merge($schedulerHandledLongOpts, $this->agentSpecifLongOptions);
    $shortOpts = $schedulerHandledOpts . $this->agentSpecifOptions;
 
    $args = getopt($shortOpts, $longOpts);
 
    $this->schedulerMode = (array_key_exists("scheduler_start", $args));
 
    $this->userId = $args['userID'];
    $this->groupId = $args['groupID'];
    $this->jobId = $args['jobId'];
 
    unset ($args['jobId']);
    unset ($args['userID']);
    unset ($args['groupID']);
 
    $this->initArsTable();
 
    if ($this->schedulerMode) {
      $this->scheduler_greet();
 
      pcntl_signal(SIGALRM, function($signo)
      {
        Agent::heartbeat_handler($signo);
      });
      pcntl_alarm(ALARM_SECS);
    }
 
    $this->args = $args;
  }
 
  static function heartbeat_handler($signo)
  {
    global $processed;
    global $alive;
 
    echo "HEART: $processed ".($alive ? '1' : '0')."\n";
    $alive = false;
    pcntl_alarm(ALARM_SECS);
  }
 
  function heartbeat($newProcessed)
  {
    if ($this->schedulerMode) {
      global $processed;
      global $alive;
 
      $processed += $newProcessed;
 
      $alive = true;
      pcntl_signal_dispatch();
    }
  }
 
  function bail($exitvalue)
  {
    debug_print_backtrace(DEBUG_BACKTRACE_IGNORE_ARGS);
    $this->scheduler_disconnect($exitvalue);
    throw new \Exception('agent fail in '.__FILE__.':'.__LINE__,$exitvalue);
  }
 
  function scheduler_disconnect($exitvalue)
  {
    if ($this->schedulerMode) {
      Agent::heartbeat_handler(SIGALRM);
      echo "BYE $exitvalue\n";
    }
  }
 
  function scheduler_greet()
  {
    echo "VERSION: ".$this->agentVersion."\n";
    echo "OK\n";
  }
 
  function initArsTable()
  {
    if (!$this->agentDao->arsTableExists($this->agentName)) {
      $this->agentDao->createArsTable($this->agentName);
    }
  }
 
  abstract protected function processUploadId($uploadId);
 
  private function scheduler_current()
  {
    ($line = fgets(STDIN));
    if ("CLOSE\n" === $line) {
      return false;
    }
    if ("END\n" === $line) {
      return false;
    }
 
    return $line;
  }
 
  function run_scheduler_event_loop()
  {
    while (false !== ($line = $this->scheduler_current())) {
      $this->heartbeat(0);
 
      $uploadId = intval($line);
      if ($uploadId <= 0) {
        continue;
      }
 
      $arsId = $this->agentDao->writeArsRecord($this->agentName, $this->agentId, $uploadId);
      if ($arsId<0) {
        print "cannot insert ars record";
        $this->bail(2);
      }
 
      try {
        $success = $this->processUploadId($uploadId);
      } catch(\Exception $e) {
        print "Caught exception while processing uploadId=$uploadId: ".$e->getMessage();
        print $e->getTraceAsString();
        $success = false;
      }
 
      $this->agentDao->writeArsRecord($this->agentName, $this->agentId, $uploadId, $arsId, $success);
 
      if (!$success) {
        print "agent failed on uploadId=$uploadId";
        $this->bail(1);
      }
    }
  }
}