Bugzilla/Elastic/Role/Object.pm


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# This Source Code Form is "Incompatible With Secondary Licenses", as
# defined by the Mozilla Public License, v. 2.0.
package Bugzilla::Elastic::Role::Object;

use 5.10.1;
use Role::Tiny;

requires qw(ES_TYPE ES_PROPERTIES es_document);
requires qw(ID_FIELD DB_TABLE);

sub ES_OBJECTS_AT_ONCE { 100 }

sub ES_SELECT_ALL_SQL {
    my ($class, $last_id) = @_;

    my $id = $class->ID_FIELD;
    my $table = $class->DB_TABLE;

    return ("SELECT $id FROM $table WHERE $id > ? ORDER BY $id", [$last_id // 0]);
}

requires qw(ES_SELECT_UPDATED_SQL);

around 'ES_PROPERTIES' => sub {
    my $orig = shift;
    my $self = shift;
    my $properties = $orig->($self, @_);
    $properties->{es_mtime} = { type => 'long' };
    $properties->{$self->ID_FIELD} = { type => 'long', analyzer => 'keyword' };

    return $properties;
};

around 'es_document' => sub {
    my ($orig, $self, $mtime) = @_;
    my $doc = $orig->($self);

    $doc->{es_mtime} = $mtime;
    $doc->{$self->ID_FIELD} = $self->id;

    return $doc;
};

1;