Skip to content

Commit f2f3b76

Browse files
MpdreamzMpdreamz
committed
Implemented external field support for terms query elastic/elasticsearch#2674 
1 parent b0d09fd commit f2f3b76

File tree

13 files changed

+365
-112
lines changed

13 files changed

+365
-112
lines changed

‎src/Nest.Tests.Integration/Integration/Query/TermToString.cs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,15 +92,15 @@ public void BoolToStringTests()
9292
.Term(p => p.BoolValue, _LookFor.BoolValue)
9393
)
9494
);
95-
this.AssertTermResults(results, 7);
95+
this.AssertTermResults(results);
9696
}
9797

9898

99-
private void AssertTermResults(IQueryResponse<ElasticSearchProject> results, int expected = 1)
99+
private void AssertTermResults(IQueryResponse<ElasticSearchProject> results)
100100
{
101101
Assert.True(results.IsValid, results.ConnectionStatus.Result);
102102
Assert.True(results.ConnectionStatus.Success, results.ConnectionStatus.Result);
103-
Assert.AreEqual(expected, results.Total);
103+
Assert.GreaterOrEqual(results.Total, 1);
104104
}
105105
}
106106
}

‎src/Nest.Tests.Unit/Nest.Tests.Unit.csproj

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@
200200
<Compile Include="Search\Query\Singles\SpanOrQueryJson.cs" />
201201
<Compile Include="Search\Query\Singles\SpanTermQueryJson.cs" />
202202
<Compile Include="Search\Query\Singles\Term\TermQueryJson.cs" />
203-
<Compile Include="Search\Query\Singles\TermsQueryJson.cs" />
203+
<Compile Include="Search\Query\Singles\Terms\TermsQueryJson.cs" />
204204
<Compile Include="Search\Query\Singles\TextPhrasePrefixQueryJson.cs" />
205205
<Compile Include="Search\Query\Singles\TextPhraseQueryJson.cs" />
206206
<Compile Include="Search\Query\Singles\TextQueryJson.cs" />
@@ -356,6 +356,7 @@
356356
</None>
357357
<None Include="packages.config" />
358358
<None Include="Search\Query\Singles\MultiMatch\TestMultiMatchJson.json" />
359+
<None Include="Search\Query\Singles\Terms\TermsQueryDescriptorUsingExternalField.json" />
359360
<None Include="Search\Query\Singles\Term\DateTimeWithCustomStringValue.json" />
360361
<None Include="Search\Query\Singles\Term\DateTimeToStringTest.json" />
361362
<None Include="Search\Query\Singles\Term\LongToStringTest.json" />

‎src/Nest.Tests.Unit/Search/Query/Singles/Terms/TermsQueryDescriptorUsingExternalField.json

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
{
2+
"from": 0,
3+
"size": 10,
4+
"query": {
5+
"terms": {
6+
"intValues": {
7+
"index": "nest_test_data",
8+
"type": "people",
9+
"id": "4",
10+
"path": "id"
11+
},
12+
"minimum_match": 2,
13+
"disable_coord": true,
14+
"_cache_key": "user_4_key"
15+
}
16+
}
17+
}

‎src/Nest.Tests.Unit/Search/Query/Singles/TermsQueryJson.cs renamed to ‎src/Nest.Tests.Unit/Search/Query/Singles/Terms/TermsQueryJson.cs

Lines changed: 27 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
1-
using NUnit.Framework;
1+
using System.Reflection;
2+
using NUnit.Framework;
23
using Nest.Tests.MockData.Domain;
34

4-
namespace Nest.Tests.Unit.Search.Query.Singles
5+
namespace Nest.Tests.Unit.Search.Query.Singles.Terms
56
{
67
[TestFixture]
7-
public class TermsQueryJson
8+
public class TermsQueryJson : BaseJsonTests
89
{
910
[Test]
1011
public void TermsQuery()
@@ -71,5 +72,28 @@ public void TermsQueryDescriptor()
7172
}";
7273
Assert.True(json.JsonEquals(expected), json);
7374
}
75+
76+
[Test]
77+
public void TermsQueryDescriptorUsingExternalField()
78+
{
79+
var s = new SearchDescriptor<ElasticSearchProject>()
80+
.From(0)
81+
.Size(10)
82+
.Query(ff => ff.
83+
TermsDescriptor(tq => tq
84+
.OnField(p=>p.IntValues)
85+
.MinimumMatch(2)
86+
.DisableCoord()
87+
.OnExternalField<Person>(ef=>ef
88+
.Path(p=>p.Id)
89+
.Id(4)
90+
)
91+
.CacheKey("user_4_key")
92+
93+
)
94+
);
95+
96+
this.JsonEquals(s, MethodInfo.GetCurrentMethod());
97+
}
7498
}
7599
}

‎src/Nest/DSL/Query/ExternalFieldDeclarationDescriptor.cs

Lines changed: 61 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,61 @@
1+
using System;
2+
using System.Globalization;
3+
using System.Linq.Expressions;
4+
using Nest.Resolvers;
5+
using Newtonsoft.Json;
6+
7+
namespace Nest
8+
{
9+
public class ExternalFieldDeclarationDescriptor<T> : IExternalFieldDeclarationDescriptor
10+
where T : class
11+
{
12+
internal Type _ClrType { get { return typeof(T); } }
13+
14+
[JsonProperty("index")]
15+
public IndexNameMarker _Index { get; set; }
16+
[JsonProperty("type")]
17+
public TypeNameMarker _Type { get; set; }
18+
[JsonProperty("id")]
19+
public string _Id { get; set; }
20+
[JsonProperty("path")]
21+
public string _Path { get; set; }
22+
23+
public ExternalFieldDeclarationDescriptor()
24+
{
25+
this._Type = new TypeNameMarker { Type = this._ClrType };
26+
this._Index = new IndexNameMarker { Type = this._ClrType };
27+
}
28+
29+
public ExternalFieldDeclarationDescriptor<T> Index(string index)
30+
{
31+
this._Index = index;
32+
return this;
33+
}
34+
public ExternalFieldDeclarationDescriptor<T> Id(string id)
35+
{
36+
this._Id = id;
37+
return this;
38+
}
39+
public ExternalFieldDeclarationDescriptor<T> Id(int id)
40+
{
41+
this._Id = id.ToString(CultureInfo.InvariantCulture);
42+
return this;
43+
}
44+
public ExternalFieldDeclarationDescriptor<T> Type(string type)
45+
{
46+
this._Type = new TypeNameMarker() { Name = type };
47+
return this;
48+
}
49+
public ExternalFieldDeclarationDescriptor<T> Path(string path)
50+
{
51+
this._Path = path;
52+
return this;
53+
}
54+
public ExternalFieldDeclarationDescriptor<T> Path(Expression<Func<T, object>> objectPath)
55+
{
56+
var fieldName = new PropertyNameResolver().Resolve(objectPath);
57+
this._Path = fieldName;
58+
return this;
59+
}
60+
}
61+
}

‎src/Nest/DSL/Query/IExternalFieldDeclarationDescriptor.cs

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
using Nest.Resolvers;
2+
3+
namespace Nest
4+
{
5+
public interface IExternalFieldDeclarationDescriptor
6+
{
7+
IndexNameMarker _Index { get; set; }
8+
TypeNameMarker _Type { get; set; }
9+
string _Id { get; set; }
10+
string _Path { get; set; }
11+
}
12+
}

‎src/Nest/DSL/Query/TermsQueryDescriptor.cs

Lines changed: 25 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
using System.Text;
55
using Newtonsoft.Json;
66
using System.Linq.Expressions;
7-
using System.Globalization;
87
using Newtonsoft.Json.Converters;
98
using Nest.Resolvers;
109
namespace Nest
@@ -17,14 +16,24 @@ public class TermsQueryDescriptor<T> : IQuery where T : class
1716
internal bool _DisableCord { get; set; }
1817
internal IEnumerable<string> _Terms { get; set; }
1918

19+
internal IExternalFieldDeclarationDescriptor _ExternalField { get; set; }
20+
21+
internal string _CacheKey { get; set; }
22+
2023
internal bool IsConditionless
2124
{
2225
get
2326
{
24-
return this._Field.IsNullOrEmpty() || !this._Terms.HasAny();
27+
return this._Field.IsNullOrEmpty()
28+
||
29+
(!this._Terms.HasAny() && this._ExternalField == null);
2530
}
2631
}
27-
32+
public TermsQueryDescriptor<T> CacheKey(string cacheKey)
33+
{
34+
this._CacheKey = cacheKey;
35+
return this;
36+
}
2837
public TermsQueryDescriptor<T> OnField(string field)
2938
{
3039
this._Field = field;
@@ -35,6 +44,19 @@ public TermsQueryDescriptor<T> OnField(Expression<Func<T, object>> objectPath)
3544
var fieldName = new PropertyNameResolver().Resolve(objectPath);
3645
return this.OnField(fieldName);
3746
}
47+
48+
public TermsQueryDescriptor<T> OnExternalField<K>(
49+
Func<ExternalFieldDeclarationDescriptor<K>, ExternalFieldDeclarationDescriptor<K>> externalFieldSelector
50+
)
51+
where K : class
52+
{
53+
externalFieldSelector.ThrowIfNull("externalFieldSelector");
54+
var descriptor = externalFieldSelector(new ExternalFieldDeclarationDescriptor<K>());
55+
this._ExternalField = descriptor;
56+
return this;
57+
}
58+
59+
3860
public TermsQueryDescriptor<T> MinimumMatch(int minMatch)
3961
{
4062
this._MinMatch = minMatch;

‎src/Nest/DSL/QueryDescriptor.cs

Lines changed: 63 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,13 @@ public QueryDescriptor()
3636
[JsonProperty(PropertyName = "ids")]
3737
internal IdsQuery IdsQuery { get; set; }
3838
[JsonProperty(PropertyName = "custom_score")]
39-
internal CustomScoreQueryDescriptor<T> CustomScoreQueryDescriptor { get; set; }
40-
[JsonProperty(PropertyName = "custom_filters_score")]
41-
internal CustomFiltersScoreDescriptor<T> CustomFiltersScoreQueryDescriptor { get; set; }
39+
internal CustomScoreQueryDescriptor<T> CustomScoreQueryDescriptor { get; set; }
40+
[JsonProperty(PropertyName = "custom_filters_score")]
41+
internal CustomFiltersScoreDescriptor<T> CustomFiltersScoreQueryDescriptor { get; set; }
4242
[JsonProperty(PropertyName = "custom_boost_factor")]
43-
internal CustomBoostFactorQueryDescriptor<T> CustomBoostFactorQueryDescriptor { get; set; }
44-
[JsonProperty(PropertyName = "constant_score")]
45-
internal ConstantScoreQueryDescriptor<T> ConstantScoreQueryDescriptor { get; set; }
43+
internal CustomBoostFactorQueryDescriptor<T> CustomBoostFactorQueryDescriptor { get; set; }
44+
[JsonProperty(PropertyName = "constant_score")]
45+
internal ConstantScoreQueryDescriptor<T> ConstantScoreQueryDescriptor { get; set; }
4646
[JsonProperty(PropertyName = "dis_max")]
4747
internal DismaxQueryDescriptor<T> DismaxQueryDescriptor { get; set; }
4848
[JsonProperty(PropertyName = "filtered")]
@@ -102,7 +102,7 @@ internal QueryDescriptor<T> CreateConditionlessQueryDescriptor(IQuery query)
102102
if (this._Strict)
103103
throw new DslException("Query resulted in a conditionless {0} query (json by approx):\n"
104104
.F(
105-
query.GetType().Name.Replace("Descriptor", "").Replace("`1","")
105+
query.GetType().Name.Replace("Descriptor", "").Replace("`1", "")
106106
)
107107
);
108108
return new QueryDescriptor<T> { IsConditionless = !this._Strict };
@@ -193,17 +193,35 @@ public BaseQuery TermsDescriptor(Action<TermsQueryDescriptor<T>> selector)
193193
if (query.IsConditionless)
194194
return CreateConditionlessQueryDescriptor(query);
195195

196-
this.TermsQueryDescriptor = new Dictionary<string, object>()
196+
if (query._ExternalField == null)
197197
{
198-
{ query._Field, query._Terms}
199-
};
198+
this.TermsQueryDescriptor = new Dictionary<string, object>()
199+
{
200+
{ query._Field, query._Terms}
201+
};
202+
}
203+
else
204+
{
205+
if (query._ExternalField._Id.IsNullOrEmpty())
206+
throw new DslException("terms query external field has no id set");
207+
208+
this.TermsQueryDescriptor = new Dictionary<string, object>()
209+
{
210+
{ query._Field, query._ExternalField}
211+
};
212+
}
213+
200214
if (query._MinMatch.HasValue)
201215
{
202216
this.TermsQueryDescriptor.Add("minimum_match", query._MinMatch);
203217
}
204218
if (query._DisableCord)
205219
{
206-
this.TermsQueryDescriptor.Add("disable_coord", query._DisableCord);
220+
this.TermsQueryDescriptor.Add("disable_coord", query._DisableCord);
221+
}
222+
if (!query._CacheKey.IsNullOrEmpty())
223+
{
224+
this.TermsQueryDescriptor.Add("_cache_key", query._CacheKey);
207225
}
208226
return new QueryDescriptor<T> { TermsQueryDescriptor = this.TermsQueryDescriptor };
209227
}
@@ -587,36 +605,36 @@ public BaseQuery CustomBoostFactor(Action<CustomBoostFactorQueryDescriptor<T>> s
587605
this.CustomBoostFactorQueryDescriptor = query;
588606
return new QueryDescriptor<T> { CustomBoostFactorQueryDescriptor = this.CustomBoostFactorQueryDescriptor };
589607
}
590-
/// <summary>
591-
/// custom_score query allows to wrap another query and customize the scoring of it optionally with a
592-
/// computation derived from other field values in the doc (numeric ones) using script expression
593-
/// </summary>
594-
public BaseQuery CustomScore(Action<CustomScoreQueryDescriptor<T>> customScoreQuery)
595-
{
596-
var query = new CustomScoreQueryDescriptor<T>();
597-
customScoreQuery(query);
598-
599-
if (query.IsConditionless)
600-
return CreateConditionlessQueryDescriptor(query);
601-
602-
this.CustomScoreQueryDescriptor = query;
603-
return new QueryDescriptor<T> { CustomScoreQueryDescriptor = this.CustomScoreQueryDescriptor };
604-
}
605-
/// <summary>
606-
/// custom_score query allows to wrap another query and customize the scoring of it optionally with a
607-
/// computation derived from other field values in the doc (numeric ones) using script or boost expression
608-
/// </summary>
609-
public BaseQuery CustomFiltersScore(Action<CustomFiltersScoreDescriptor<T>> customFiltersScoreQuery)
610-
{
611-
var query = new CustomFiltersScoreDescriptor<T>();
612-
customFiltersScoreQuery(query);
613-
614-
if (query.IsConditionless)
615-
return CreateConditionlessQueryDescriptor(query);
616-
617-
this.CustomFiltersScoreQueryDescriptor = query;
618-
return new QueryDescriptor<T> { CustomFiltersScoreQueryDescriptor = this.CustomFiltersScoreQueryDescriptor };
619-
}
608+
/// <summary>
609+
/// custom_score query allows to wrap another query and customize the scoring of it optionally with a
610+
/// computation derived from other field values in the doc (numeric ones) using script expression
611+
/// </summary>
612+
public BaseQuery CustomScore(Action<CustomScoreQueryDescriptor<T>> customScoreQuery)
613+
{
614+
var query = new CustomScoreQueryDescriptor<T>();
615+
customScoreQuery(query);
616+
617+
if (query.IsConditionless)
618+
return CreateConditionlessQueryDescriptor(query);
619+
620+
this.CustomScoreQueryDescriptor = query;
621+
return new QueryDescriptor<T> { CustomScoreQueryDescriptor = this.CustomScoreQueryDescriptor };
622+
}
623+
/// <summary>
624+
/// custom_score query allows to wrap another query and customize the scoring of it optionally with a
625+
/// computation derived from other field values in the doc (numeric ones) using script or boost expression
626+
/// </summary>
627+
public BaseQuery CustomFiltersScore(Action<CustomFiltersScoreDescriptor<T>> customFiltersScoreQuery)
628+
{
629+
var query = new CustomFiltersScoreDescriptor<T>();
630+
customFiltersScoreQuery(query);
631+
632+
if (query.IsConditionless)
633+
return CreateConditionlessQueryDescriptor(query);
634+
635+
this.CustomFiltersScoreQueryDescriptor = query;
636+
return new QueryDescriptor<T> { CustomFiltersScoreQueryDescriptor = this.CustomFiltersScoreQueryDescriptor };
637+
}
620638
/// <summary>
621639
/// A query that matches documents matching boolean combinations of other queries. The bool query maps to
622640
/// Lucene BooleanQuery.
@@ -727,8 +745,8 @@ public BaseQuery Wildcard(Expression<Func<T, object>> fieldDescriptor
727745
/// one of the wildcards * or ?. The wildcard query maps to Lucene WildcardQuery.
728746
/// </summary>
729747
public BaseQuery Wildcard(
730-
string field,
731-
string value,
748+
string field,
749+
string value,
732750
double? Boost = null,
733751
RewriteMultiTerm? Rewrite = null)
734752
{
@@ -764,8 +782,8 @@ public BaseQuery Prefix(Expression<Func<T, object>> fieldDescriptor
764782
/// The prefix query maps to Lucene PrefixQuery.
765783
/// </summary>
766784
public BaseQuery Prefix(
767-
string field,
768-
string value,
785+
string field,
786+
string value,
769787
double? Boost = null,
770788
RewriteMultiTerm? Rewrite = null)
771789
{

0 commit comments

Comments
 (0)