1 |
|
|
2 |
|
|
3 |
|
|
4 |
|
|
5 |
|
|
6 |
|
|
7 |
|
|
8 |
|
|
9 |
|
|
10 |
|
|
11 |
|
|
12 |
|
|
13 |
|
|
14 |
|
|
15 |
|
|
16 |
|
|
17 |
|
|
18 |
|
|
19 |
|
|
20 |
|
package org.xwiki.search.solr.internal.reference; |
21 |
|
|
22 |
|
import java.util.ArrayList; |
23 |
|
import java.util.List; |
24 |
|
|
25 |
|
import javax.inject.Inject; |
26 |
|
import javax.inject.Named; |
27 |
|
import javax.inject.Provider; |
28 |
|
import javax.inject.Singleton; |
29 |
|
|
30 |
|
import org.apache.solr.client.solrj.util.ClientUtils; |
31 |
|
import org.xwiki.component.annotation.Component; |
32 |
|
import org.xwiki.model.EntityType; |
33 |
|
import org.xwiki.model.reference.EntityReference; |
34 |
|
import org.xwiki.model.reference.EntityReferenceSerializer; |
35 |
|
import org.xwiki.query.QueryException; |
36 |
|
import org.xwiki.query.QueryManager; |
37 |
|
import org.xwiki.search.solr.internal.api.FieldUtils; |
38 |
|
import org.xwiki.search.solr.internal.api.SolrIndexerException; |
39 |
|
|
40 |
|
import com.google.common.collect.Iterables; |
41 |
|
|
42 |
|
|
43 |
|
|
44 |
|
|
45 |
|
@version |
46 |
|
@since |
47 |
|
|
48 |
|
@Component |
49 |
|
@Named("space") |
50 |
|
@Singleton |
|
|
| 91.7% |
Uncovered Elements: 2 (24) |
Complexity: 4 |
Complexity Density: 0.18 |
|
51 |
|
public class SpaceSolrReferenceResolver extends AbstractSolrReferenceResolver |
52 |
|
{ |
53 |
|
|
54 |
|
|
55 |
|
|
56 |
|
@Inject |
57 |
|
@Named("document") |
58 |
|
private Provider<SolrReferenceResolver> documentResolverProvider; |
59 |
|
|
60 |
|
|
61 |
|
|
62 |
|
|
63 |
|
@Inject |
64 |
|
@Named("wiki") |
65 |
|
private Provider<SolrReferenceResolver> wikiResolverProvider; |
66 |
|
|
67 |
|
|
68 |
|
|
69 |
|
|
70 |
|
@Inject |
71 |
|
private QueryManager queryManager; |
72 |
|
|
73 |
|
@Inject |
74 |
|
@Named("local") |
75 |
|
private EntityReferenceSerializer<String> localEntityReferenceSerializer; |
76 |
|
|
|
|
| 84.6% |
Uncovered Elements: 2 (13) |
Complexity: 3 |
Complexity Density: 0.23 |
|
77 |
10 |
@Override... |
78 |
|
public List<EntityReference> getReferences(EntityReference spaceReference) throws SolrIndexerException |
79 |
|
{ |
80 |
10 |
List<EntityReference> result = new ArrayList<EntityReference>(); |
81 |
10 |
EntityReference wikiReference = spaceReference.extractReference(EntityType.WIKI); |
82 |
10 |
String localSpaceReference = this.localEntityReferenceSerializer.serialize(spaceReference); |
83 |
|
|
84 |
|
|
85 |
|
|
86 |
|
|
87 |
10 |
List<String> documentNames; |
88 |
10 |
try { |
89 |
10 |
documentNames = |
90 |
|
this.queryManager.getNamedQuery("getSpaceDocsName").setWiki(wikiReference.getName()) |
91 |
|
.bindValue("space", localSpaceReference).execute(); |
92 |
|
} catch (QueryException e) { |
93 |
0 |
throw new SolrIndexerException("Failed to query space [" + spaceReference + "] documents", e); |
94 |
|
} |
95 |
|
|
96 |
10 |
for (String documentName : documentNames) { |
97 |
18 |
EntityReference documentReference = new EntityReference(documentName, EntityType.DOCUMENT, spaceReference); |
98 |
|
|
99 |
18 |
try { |
100 |
18 |
Iterables.addAll(result, this.documentResolverProvider.get().getReferences(documentReference)); |
101 |
|
} catch (Exception e) { |
102 |
0 |
this.logger.error("Failed to resolve references for document [" + documentReference + "]", e); |
103 |
|
} |
104 |
|
} |
105 |
|
|
106 |
10 |
return result; |
107 |
|
} |
108 |
|
|
|
|
| 100% |
Uncovered Elements: 0 (9) |
Complexity: 1 |
Complexity Density: 0.11 |
|
109 |
3 |
@Override... |
110 |
|
public String getQuery(EntityReference reference) throws SolrIndexerException |
111 |
|
{ |
112 |
3 |
StringBuilder builder = new StringBuilder(); |
113 |
|
|
114 |
3 |
EntityReference wikiReference = reference.extractReference(EntityType.WIKI); |
115 |
3 |
builder.append(wikiResolverProvider.get().getQuery(wikiReference)); |
116 |
|
|
117 |
3 |
builder.append(QUERY_AND); |
118 |
|
|
119 |
3 |
builder.append(FieldUtils.SPACE_EXACT); |
120 |
3 |
builder.append(':'); |
121 |
3 |
String localSpaceReference = this.localEntityReferenceSerializer.serialize(reference); |
122 |
3 |
builder.append(ClientUtils.escapeQueryChars(localSpaceReference)); |
123 |
|
|
124 |
3 |
return builder.toString(); |
125 |
|
} |
126 |
|
} |