You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
4.2 KiB
100 lines
4.2 KiB
/**
|
|
* RankingSolr_p
|
|
* Copyright 2012 by Michael Peter Christen
|
|
* First released 30.11.2012 at http://yacy.net
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with this program in the file lgpl21.txt
|
|
* If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import java.util.Map;
|
|
|
|
import net.yacy.cora.federate.solr.Boost;
|
|
import net.yacy.cora.federate.solr.YaCySchema;
|
|
import net.yacy.cora.protocol.RequestHeader;
|
|
import net.yacy.search.Switchboard;
|
|
import net.yacy.search.SwitchboardConstants;
|
|
import net.yacy.search.query.SearchEventCache;
|
|
import net.yacy.server.serverObjects;
|
|
import net.yacy.server.serverSwitch;
|
|
|
|
public class RankingSolr_p {
|
|
|
|
|
|
public static serverObjects respond(@SuppressWarnings("unused") final RequestHeader header, final serverObjects post, final serverSwitch env) {
|
|
final Switchboard sb = (Switchboard) env;
|
|
|
|
// clean up all search events
|
|
SearchEventCache.cleanupEvents(true);
|
|
|
|
if (post != null && post.containsKey("EnterDoublecheck")) {
|
|
Boost.RANKING.setMinTokenLen(post.getInt("minTokenLen", 3));
|
|
Boost.RANKING.setQuantRate(post.getFloat("quantRate", 0.5f));
|
|
sb.setConfig(SwitchboardConstants.SEARCH_RANKING_SOLR_DOUBLEDETECTION_MINLENGTH, Boost.RANKING.getMinTokenLen());
|
|
sb.setConfig(SwitchboardConstants.SEARCH_RANKING_SOLR_DOUBLEDETECTION_QUANTRATE, Boost.RANKING.getQuantRate());
|
|
}
|
|
|
|
if (post != null && post.containsKey("ResetDoublecheck")) {
|
|
Boost.RANKING.setMinTokenLen(3);
|
|
Boost.RANKING.setQuantRate(0.5f);
|
|
sb.setConfig(SwitchboardConstants.SEARCH_RANKING_SOLR_DOUBLEDETECTION_MINLENGTH, Boost.RANKING.getMinTokenLen());
|
|
sb.setConfig(SwitchboardConstants.SEARCH_RANKING_SOLR_DOUBLEDETECTION_QUANTRATE, Boost.RANKING.getQuantRate());
|
|
}
|
|
|
|
if (post != null && post.containsKey("EnterRanking")) {
|
|
StringBuilder boostString = new StringBuilder(); // SwitchboardConstants.SEARCH_RANKING_SOLR_BOOST;
|
|
for (Map.Entry<String, String> entry: post.entrySet()) {
|
|
if (entry.getKey().startsWith("boost")) {
|
|
String fieldName = entry.getKey().substring(6);
|
|
YaCySchema field = YaCySchema.valueOf(fieldName);
|
|
if (field == null) continue;
|
|
try {
|
|
float boost = Float.parseFloat(entry.getValue());
|
|
if (boostString.length() > 0) boostString.append(',');
|
|
boostString.append(field.getSolrFieldName()).append('^').append(Float.toString(boost));
|
|
} catch (NumberFormatException e) {
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
if (boostString.length() > 0) {
|
|
String s = boostString.toString();
|
|
sb.setConfig(SwitchboardConstants.SEARCH_RANKING_SOLR_BOOST, s);
|
|
Boost.RANKING.updateBoosts(s);
|
|
}
|
|
|
|
}
|
|
|
|
if (post != null && post.containsKey("ResetRanking")) {
|
|
Boost.RANKING.initDefaults();
|
|
}
|
|
|
|
final serverObjects prop = new serverObjects();
|
|
prop.put("minTokenLen", Boost.RANKING.getMinTokenLen());
|
|
prop.put("quantRate", Boost.RANKING.getQuantRate());
|
|
int i = 0;
|
|
for (Map.Entry<YaCySchema, Float> entry: Boost.RANKING.entrySet()) {
|
|
YaCySchema field = entry.getKey();
|
|
float boost = entry.getValue();
|
|
prop.put("boosts_" + i + "_field", field.getSolrFieldName());
|
|
prop.put("boosts_" + i + "_boost", Float.toString(boost));
|
|
i++;
|
|
}
|
|
prop.put("boosts", i);
|
|
|
|
return prop;
|
|
}
|
|
|
|
}
|