WebSVN – SmartDukaan – Diff – //trunk/profitmandi-dao/src/main/java/com/spice/profitmandi/dao/service/solr/FofoSolr.java

 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.apache.solr.client.solrj.SolrClient;
 import org.apache.solr.client.solrj.impl.HttpSolrClient;
-import org.apache.solr.client.solrj.response.UpdateResponse;
+import org.apache.solr.client.solrj.response.schema.SchemaResponse.UpdateResponse;
 import org.apache.solr.common.SolrInputDocument;
 import org.json.JSONArray;
 import org.json.JSONException;
 import org.json.JSONObject;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 import com.google.gson.Gson;
 import com.spice.profitmandi.dao.entity.catalog.TagListing;
 import com.spice.profitmandi.dao.entity.catalog.TagRanking;
+import com.spice.profitmandi.dao.entity.dtr.WebProductListing;
 import com.spice.profitmandi.dao.repository.catalog.TagListingRepository;
 import com.spice.profitmandi.dao.repository.catalog.TagRankingRepository;
 import com.spice.profitmandi.dao.repository.dtr.Mongo;
+import com.spice.profitmandi.dao.repository.dtr.WebListingRepository;
+import com.spice.profitmandi.dao.repository.dtr.WebProductListingRepository;
 import com.spice.profitmandi.service.tag.ItemTagModel;
 import in.shop2020.model.v1.catalog.status;
 @Component
 	TagRankingRepository tagRankingRepository;
 	@Autowired
 	private Mongo contentMongoClient;
+	@Autowired
+	private WebListingRepository webListingRepository;
+	@Autowired
+	private WebProductListingRepository webProductListingRepository;
 	private static final Logger logger = LogManager.getLogger(FofoSolr.class);
 	private static final List<Integer> CATEGORY_MASTER = Arrays.asList(10006, 10010, 14202, 14203);
 	String solrPath = "http://localhost:8984/solr/demo";
+		}
 		return availabilityItemMap;
+	}
+	public Map<Integer, List<String>> getLabels() {
+		List<String> labels = Arrays.asList("partner-best-sellers", "partner-price-drop", "new-launches");
+		Map<Integer, String> webListing = webListingRepository.selectByUrls(labels).stream()
+				.collect(Collectors.toMap(x -> x.getId(), x -> x.getUrl()));
+		Map<Integer, List<WebProductListing>> webProductListingMap = webProductListingRepository
+				.selectAllByWebListingIds(new ArrayList<>(webListing.keySet())).stream()
+				.collect(Collectors.groupingBy(x -> x.getWebListingId()));
+		Map<Integer, List<String>> catalogLabelMap = new HashMap<>();
+		for (Entry<Integer, List<WebProductListing>> webProductListingEntry : webProductListingMap.entrySet()) {
+			for (WebProductListing webProduct : webProductListingEntry.getValue()) {
+				if (!catalogLabelMap.containsKey(webProduct.getEntityId())) {
+					catalogLabelMap.put(webProduct.getEntityId(), new ArrayList<>());
+				}
+				List<String> itemLabels = catalogLabelMap.get(webProduct.getEntityId());
+				itemLabels.add(webListing.get(webProduct.getWebListingId()));
+				catalogLabelMap.put(webProduct.getEntityId(), itemLabels);
+			}
+		}
+		return catalogLabelMap;
+	}
 	public void populateTagItems() throws Exception {
 		Map<String, Map<String, Map<String, Integer>>> availabilityItemMap = getItemMap();
 		List<TagRanking> tagRankingList = tagRankingRepository.getAllTagRanking();
 		List<Integer> rankingList = new ArrayList<>();
 		Map<Integer, String> featureMap = new HashMap<>();
 			rankingList.add(tagRanking.getCatalogItemId());
 			featureMap.put(tagRanking.getCatalogItemId(), tagRanking.getFeature());
+		}
 		Map<Integer, Map<String, Object>> catalogMap = new HashMap<>();
-		List<status> statuses = new ArrayList<>();
+		List<status> statuses = Arrays.asList(status.ACTIVE, status.PAUSED_BY_RISK, status.PARTIALLY_ACTIVE);
-		statuses.add(status.ACTIVE);
+		Map<Integer, List<String>> catalogLabelMap = this.getLabels();
-		statuses.add(status.PAUSED_BY_RISK);
+		logger.info("catalogLabelMap {}", catalogLabelMap);
+		/*
-		statuses.add(status.PARTIALLY_ACTIVE);
+		 * List<ItemTagModel> tuples =
-	/*	List<ItemTagModel> tuples = tagListingRepository.getAllItemTagByStatus(statuses).stream()
+		 * tagListingRepository.getAllItemTagByStatus(statuses).stream() .filter(x ->
-				.filter(x -> x.getItem().getCatalogItemId() == 1023747).collect(Collectors.toList());*/
+		 * x.getItem().getCatalogItemId() == 1023747).collect(Collectors.toList());
+		 */
 		List<ItemTagModel> tuples = tagListingRepository.getAllItemTagByStatus(statuses);
 		Map<String, Object> projection = new HashMap<>();
 		projection.put("defaultImageUrl", 1);
-		List<String> excludeBrands = Arrays.asList("Live Demo", "FOC", "Dummy", "FOC HANDSET");
+		List<String> excludeBrands = Arrays.asList("Live Demo", "Dummy", "FOC HANDSET");
 		for (ItemTagModel result : tuples) {
 			TagListing tag = result.getTagListing();
 			com.spice.profitmandi.dao.entity.catalog.Item item = result.getItem();
 			if (excludeBrands.contains(item.getBrand())) {
 				Map<String, Object> catalogObj = new HashMap<>();
 				catalogObj.put("feature", "");
 				catalogObj.put("title",
 						String.join(" ", Arrays.asList(item.getBrand(), item.getModelName(), item.getModelNumber())
 								.stream().filter(s -> s != null && !s.isEmpty()).collect(Collectors.toList())));
-				catalogObj.put("brand", item.getBrand());
+				catalogObj.put("brand", new String[] { item.getBrand() });
 				if (item.getBrand().equals("Huawei") || item.getBrand().equals("Honor")) {
 					catalogObj.put("brand", new String[] { "Huawei", "Honor" });
+				}
 				if (item.getBrand().equals("Mi") || item.getBrand().equals("Xiaomi")
 						|| item.getBrand().equals("Redmi")) {
 				if (item.getCategoryId() == 10006) {
 					catalogObj.put("categoryId", 3);
+				}
 				catalogObj.put("subCategoryId", item.getCategoryId());
+				catalogObj.put("create_timestamp",
-				catalogObj.put("create_timestamp", tag.getCreatedTimestamp().atZone(ZoneId.systemDefault())
+						tag.getCreatedTimestamp().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
-					    .toInstant().toEpochMilli());
+				if (catalogLabelMap.containsKey(item.getCatalogItemId())
+						&& catalogLabelMap.get(item.getCatalogItemId()) != null) {
+					catalogObj.put("labels", catalogLabelMap.get(item.getCatalogItemId()));
+				}
 				catalogMap.put(item.getCatalogItemId(), catalogObj);
 				catalogObj.put("avColor", new HashMap<>());
+			}
 			Map<String, Object> catalogObj = catalogMap.get(item.getCatalogItemId());
 			Map<Integer, Integer> avColorMap = (Map<Integer, Integer>) catalogObj.get("avColor");
 			SolrInputDocument catalogSolrObj = new SolrInputDocument();
 			catalogSolrObj.setField("id", "catalog" + catalogId);
 			catalogSolrObj.setField("rank_i", catalogValMap.get("rank"));
 			catalogSolrObj.setField("title_s", catalogValMap.get("title"));
-		//	catalogSolrObj.setField("_childDocuments_", itemObjs);
+			// catalogSolrObj.setField("_childDocuments_", itemObjs);
 			catalogSolrObj.addChildDocuments(itemObjs);
 			catalogSolrObj.setField("child_b", itemObjs.size() > 0);
 			catalogSolrObj.setField("catalogId_i", catalogId);
 			catalogSolrObj.setField("imageUrl_s",
 					catalogValMap.get("imageUrl").toString().replace("saholic", "smartdukaan"));
 			catalogSolrObj.setField("feature_s", catalogValMap.get("feature"));
 			catalogSolrObj.setField("brand_ss", catalogValMap.get("brand"));
 			catalogSolrObj.setField("create_s", catalogValMap.get("create_timestamp"));
 			catalogSolrObj.setField("categoryId_i", catalogValMap.get("categoryId"));
 			catalogSolrObj.setField("subCategoryId_i", catalogValMap.get("subCategoryId"));
+			catalogSolrObj.setField("label_ss", catalogValMap.get("labels"));
 			catalogSolrObj.setField("w7573_i", 0);
 			catalogSolrObj.setField("w7678_i", 0);
 			catalogSolrObj.setField("w7720_i", 0);
 			catalogSolrObj.setField("w8468_i", 0);
 			catalogSolrObj.setField("w8889_i", 0);
 			catalogSolrObj.setField("w8947_i", 0);
 			catalogSolrObj.setField("w9213_i", 0);
 			catalogSolrObj.setField("w9203_i", 0);
 			catalogSolrObj.setField("mop_f", mop);
+			catalogSolrObj.setField("show_default_b", false);
+			String[] brands = (String[]) catalogValMap.get("brand");
+			for (String brand : brands) {
+				if (brand != "FOC") {
+					catalogSolrObj.setField("show_default_b", true);
+				}
+			}
 			Map<Integer, Integer> avColorMap = (Map<Integer, Integer>) catalogValMap.get("avColor");
 			for (Entry<Integer, Integer> avColorEntry : avColorMap.entrySet()) {
 				int whId = avColorEntry.getKey();
 				int color = (int) avColorEntry.getValue();
 		logger.info("catalogSolrObjs {}", catalogSolrObjs);
 		solr.deleteByQuery("*:*");
 		solr.add(catalogSolrObjs);
-		UpdateResponse updateResponse = solr.commit();
+		org.apache.solr.client.solrj.response.UpdateResponse updateResponse = solr.commit();
 		logger.info("solr {}", updateResponse.getStatus());
+	}
 	/*

Subversion Repositories SmartDukaan

(root)//trunk/profitmandi-dao/src/main/java/com/spice/profitmandi/dao/service/solr/FofoSolr.java – Rev 31604 → 31612