private void initEvaluation() {
int port = 8983;
RecommenderEvaluator recommenderEval = new RecommenderEvaluator();
UserService userService = new UserService("localhost", port, "collection3");
RecommendService recService = new RecommendService("localhost", port, "collection2");
ItemService itemService = new ItemService("localhost", port, "collection1");
SolrServiceContainer.getInstance().setUserService(userService);
SolrServiceContainer.getInstance().setRecommendService(recService);
SolrServiceContainer.getInstance().setItemService(itemService);
users = recommenderEval.getAllUsers();
System.out.println("User Size " + users.size());
}
@Override
public RecommendResponse recommend(RecommendQuery query, Integer maxReuslts) {
productsToFilter = new ArrayList<String>();
String queryString = "*:*";
String sortCriteria = "user_count_purchased desc";
ModifiableSolrParams solrParams = new ModifiableSolrParams();
QueryResponse response = null;
RecommendResponse searchResponse = new RecommendResponse();
try {
String filterQueryString =
RecommendationQueryUtils.buildFilterForContentBasedFiltering(contentFilter);
if (query.getProductIds() != null && query.getProductIds().size() > 0) {
productsToFilter.addAll(query.getProductIds());
}
if (query.getUser() != null) {
if (alreadyBoughtProducts != null) {
productsToFilter.addAll(alreadyBoughtProducts);
}
}
if (productsToFilter != null && productsToFilter.size() > 0) {
if (filterQueryString.length() > 0) {
filterQueryString += " OR ";
}
filterQueryString +=
RecommendationQueryUtils.buildFilterForAlreadyBoughtProducts(productsToFilter);
}
solrParams.set("q", queryString);
solrParams.set("fq", filterQueryString);
solrParams.set("sort", sortCriteria);
solrParams.set("rows", maxReuslts);
response =
SolrServiceContainer.getInstance()
.getRecommendService()
.getSolrServer()
.query(solrParams);
// fill response object
List<String> extractedRecommendations =
RecommendationQueryUtils.extractRecommendationIds(
response.getBeans(CustomerAction.class));
searchResponse.setResultItems(extractedRecommendations);
searchResponse.setElapsedTime(response.getElapsedTime());
SolrDocumentList docResults = response.getResults();
searchResponse.setNumFound(docResults.getNumFound());
} catch (SolrServerException e) {
e.printStackTrace();
searchResponse.setNumFound(0);
searchResponse.setResultItems(new ArrayList<String>());
searchResponse.setElapsedTime(-1);
}
return searchResponse;
}
@Override
public RecommendResponse recommend(RecommendQuery query, Integer maxReuslts) {
ModifiableSolrParams solrParams = new ModifiableSolrParams();
QueryResponse response = null;
RecommendResponse searchResponse = new RecommendResponse();
long step0ElapsedTime = 0;
long step1ElapsedTime;
List<String> recommendations = new ArrayList<String>();
try {
// STEP 0 - get products from a user
if (query.getUser() != null) {
if (query.getProductIds() == null || query.getProductIds().size() == 0) {
if (alreadyBoughtProducts != null) {
query.setProductIds(alreadyBoughtProducts);
} else {
}
}
}
solrParams = getInteractionsFromMeAndUSersThatILikedOrCommented(query.getUser());
response =
SolrServiceContainer.getInstance()
.getSocialActionService()
.getSolrServer()
.query(solrParams);
step1ElapsedTime = response.getElapsedTime();
List<SocialAction> socialUsers = response.getBeans(SocialAction.class);
if (socialUsers.size() == 0) {
searchResponse.setNumFound(0);
searchResponse.setResultItems(recommendations);
searchResponse.setElapsedTime(-1);
return searchResponse;
}
SocialAction currentUserInteractions = socialUsers.get(0);
if (currentUserInteractions.getUserId().equals(query.getUser())) {
socialUsers.remove(0);
} else {
currentUserInteractions = null;
}
final Map<String, Integer> userInteractionMap = new HashMap<String, Integer>();
if (currentUserInteractions != null) {
List<String> usersThatPostedASnapshopToMe =
currentUserInteractions.getUsersThatPostedASnapshopToMe();
if (usersThatPostedASnapshopToMe != null) {
fillInteractions(usersThatPostedASnapshopToMe, userInteractionMap);
}
}
for (SocialAction socialUser : socialUsers) {
Integer userInteraction = userInteractionMap.get(socialUser.getUserId());
if (userInteraction == null) {
userInteraction = 0;
}
if (socialUser.getUsersThatPostedASnapshopToMe() != null) {
userInteraction +=
Collections.frequency(socialUser.getUsersThatPostedASnapshopToMe(), query.getUser());
}
userInteractionMap.put(socialUser.getUserId(), userInteraction);
}
Comparator<String> interactionCountComparator =
new Comparator<String>() {
@Override
public int compare(String a, String b) {
if (userInteractionMap.get(a) > userInteractionMap.get(b)) {
return -1;
} else if (userInteractionMap.get(a).equals(userInteractionMap.get(b))) {
return 0;
} else {
return 1;
}
}
};
TreeMap<String, Integer> sorted_map =
new TreeMap<String, Integer>(interactionCountComparator);
sorted_map.putAll(userInteractionMap);
solrParams = getSTEP2Params(query, maxReuslts, sorted_map);
// TODO Facet for confidence value
response =
SolrServiceContainer.getInstance().getResourceService().getSolrServer().query(solrParams);
// fill response object
List<Resource> beans = response.getBeans(Resource.class);
searchResponse.setResultItems(RecommendationQueryUtils.extractRecommendationIds(beans));
searchResponse.setElapsedTime(
step0ElapsedTime + step1ElapsedTime + response.getElapsedTime());
SolrDocumentList docResults = response.getResults();
searchResponse.setNumFound(docResults.getNumFound());
} catch (Exception e) {
System.out.println(solrParams);
e.printStackTrace();
searchResponse.setNumFound(0);
searchResponse.setResultItems(recommendations);
searchResponse.setElapsedTime(-1);
}
return searchResponse;
}