您的位置：首页 > 编程语言

solr基本操作代码示例

2015-09-24 12:44 369 查看

1、基本操作类
?

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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;

import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.impl.CommonsHttpSolrServer;
import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.FacetField.Count;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.SolrInputDocument;

import com.xzhe.goods.bo.Goods;
import com.xzhe.goods.dao.impl.GoodsDAO;

public class SolrSearch {
private static String tomcat_solr = "http://localhost:8983/solr";
private static CommonsHttpSolrServer solr = null;
private static GoodsDAO goodsDAO = GoodsDAO.getInstance();
private static List<Goods> goodList = new ArrayList<Goods>();

// 初始化solr服务
public static void initiate() {
try {
solr = new CommonsHttpSolrServer(tomcat_solr);
solr.setConnectionTimeout(100);
solr.setDefaultMaxConnectionsPerHost(100);
solr.setMaxTotalConnections(100);
} catch (Exception e) {
System.out.println("请检查tomcat服务器或端口是否开启!");
e.printStackTrace();
}
}

// 检测good是否有字段为空
public static Boolean CheckGood(Goods good) {
if (null == good.getTitle() || good.getTitle().length() == 0) {
return false;
} else if (null == good.getGoodsId()) {
return false;
} else if (null == good.getStartTime()) {
return false;
} else if (null == good.getDiscount()) {
return false;
} else if (null == good.getCurPrice()) {
return false;
} else if (null == good.getSiteId()) {
return false;
} else if (null == good.getBuyNum()) {
return false;
} else {
return true;
}
}

// 将数据库中的时间转换为solr可接受的格式
public static String Convertime(String time) {
time = time.replace(" ", "T");
time = time + "00Z";
return time;
}

// 添加list到索引，flag为false，在原有基础上添加，为false，重新添加、原有索引不会被删除
public static void addGoods(List<Goods> list, Boolean flag) {
if (flag) {
try {
solr.deleteByQuery("*:*");
} catch (Exception e) {
e.printStackTrace();
}
}

Collection<SolrInputDocument> docs = new ArrayList<SolrInputDocument>();

for (int i = 0; i < list.size(); i++) {
Goods good = list.get(i);
if (CheckGood(good)) {
SolrInputDocument doc = new SolrInputDocument();
doc.addField("goodid", good.getGoodsId());
doc.addField("title", good.getTitle());
doc.addField("siteid", good.getSiteId());
doc.addField("buynum", good.getBuyNum());
doc.addField("starttime", Convertime(good.getStartTime()
.toString()));
doc.addField("discount", good.getDiscount());
doc.addField("curprice", good.getCurPrice());
docs.add(doc);
}
}

try {
solr.add(docs);
solr.optimize();
solr.commit();
} catch (Exception e) {
e.printStackTrace();
}
}

// 转换good到beans
public static GoodsBeans ConvertGoodstoBeans(Goods good) {
if (!CheckGood(good)) {
return null;
}
GoodsBeans beans = new GoodsBeans(good.getGoodsId().toString(),
good.getTitle(), good.getSiteId().toString(), good.getBuyNum(),
good.getStartTime(), good.getDiscount(), good.getCurPrice());
return beans;
}

// 添加beans到索引
public static void addGoodsBeans(List<GoodsBeans> beansList) {
try {
solr.addBeans(beansList);
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
solr.optimize();
solr.commit();
} catch (Exception e) {
e.printStackTrace();
}
}
}

// 删除所有索引
public static void DeleteAllIndex() {
try {
solr.deleteByQuery("*:*");
solr.commit();
} catch (Exception e) {
e.printStackTrace();
}
}

// 根据ID删除索引
public static void DeleteIndex(List<String> ids) {
try {
solr.deleteById(ids);
solr.commit();
} catch (Exception e) {
e.printStackTrace();
}
}

// 进行搜索，field、key为查询（值或范围均可），start为起始查询位置，row为返回结果个数，sortfield为排序字段，flag中true升序、false降序，hightlight选择是否高亮返回，高亮字段为title
public static QueryResponse Search(String[] field, String[] key, int start,
int count, String[] sortfield, Boolean[] flag, Boolean hightlight) {
if (null == field || null == key || field.length != key.length) {
return null;
}
if (null == sortfield || null == flag
|| sortfield.length != flag.length) {
return null;
}

SolrQuery query = null;
try {
if (field[0].equals("title")) {
// query = new SolrQuery("*" + key + "*");
query = new SolrQuery(field[0] + ":" + key[0]);
} else {
query = new SolrQuery(field[0] + ":" + key[0]);
}
for (int i = 0; i < field.length; i++) {
if (field[i].equals("title")) {
// query = new SolrQuery("*" + key + "*");
query.addFilterQuery(field[i] + ":" + key[i]);
} else {
query.addFilterQuery(field[i] + ":" + key[i]);
}
}
query.setStart(start);
query.setRows(count);
for (int i = 0; i < sortfield.length; i++) {
if (flag[i]) {
query.addSortField(sortfield[i], SolrQuery.ORDER.asc);
} else {
query.addSortField(sortfield[i], SolrQuery.ORDER.desc);
}
}
if (null != hightlight) {
query.setHighlight(true); // 开启高亮组件
query.addHighlightField("title");// 高亮字段
query.setHighlightSimplePre("<font color=\"red\">");// 标记
query.setHighlightSimplePost("</font>");
query.setHighlightSnippets(1);
query.setHighlightFragsize(1000);
}
} catch (Exception e) {
e.printStackTrace();
}
System.out.println(query.toString());

QueryResponse rsp = null;
try {
rsp = solr.query(query);
} catch (Exception e) {
e.printStackTrace();
return null;
}
return rsp;
}

// 自动补全
public static String[] autoComplete(String prefix, int min) {
String words[] = null;
StringBuffer sb = new StringBuffer("");
SolrQuery query = new SolrQuery("*.*");
QueryResponse rsp = new QueryResponse();
try {
query.setFacet(true);
query.setQuery("*:*");
query.setFacetPrefix(prefix);
query.addFacetField("title");
rsp = solr.query(query);
} catch (Exception e) {
// TODO: handle exception
e.printStackTrace();
return null;
}

if (null != rsp) {
FacetField ff = rsp.getFacetField("title");
List<Count> countList = ff.getValues();
if (null == countList) {
return null;
}
for (int i = 0; i < countList.size(); i++) {
String tmp[] = countList.get(i).toString().split(" ");
if (tmp[0].length() < 2) {
continue;
}
sb.append(tmp[0] + " ");
min--;
if (min == 0) {
break;
}
}
words = sb.toString().split(" ");
} else {
return null;
}
return words;
}

public static void main(String[] args) {
initiate();

// 建立索引
// goodList = goodsDAO.findAll();
// System.out.println("所有商品载入完成！");
// DeleteAllIndex();
// System.out.println("原有索引已清除！");
// System.out.println("添加索引开始！");
// long starttime = System.currentTimeMillis();
// addGoods(goodList, true);
// long endtime = System.currentTimeMillis();
// System.out.println("共耗时" + (endtime - starttime) + "ms！");
// System.out.println("添加索引完成！");

// 进行查询
SolrDocumentList solrList = null;
QueryResponse rsp = null;

// rsp = Search("title", "*变形金* 蓝精灵", 0, 10, "buynum", false, true);
// String field[] = { "title", "buynum", "discount", "starttime" };
// String key[] = { "变形金刚哈利波特", "[90 TO 100]", "[2.0 TO 3.0]",
// "[2011-07-18T00:00:00.000Z TO 2011-07-19T00:00:00.000Z]" };
String field[] = {"title"};
String key[] = {"牛奶"};
String sortfield[] = { "buynum" };
Boolean flag[] = { false };
long starttime = System.currentTimeMillis();
rsp = Search(field, key, 0, 10, sortfield, flag, true);
long endtime = System.currentTimeMillis();
System.out.println("共耗时" + (endtime - starttime) + "ms！");

if (null != rsp) {
solrList = rsp.getResults();
for (int i = 0; i < solrList.size(); i++) {
System.out.println(solrList.get(i).toString());
}
// 高亮显示部分
System.out.println("搜索结果共" + solrList.size() + "条！");
System.out.println("");
Map<String, Map<String, List<String>>> hightlight = rsp
.getHighlighting();
List<GoodsBeans> tmpLists = rsp.getBeans(GoodsBeans.class);
for (int i = 0; i < tmpLists.size(); i++) {
String hlString = hightlight.get(tmpLists.get(i).getGoodId())
.get("title").toString();
if (null != hlString) {
// System.out.println(hlString);
}
}
}

// 自动补全
// String words[] = autoComplete("哈利", 10);
// if (null != words) {
// System.out.println(words.length);
// for (int i = 0; i < words.length; i++) {
// System.out.println(words[i]);
// }
// }
}

}

2、bean
?

import java.util.Date;

import org.apache.solr.client.solrj.beans.Field;

public class GoodsBeans {
@Field
private String goodid;
@Field
private String title;
@Field
private String siteid;
@Field
private int buynum;
@Field
private Date starttime;
@Field
private double discount;
@Field
private double curprice;

public String getTitle() {
return title;
}

public void setTitle(String title) {
this.title = title;
}

public String getSiteid() {
return siteid;
}

public void setSiteid(String siteid) {
this.siteid = siteid;
}

public int getBuynum() {
return buynum;
}

public void setBuynum(int buynum) {
this.buynum = buynum;
}

public Date getStarttime() {
return starttime;
}

public void setStarttime(Date starttime) {
this.starttime = starttime;
}

public double getDiscount() {
return discount;
}

public void setDiscount(double discount) {
this.discount = discount;
}

public double getCurprice() {
return curprice;
}

public void setCurprice(double curprice) {
this.curprice = curprice;
}

public void setGoodid(String goodid) {
this.goodid = goodid;
}

public String getGoodId() {
return goodid;
}

public GoodsBeans(){

}

public GoodsBeans(String goodid, String title, String siteid, int buynum,
Date starttime, double discount, double curprice) {
this.goodid = goodid;
this.title = title;
this.siteid = siteid;
this.buynum = buynum;
this.starttime = starttime;
this.discount = discount;
this.curprice = curprice;
}
}

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航