公司開發使用的搜索引擎核心是Solr，但是應為業務原因，需要相對復雜權限機制。
1）通過Solr的filterQuery可以實現field過濾，實現過濾項的效果。索引
A{filter1:a,field2:a,field3:a,field4:1}
B{filter1:b,field2:b,field3:b,field4:2}
C{filter1:c,field2:c,field3:c,field4:3}
過濾條件
fq=field:b
結果
B{filter1:b,field2:b,field3:b,field4:b}
2）通過Solr的數字區間，也能方便實現密級的權限方式。索引
A{filter1:a,field2:a,field3:a,field4:1}
B{filter1:b,field2:b,field3:b,field4:2}
C{filter1:c,field2:c,field3:c,field4:3}
過濾條件
fq=field4:[2 TO *]
結果
B{filter1:b,field2:b,field3:b,field4:2}
C{filter1:c,field2:c,field3:c,field4:3}
3）但是真正常用的是多條件的、並集式的過濾項的效果（如：過濾出包涵filed1:a和filed1:b的所有索引）。
在網上搜索了很久，也問了一些做搜索很長時間的同事，都沒有找到解決方法，郁悶啊！最后決定自己仔細的把Solr源碼讀一讀，看看有沒有解決方法（一直覺得開發Solr大牛不太可能沒有開發這個功能），功夫不負有心人，花了1天時間終於找到了完美的解決方法。
先說一下Solr的過濾機制是怎么實現的：
1、查詢時將參數傳入Request，經過處理，會調用SolrCore.execute()方法。
2、SolrCore.execute()會調用RequestHandlerBase.handleRequest()方法。
3、RequestHandlerBase.handleRequest()會調用Component，默認有6個Component。分別是QueryComponent、FacetComponent、MoreLikeThisComponent、HighlightComponent、StatsComponent、DebugComponent
4、fq是在QueryComponent解析的，
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
String[] fqs = req.getParams().getParams(CommonParams.FQ);
if
(fqs!=
null
&& fqs.length!=
0
) {
List<Query> filters = rb.getFilters();
if
(filters==
null
) {
filters =
new
ArrayList<Query>(fqs.length);
}
for
(String fq : fqs) {
if
(fq !=
null
&& fq.trim().length()!=
0
) {
QParser fqp = QParser.getParser(fq,
null
, req);
filters.add(fqp.getQuery());
}
}
// only set the filters if they are not empty otherwise
// fq=&someotherParam= will trigger all docs filter for every request
// if filter cache is disabled
if
(!filters.isEmpty()) {
rb.setFilters( filters );
}
}
|
實現對fq的解析，往里跟蹤發現，是通過QueryParser.parse()將過濾條件轉換成Query的。這個就是問題關鍵，既然filed:value形式的過濾條件可以實現，那filed:{value1 or value2}不就可以實現我想要的多值並集試的過濾嗎？（QueryParser.parse()支持查詢語法）.
問題又來了，通過多次測試，發現value不能支持寫查詢表達式（糾結啊）。再一次耐心把代碼看一下，突然發現一段奇怪的代碼，QueryParsing.parseLocalParams()方法
|
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
|
int
off = start;
if
(!txt.startsWith(LOCALPARAM_START, off))
return
start;
StrParser p =
new
StrParser(txt, start, txt.length());
p.pos +=
2
;
// skip over "{!"
for
(; ; ) {
/*
if (p.pos>=txt.length()) {
throw new ParseException("Missing '}' parsing local params '" + txt + '"');
}
*/
char
ch = p.peek();
if
(ch == LOCALPARAM_END) {
return
p.pos +
1
;
}
String id = p.getId();
if
(id.length() ==
0
) {
throw
new
ParseException(
"Expected identifier '}' parsing local params '"
+ txt + '"');
}
String val =
null
;
ch = p.peek();
if
(ch !=
'='
) {
// single word... treat {!func} as type=func for easy lookup
val = id;
id = TYPE;
}
else
{
// saw equals, so read value
p.pos++;
ch = p.peek();
boolean
deref =
false
;
if
(ch ==
'$'
) {
p.pos++;
ch = p.peek();
deref =
true
;
// dereference whatever value is read by treating it as a variable name
}
if
(ch ==
'\"'
|| ch ==
'\''
) {
val = p.getQuotedString();
}
else
{
// read unquoted literal ended by whitespace or '}'
// there is no escaping.
int
valStart = p.pos;
for
(; ; ) {
if
(p.pos >= p.end) {
throw
new
ParseException(
"Missing end to unquoted value starting at "
+ valStart +
" str='"
+ txt +
"'"
);
}
char
c = p.val.charAt(p.pos);
if
(c == LOCALPARAM_END || Character.isWhitespace(c)) {
val = p.val.substring(valStart, p.pos);
break
;
}
p.pos++;
}
}
if
(deref) {
// dereference parameter
if
(params !=
null
) {
val = params.get(val);
}
}
}
if
(target !=
null
) target.put(id, val);
}
|
仔細看了一下，發現用於解析類似{!dismax v=$auth}這樣的字符傳（一直都不太想看這段代碼，太長了），作用是將$auth替換為傳入solr的參數，解析為{!dismax v=field4:2}，后用dismax轉換為Query，添加到過濾查詢里。
最終解決方法是：索引
A{filter1:a,field2:a,field3:a,field4:1}
B{filter1:b,field2:b,field3:b,field4:2}
C{filter1:c,field2:c,field3:c,field4:3}
|
1
2
3
4
5
|
Map<String,Object> paramMap =
new
HashMap<String,Object>();
SearchParams params =
new
SearchParams(paramMap) ;
paramMap.put(
"auth"
,
"filed4:1 filed4:3"
);
//將過濾查詢語句放入參數里
query.add(params);
query.addFilterQuery(
"{!dismax v=$auth}"
);
//通過dismax轉換為query，加入查詢語句，大功告成
|
結果：
A{filter1:a,field2:a,field3:a,field4:1}
C{filter1:c,field2:c,field3:c,field4:3}
附錄
文中使用類的全路徑
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
org.apache.solr.core.SolrCore
org.apache.solr.handler.StandardRequestHandler
org.apache.solr.handler.RequestHandlerBase.handleRequest
org.apache.solr.handler.StandardRequestHandler
org.apache.solr.handler.component.SearchHandler.handleRequestBody
org.apache.solr.handler.component.QueryComponent
org.apache.solr.handler.component.FacetComponent
org.apache.solr.handler.component.MoreLikeThisComponent
org.apache.solr.handler.component.HighlightComponent
org.apache.solr.handler.component.StatsComponent
org.apache.solr.handler.component.DebugComponent
org.apache.lucene.queryParser.QueryParser
org.apache.solr.search.QueryParsing
|