blob: 00e62725f091d7cb4ee5a10d6028986e79386356 (
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// Copyright 2012-present Oliver Eilhard. All rights reserved.
// Use of this source code is governed by a MIT-license.
// See http://olivere.mit-license.org/license.txt for details.
package elastic
// GeoDistanceQuery filters documents that include only hits that exists
// within a specific distance from a geo point.
//
// For more details, see:
// https://www.elastic.co/guide/en/elasticsearch/reference/6.0/query-dsl-geo-distance-query.html
type GeoDistanceQuery struct {
name string
distance string
lat float64
lon float64
geohash string
distanceType string
queryName string
}
// NewGeoDistanceQuery creates and initializes a new GeoDistanceQuery.
func NewGeoDistanceQuery(name string) *GeoDistanceQuery {
return &GeoDistanceQuery{name: name}
}
func (q *GeoDistanceQuery) GeoPoint(point *GeoPoint) *GeoDistanceQuery {
q.lat = point.Lat
q.lon = point.Lon
return q
}
func (q *GeoDistanceQuery) Point(lat, lon float64) *GeoDistanceQuery {
q.lat = lat
q.lon = lon
return q
}
func (q *GeoDistanceQuery) Lat(lat float64) *GeoDistanceQuery {
q.lat = lat
return q
}
func (q *GeoDistanceQuery) Lon(lon float64) *GeoDistanceQuery {
q.lon = lon
return q
}
func (q *GeoDistanceQuery) GeoHash(geohash string) *GeoDistanceQuery {
q.geohash = geohash
return q
}
func (q *GeoDistanceQuery) Distance(distance string) *GeoDistanceQuery {
q.distance = distance
return q
}
func (q *GeoDistanceQuery) DistanceType(distanceType string) *GeoDistanceQuery {
q.distanceType = distanceType
return q
}
func (q *GeoDistanceQuery) QueryName(queryName string) *GeoDistanceQuery {
q.queryName = queryName
return q
}
// Source returns JSON for the function score query.
func (q *GeoDistanceQuery) Source() (interface{}, error) {
// {
// "geo_distance" : {
// "distance" : "200km",
// "pin.location" : {
// "lat" : 40,
// "lon" : -70
// }
// }
// }
source := make(map[string]interface{})
params := make(map[string]interface{})
if q.geohash != "" {
params[q.name] = q.geohash
} else {
location := make(map[string]interface{})
location["lat"] = q.lat
location["lon"] = q.lon
params[q.name] = location
}
if q.distance != "" {
params["distance"] = q.distance
}
if q.distanceType != "" {
params["distance_type"] = q.distanceType
}
if q.queryName != "" {
params["_name"] = q.queryName
}
source["geo_distance"] = params
return source, nil
}
|
