Query statistic how many runs in a timeframe are at which step

[MAILPOET-5091]
This commit is contained in:
David Remer
2023-06-28 12:51:40 +03:00
committed by Aschepikov
parent ae57f81c14
commit d2fce2014a
2 changed files with 93 additions and 0 deletions

View File

@@ -176,6 +176,32 @@ class AutomationRunStorage {
}
}
public function getAutomationStepStatisticForTimeFrame(int $automationId, string $status, \DateTimeImmutable $after, \DateTimeImmutable $before, int $versionId = null): array {
$table = esc_sql($this->table);
$where = "automation_id = %d
AND `status` = %s
AND created_at BETWEEN %s AND %s";
if ($versionId) {
$where .= " AND version_id = %d";
}
$sql = "
SELECT
COUNT(id) AS `count`,
next_step_id
FROM $table as log
WHERE $where
GROUP BY next_step_id
";
$sql = $versionId ?
$this->wpdb->prepare($sql, $automationId, $status, $after->format('Y-m-d H:i:s'), $before->format('Y-m-d H:i:s'), $versionId) :
$this->wpdb->prepare($sql, $automationId, $status, $after->format('Y-m-d H:i:s'), $before->format('Y-m-d H:i:s'));
$sql = is_string($sql) ? $sql : '';
$result = $this->wpdb->get_results($sql, ARRAY_A);
return is_array($result) ? $result : [];
}
public function truncate(): void {
$table = esc_sql($this->table);
$this->wpdb->query("TRUNCATE $table");

View File

@@ -0,0 +1,67 @@
<?php declare(strict_types = 1);
namespace MailPoet\Test\Automation\Engine\Storage;
use MailPoet\Automation\Engine\Data\AutomationRun;
use MailPoet\Automation\Engine\Storage\AutomationRunStorage;
class AutomationRunStorageTest extends \MailPoetTest {
/** @var AutomationRunStorage */
private $testee;
public function _before() {
$this->testee = $this->diContainer->get(AutomationRunStorage::class);
}
public function testAutomationStepStatisticForTimeFrame() {
global $wpdb;
$expected = [
[
'count' => '2',
'next_step_id' => 'step-1',
],
[
'count' => '1',
'next_step_id' => 'step-2',
],
];
$timeFrame = [
'after' => new \DateTimeImmutable('2020-01-01 00:00:00'),
'before' => new \DateTimeImmutable('2020-01-02 00:00:00'),
];
$status = AutomationRun::STATUS_RUNNING;
$automationId = 1;
$this->testee->truncate();
$sql = "insert into " . $wpdb->prefix .
"mailpoet_automation_runs" .
"(automation_id, version_id, created_at, `status`, trigger_key, next_step_id) values" .
"($automationId, 1, '2019-12-31 23:59:59', '$status', 'trigger_key', 'step-1')," . // Outside of timeframe
"($automationId, 1, '2020-01-01 00:00:00', '$status', 'trigger_key', 'step-1')," . // Should match
"($automationId, 1, '2020-01-01 00:00:00', '$status', 'trigger_key', 'step-2')," . // Should match
"($automationId, 2, '2020-01-02 00:00:00', '$status', 'trigger_key', 'step-1')," . // Should match when version not 1
"($automationId, 2, '2020-01-02 00:00:01', '$status', 'trigger_key', 'step-1')," . // Outside of timeframe
"($automationId, 1, '2020-01-01 00:00:00', 'complete', 'trigger_key', 'step-1')," . // Wrong status
"(2, 1, '2020-01-01 00:00:00', '$status', 'trigger_key', 'step-1')"; // Wrong automation id
$this->assertNotFalse($wpdb->query($sql));
$result = $this->testee->getAutomationStepStatisticForTimeFrame($automationId, $status, $timeFrame['after'], $timeFrame['before']);
$this->assertEquals($expected, $result);
$versionId = 1;
$result = $this->testee->getAutomationStepStatisticForTimeFrame($automationId, $status, $timeFrame['after'], $timeFrame['before'], $versionId);
$expected = [
[
'count' => '1',
'next_step_id' => 'step-1',
],
[
'count' => '1',
'next_step_id' => 'step-2',
],
];
$this->assertEquals($expected, $result);
}
}