File: search_filters_regexp.go

package info (click to toggle)
golang-gopkg-olivere-elastic.v2 2.0.12-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, buster, forky, sid, trixie
  • size: 1,964 kB
  • sloc: makefile: 17
file content (90 lines) | stat: -rw-r--r-- 2,281 bytes parent folder | download | duplicates (2)
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
// Copyright 2012-2015 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

// RegexpFilter allows filtering for regular expressions.
// See http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-regexp-filter.html
// and http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-regexp-query.html#regexp-syntax
// for details.
type RegexpFilter struct {
	Filter
	name                  string
	regexp                string
	flags                 *string
	maxDeterminizedStates *int
	cache                 *bool
	cacheKey              string
	filterName            string
}

// NewRegexpFilter sets up a new RegexpFilter.
func NewRegexpFilter(name, regexp string) RegexpFilter {
	return RegexpFilter{name: name, regexp: regexp}
}

// Flags sets the regexp flags.
// See http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/query-dsl-regexp-query.html#_optional_operators
// for details.
func (f RegexpFilter) Flags(flags string) RegexpFilter {
	f.flags = &flags
	return f
}

func (f RegexpFilter) MaxDeterminizedStates(maxDeterminizedStates int) RegexpFilter {
	f.maxDeterminizedStates = &maxDeterminizedStates
	return f
}

func (f RegexpFilter) Cache(cache bool) RegexpFilter {
	f.cache = &cache
	return f
}

func (f RegexpFilter) CacheKey(cacheKey string) RegexpFilter {
	f.cacheKey = cacheKey
	return f
}

func (f RegexpFilter) FilterName(filterName string) RegexpFilter {
	f.filterName = filterName
	return f
}

func (f RegexpFilter) Source() interface{} {
	// {
	//   "regexp" : {
	//     "..." : "..."
	//   }
	// }

	source := make(map[string]interface{})

	params := make(map[string]interface{})
	source["regexp"] = params

	if f.flags == nil {
		params[f.name] = f.regexp
	} else {
		x := make(map[string]interface{})
		x["value"] = f.regexp
		x["flags"] = *f.flags
		if f.maxDeterminizedStates != nil {
			x["max_determinized_states"] = *f.maxDeterminizedStates
		}
		params[f.name] = x
	}

	if f.filterName != "" {
		params["_name"] = f.filterName
	}
	if f.cache != nil {
		params["_cache"] = *f.cache
	}
	if f.cacheKey != "" {
		params["_cache_key"] = f.cacheKey
	}

	return source
}