通用包
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
utils/pkg/es/search_queries_script.go

51 lines
1.2 KiB

// 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 es
import "errors"
// ScriptQuery allows to define scripts as filters.
//
// For details, see
// https://www.elastic.co/guide/en/elasticsearch/reference/7.0/query-dsl-script-query.html
type ScriptQuery struct {
script *Script
queryName string
}
// NewScriptQuery creates and initializes a new ScriptQuery.
func NewScriptQuery(script *Script) *ScriptQuery {
return &ScriptQuery{
script: script,
}
}
// QueryName sets the query name for the filter that can be used
// when searching for matched_filters per hit
func (q *ScriptQuery) QueryName(queryName string) *ScriptQuery {
q.queryName = queryName
return q
}
// Source returns JSON for the query.
func (q *ScriptQuery) Source() (interface{}, error) {
if q.script == nil {
return nil, errors.New("ScriptQuery expected a script")
}
source := make(map[string]interface{})
params := make(map[string]interface{})
source["script"] = params
src, err := q.script.Source()
if err != nil {
return nil, err
}
params["script"] = src
if q.queryName != "" {
params["_name"] = q.queryName
}
return source, nil
}