src/applications/fact/chart/PhabricatorFactChartFunction.php

   1 <?php
   2
   3 final class PhabricatorFactChartFunction
   4   extends PhabricatorChartFunction {
   5
   6   const FUNCTIONKEY = 'fact';
   7
   8   private $fact;
   9   private $map;
  10   private $refs;
  11
  12   protected function newArguments() {
  13     $key_argument = $this->newArgument()
  14       ->setName('fact-key')
  15       ->setType('fact-key');
  16
  17     $parser = $this->getArgumentParser();
  18     $parser->parseArgument($key_argument);
  19
  20     $fact = $this->getArgument('fact-key');
  21     $this->fact = $fact;
  22
  23     return $fact->getFunctionArguments();
  24   }
  25
  26   public function loadData() {
  27     $fact = $this->fact;
  28
  29     $key_id = id(new PhabricatorFactKeyDimension())
  30       ->newDimensionID($fact->getKey());
  31     if (!$key_id) {
  32       $this->map = array();
  33       return;
  34     }
  35
  36     $table = $fact->newDatapoint();
  37     $conn = $table->establishConnection('r');
  38     $table_name = $table->getTableName();
  39
  40     $where = array();
  41
  42     $where[] = qsprintf(
  43       $conn,
  44       'keyID = %d',
  45       $key_id);
  46
  47     $parser = $this->getArgumentParser();
  48
  49     $parts = $fact->buildWhereClauseParts($conn, $parser);
  50     foreach ($parts as $part) {
  51       $where[] = $part;
  52     }
  53
  54     $data = queryfx_all(
  55       $conn,
  56       'SELECT id, value, epoch FROM %T WHERE %LA ORDER BY epoch ASC',
  57       $table_name,
  58       $where);
  59
  60     $map = array();
  61     $refs = array();
  62     if ($data) {
  63       foreach ($data as $row) {
  64         $ref = (string)$row['id'];
  65         $value = (int)$row['value'];
  66         $epoch = (int)$row['epoch'];
  67
  68         if (!isset($map[$epoch])) {
  69           $map[$epoch] = 0;
  70         }
  71
  72         $map[$epoch] += $value;
  73
  74         if (!isset($refs[$epoch])) {
  75           $refs[$epoch] = array();
  76         }
  77
  78         $refs[$epoch][] = $ref;
  79       }
  80     }
  81
  82     $this->map = $map;
  83     $this->refs = $refs;
  84   }
  85
  86   public function getDomain() {
  87     $min = head_key($this->map);
  88     $max = last_key($this->map);
  89
  90     return new PhabricatorChartInterval($min, $max);
  91   }
  92
  93   public function newInputValues(PhabricatorChartDataQuery $query) {
  94     return array_keys($this->map);
  95   }
  96
  97   public function evaluateFunction(array $xv) {
  98     $map = $this->map;
  99
 100     $yv = array();
 101
 102     foreach ($xv as $x) {
 103       if (isset($map[$x])) {
 104         $yv[] = $map[$x];
 105       } else {
 106         $yv[] = null;
 107       }
 108     }
 109
 110     return $yv;
 111   }
 112
 113   public function getDataRefs(array $xv) {
 114     return array_select_keys($this->refs, $xv);
 115   }
 116
 117   public function loadRefs(array $refs) {
 118     $fact = $this->fact;
 119
 120     $datapoint_table = $fact->newDatapoint();
 121     $conn = $datapoint_table->establishConnection('r');
 122
 123     $dimension_table = new PhabricatorFactObjectDimension();
 124
 125     $where = array();
 126
 127     $where[] = qsprintf(
 128       $conn,
 129       'p.id IN (%Ld)',
 130       $refs);
 131
 132
 133     $rows = queryfx_all(
 134       $conn,
 135       'SELECT
 136           p.id id,
 137           p.value,
 138           od.objectPHID objectPHID,
 139           dd.objectPHID dimensionPHID
 140         FROM %R p
 141           LEFT JOIN %R od ON od.id = p.objectID
 142           LEFT JOIN %R dd ON dd.id = p.dimensionID
 143           WHERE %LA',
 144       $datapoint_table,
 145       $dimension_table,
 146       $dimension_table,
 147       $where);
 148     $rows = ipull($rows, null, 'id');
 149
 150     $results = array();
 151
 152     foreach ($refs as $ref) {
 153       if (!isset($rows[$ref])) {
 154         continue;
 155       }
 156
 157       $row = $rows[$ref];
 158
 159       $results[$ref] = array(
 160         'objectPHID' => $row['objectPHID'],
 161         'dimensionPHID' => $row['dimensionPHID'],
 162         'value' => (float)$row['value'],
 163       );
 164     }
 165
 166     return $results;
 167   }
 168
 169 }