blob: 3b0a107153a4d1a6313a88a1748023511c24ddb8 (
plain)
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
<?php
namespace Elastica\Rescore;
use Elastica\Query as BaseQuery;
/**
* Query Rescore.
*
* @author Jason Hu <mjhu91@gmail.com>
*
* @link http://www.elastic.co/guide/en/elasticsearch/reference/current/search-request-rescore.html
*/
class Query extends AbstractRescore
{
/**
* Constructor.
*
* @param string|\Elastica\Query\AbstractQuery $rescoreQuery
* @param string|\Elastica\Query\AbstractQuery $query
*/
public function __construct($query = null)
{
$this->setParam('query', array());
$this->setRescoreQuery($query);
}
/**
* Override default implementation so params are in the format
* expected by elasticsearch.
*
* @return array Rescore array
*/
public function toArray()
{
$data = $this->getParams();
if (!empty($this->_rawParams)) {
$data = array_merge($data, $this->_rawParams);
}
return $data;
}
/**
* Sets rescoreQuery object.
*
* @param string|\Elastica\Query|\Elastica\Query\AbstractQuery $query
*
* @return $this
*/
public function setRescoreQuery($rescoreQuery)
{
$query = BaseQuery::create($rescoreQuery);
$data = $query->toArray();
$query = $this->getParam('query');
$query['rescore_query'] = $data['query'];
return $this->setParam('query', $query);
}
/**
* Sets query_weight.
*
* @param float $weight
*
* @return $this
*/
public function setQueryWeight($weight)
{
$query = $this->getParam('query');
$query['query_weight'] = $weight;
return $this->setParam('query', $query);
}
/**
* Sets rescore_query_weight.
*
* @param float $size
*
* @return $this
*/
public function setRescoreQueryWeight($weight)
{
$query = $this->getParam('query');
$query['rescore_query_weight'] = $weight;
return $this->setParam('query', $query);
}
}
|