0ct0pu5/fess
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix #31

Browse source
This commit is contained in:
Shinsuke Sugaya 2013-11-08 07:13:25 +09:00
parent 6a1101b540
commit a5f01c4039
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
src/main/java/jp/sf/fess/ds/impl/IndexUpdateCallbackImpl.java
View file

@ -22,6 +22,8 @@ import java.util.Map;
import java.util.concurrent.atomic.AtomicLong;
import jp.sf.fess.FessSystemException;
import jp.sf.fess.db.cbean.ClickLogCB;
import jp.sf.fess.db.exbhv.ClickLogBhv;
import jp.sf.fess.ds.IndexUpdateCallback;
import jp.sf.fess.helper.CrawlingSessionHelper;
@ -39,6 +41,10 @@ public class IndexUpdateCallbackImpl implements IndexUpdateCallback {
public int maxDocumentCacheSize = 10;
public boolean clickCountEnabled = true;
public String clickCountField = "clickCount_i";
protected volatile AtomicLong documentSize = new AtomicLong(0);
protected volatile long commitPerCount = 0;
@ -82,6 +88,10 @@ public class IndexUpdateCallbackImpl implements IndexUpdateCallback {
doc.addField(entry.getKey(), entry.getValue());
}
if (clickCountEnabled) {
addClickCountField(doc, dataMap.get("url").toString());
}
docList.add(doc);
if (logger.isDebugEnabled()) {
logger.debug("Added the document. "
@ -147,6 +157,19 @@ public class IndexUpdateCallbackImpl implements IndexUpdateCallback {
docList.clear();
}
protected void addClickCountField(final SolrInputDocument doc,
final String url) {
final ClickLogBhv clickLogBhv = SingletonS2Container
.getComponent(ClickLogBhv.class);
final ClickLogCB cb = new ClickLogCB();
cb.query().setUrl_Equal(url);
final int count = clickLogBhv.selectCount(cb);
doc.addField(clickCountField, count);
if (logger.isDebugEnabled()) {
logger.debug("Click Count: " + count + ", url: " + url);
}
}
@Override
public long getDocumentSize() {
return documentSize.get();