SolrReindexTest_ItemExtension::augmentSQL()   A
last analyzed

Complexity

Conditions 3
Paths 2

Size

Total Lines 6
Code Lines 4

Duplication

Lines 0
Ratio 0 %

Importance

Changes 3
Bugs 0 Features 0
Metric Value
cc 3
eloc 4
c 3
b 0
f 0
nc 2
nop 2
dl 0
loc 6
rs 10
1
<?php
2
3
namespace SilverStripe\FullTextSearch\Tests\SolrReindexTest;
4
5
use SilverStripe\ORM\DataExtension;
6
use SilverStripe\Dev\TestOnly;
7
use SilverStripe\ORM\Queries\SQLSelect;
8
use SilverStripe\ORM\DataQuery;
9
use SilverStripe\FullTextSearch\Tests\SolrReindexTest\SolrReindexTest_Variant;
10
use SilverStripe\Core\Convert;
11
12
/**
13
 * Select only records in the current variant
14
 */
15
class SolrReindexTest_ItemExtension extends DataExtension implements TestOnly
16
{
17
    /**
18
     * Filter records on the current variant
19
     *
20
     * @param SQLSelect $query
21
     * @param DataQuery $dataQuery
22
     */
23
    public function augmentSQL(SQLSelect $query, DataQuery $dataQuery = null)
24
    {
25
        $variant = SolrReindexTest_Variant::get_current();
26
        if ($variant !== null && !$query->filtersOnID()) {
27
            $sqlVariant = Convert::raw2sql($variant);
28
            $query->addWhere("\"Variant\" = '{$sqlVariant}'");
29
        }
30
    }
31
}
32