es 查询封装 新增 match_all

pkg/es/search_queries_match_all.go

@@ -0,0 +1,67 @@
+// 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 "encoding/json"
+
+var nilByte = []byte("null")
+
+// MatchAllQuery is the most simple query, which matches all documents,
+// giving them all a _score of 1.0.
+//
+// For more details, see
+// https://www.elastic.co/guide/en/elasticsearch/reference/7.0/query-dsl-match-all-query.html
+type MatchAllQuery struct {
+	boost     *float64
+	queryName string
+}
+
+// NewMatchAllQuery creates and initializes a new match all query.
+func NewMatchAllQuery() *MatchAllQuery {
+	return &MatchAllQuery{}
+}
+
+// Boost sets the boost for this query. Documents matching this query will
+// (in addition to the normal weightings) have their score multiplied by the
+// boost provided.
+func (q *MatchAllQuery) Boost(boost float64) *MatchAllQuery {
+	q.boost = &boost
+	return q
+}
+
+// QueryName sets the query name.
+func (q *MatchAllQuery) QueryName(name string) *MatchAllQuery {
+	q.queryName = name
+	return q
+}
+
+// Source returns JSON for the match all query.
+func (q *MatchAllQuery) Source() (interface{}, error) {
+	// {
+	//   "match_all" : { ... }
+	// }
+	source := make(map[string]interface{})
+	params := make(map[string]interface{})
+	source["match_all"] = params
+	if q.boost != nil {
+		params["boost"] = *q.boost
+	}
+	if q.queryName != "" {
+		params["_name"] = q.queryName
+	}
+	return source, nil
+}
+
+// MarshalJSON enables serializing the type as JSON.
+func (q *MatchAllQuery) MarshalJSON() ([]byte, error) {
+	if q == nil {
+		return nilByte, nil
+	}
+	src, err := q.Source()
+	if err != nil {
+		return nil, err
+	}
+	return json.Marshal(src)
+}

pkg/es/search_queries_match_all_test.go

@@ -0,0 +1,74 @@
+// 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 (
+	"encoding/json"
+	"testing"
+)
+
+func TestMatchAllQuery(t *testing.T) {
+	q := NewMatchAllQuery()
+	src, err := q.Source()
+	if err != nil {
+		t.Fatal(err)
+	}
+	data, err := json.Marshal(src)
+	if err != nil {
+		t.Fatalf("marshaling to JSON failed: %v", err)
+	}
+	got := string(data)
+	expected := `{"match_all":{}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestMatchAllQueryWithBoost(t *testing.T) {
+	q := NewMatchAllQuery().Boost(3.14)
+	src, err := q.Source()
+	if err != nil {
+		t.Fatal(err)
+	}
+	data, err := json.Marshal(src)
+	if err != nil {
+		t.Fatalf("marshaling to JSON failed: %v", err)
+	}
+	got := string(data)
+	expected := `{"match_all":{"boost":3.14}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestMatchAllQueryWithQueryName(t *testing.T) {
+	q := NewMatchAllQuery().QueryName("qname")
+	src, err := q.Source()
+	if err != nil {
+		t.Fatal(err)
+	}
+	data, err := json.Marshal(src)
+	if err != nil {
+		t.Fatalf("marshaling to JSON failed: %v", err)
+	}
+	got := string(data)
+	expected := `{"match_all":{"_name":"qname"}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestMatchAllMarshalJSON(t *testing.T) {
+	in := NewMatchAllQuery().Boost(3.14)
+	data, err := json.Marshal(in)
+	if err != nil {
+		t.Fatal(err)
+	}
+	got := string(data)
+	expected := `{"match_all":{"boost":3.14}}`
+	if got != expected {
+		t.Errorf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}