blob: 5041f3edba8a58b6961d8cd595a4a548c3a7322e (
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
|
<?php
namespace Elastica\Rescore;
use Elastica\Query as BaseQuery;
/**
* Query Rescore
*
* @category Xodoa
* @package Elastica
* @author Jason Hu <mjhu91@gmail.com>
* @link http://www.elasticsearch.org/guide/reference/api/search/rescore/
*/
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 \Elastica\Query\Rescore
*/
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 \Elastica\Query\Rescore
*/
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 \Elastica\Query\Rescore
*/
public function setRescoreQueryWeight($weight)
{
$query = $this->getParam('query');
$query['rescore_query_weight'] = $weight;
return $this->setParam('query', $query);
}
}
|
