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.common.SolrInputDocument;
+import org.apache.solr.common.SolrInputField;
 import org.json.JSONArray;
 import org.json.JSONException;
 import org.json.JSONObject;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
         for (TagRanking tagRanking : tagRankingList) {
             rankingList.add(tagRanking.getCatalogItemId());
             featureMap.put(tagRanking.getCatalogItemId(), tagRanking.getFeature());
+        }
         //Get Similar Models
-        List<SimilarModel> similarModels = similarModelRepository.selectAll();
+        /*List<SimilarModel> similarModels = similarModelRepository.selectAll();
-        Map<Integer, List<Integer>> similarModelsMap = similarModels.stream().collect(Collectors.groupingBy(SimilarModel::getCatalogId, Collectors.mapping(x -> x.getSimilarCatalogId(), Collectors.toList())));
+        Map<Integer, List<Integer>> similarModelsMap = similarModels.stream().collect(Collectors.groupingBy(SimilarModel::getCatalogId, Collectors.mapping(x -> x.getSimilarCatalogId(), Collectors.toList())));*/
         Map<Integer, Map<String, Object>> catalogMap = new HashMap<>();
         List<status> statuses = Arrays.asList(status.ACTIVE, status.PAUSED_BY_RISK, status.PARTIALLY_ACTIVE);
         Map<Integer, List<String>> catalogLabelMap = this.getLabels();
         logger.info("catalogLabelMap {}", catalogLabelMap);
-        /*
-         * List<ItemTagModel> tuples =
-         * tagListingRepository.getAllItemTagByStatus(statuses).stream() .filter(x ->
-         * x.getItem().getCatalogItemId() == 1023747).collect(Collectors.toList());
-         */
         List<ItemTagModel> itemTagModels = tagListingRepository.getAllItemTagByStatus(statuses);
         Map<String, Object> projection = new HashMap<>();
         projection.put("defaultImageUrl", 1);
             if (excludeBrands.contains(item.getBrand())) {
                 continue;
+            }
             if (!catalogMap.containsKey(item.getCatalogItemId())) {
                 if (!priceModelMap.containsKey(tagListing.getSellingPrice()) && item.getCategoryId() == 10006) {
                     priceModelMap.put(tagListing.getSellingPrice(), new ArrayList<>());
+                }
-                if(item.getCategoryId()==10006) {
+                if (item.getCategoryId() == 10006) {
                     List<Integer> priceModels = priceModelMap.get(tagListing.getSellingPrice());
                     priceModels.add(item.getCatalogItemId());
+                }
                 Map<String, Object> catalogObj = new HashMap<>();
+                catalogObj.put("stockColor", 0);
                 catalogObj.put("feature", "");
                 catalogObj.put("hsnCode", item.getHsnCode());
                 catalogObj.put("title",
                         String.join(" ", Arrays.asList(item.getBrand(), item.getModelName(), item.getModelNumber())
                                 .stream().filter(s -> s != null && !s.isEmpty()).collect(Collectors.toList())));
             Map<String, Object> catalogObj = catalogMap.get(item.getCatalogItemId());
             Map<Integer, Integer> avColorMap = (Map<Integer, Integer>) catalogObj.get("avColor");
             if (availabilityItemMap.containsKey(String.valueOf(item.getId()))) {
+                int itemStockColor = 0;
                 for (Map.Entry<String, Map<String, Integer>> entry : availabilityItemMap
                         .get(String.valueOf(item.getId())).entrySet()) {
                     String warehouseId = entry.getKey();
                     Map<String, Integer> avMap = entry.getValue();
                     if (!avColorMap.containsKey(Integer.parseInt(warehouseId))) {
                         color = Math.max(color, 0);
+                    }
                     avColorMap.put(Integer.parseInt(warehouseId), color);
                     catalogObj.put("avColor", avColorMap);
+                    if (color > itemStockColor) {
+                        itemStockColor = color;
+                    }
+                }
+                if (itemStockColor > 0 && itemStockColor > (Integer) catalogObj.get("stockColor")) {
+                    catalogObj.put("stockColor", itemStockColor);
+                }
+            }
             Map<Integer, Object> itemColorMap = (Map<Integer, Object>) catalogObj.get("items");
             if (!itemColorMap.containsKey(item.getId())) {
             boolean active = false;
             // Map<Integer, Object> itemsMap = (Map<String, Object>)
             // catalogValMap.get("items");
             List<SolrInputDocument> itemObjs = new ArrayList<>();
             float mop = 0;
-            float dp = 0;
             for (Entry<Integer, Object> itemEntry : itemsMap.entrySet()) {
                 int itemId = itemEntry.getKey();
                 Map<String, Object> itemMap = (Map<String, Object>) itemEntry.getValue();
                 List<TagListing> tags = (List<TagListing>) itemMap.get("tagPricing");
                     itemObj.setField("mop_f", taglist.getMop());
                     itemObj.setField("sellingPrice_f", taglist.getSellingPrice());
                     itemObj.setField("active_b", taglist.isActive());
                     itemObj.setField("hot_deal_b", taglist.isHotDeals());
                     mop = taglist.getMop();
-                    dp = taglist.getSellingPrice();
+                    catalogValMap.put("dp", taglist.getSellingPrice());
+                }
                 itemObjs.add(itemObj);
+            }
-            float starPrice = dp - 500f;
-            float endPrice = dp + 1000f;
-            List<Integer> similalarModels = null;
-            if ((Integer) catalogValMap.get("categoryId") == 3) {
-                similalarModels = new ArrayList<>();
-                for (Entry<Float, List<Integer>> floatListEntry : priceModelMap.entrySet()) {
-                    float modelPrice = floatListEntry.getKey();
-                    if (modelPrice >= starPrice && modelPrice <= endPrice) {
-                        similalarModels.addAll(floatListEntry.getValue());
-                    }
-                    if (modelPrice > endPrice) break;
-                }
-                similalarModels.remove(Integer.valueOf(catalogId));
-            }
-            logger.info("catalogValMap {}", catalogValMap);
+            //logger.info("catalogValMap {}", catalogValMap);
-            logger.info("rank {}", catalogValMap.get("rank"));
+            //logger.info("rank {}", catalogValMap.get("rank"));
             SolrInputDocument catalogSolrObj = new SolrInputDocument();
             catalogSolrObj.setField("id", "catalog" + catalogId);
             catalogSolrObj.setField("rank_i", catalogValMap.get("rank"));
-            catalogSolrObj.setField("similarModels_ii", similalarModels == null ? Collections.EMPTY_LIST : similalarModels);
             catalogSolrObj.setField("title_s", catalogValMap.get("title"));
             // catalogSolrObj.setField("_childDocuments_", itemObjs);
             catalogSolrObj.addChildDocuments(itemObjs);
             catalogSolrObj.setField("child_b", itemObjs.size() > 0);
             catalogSolrObj.setField("catalogId_i", catalogId);
             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"));
-            for (Integer warehouseId : ProfitMandiConstants.WAREHOUSE_MAP.keySet()) {
-                catalogSolrObj.setField("w" + warehouseId + "_i", 0);
-            }
             catalogSolrObj.setField("mop_f", mop);
             catalogSolrObj.setField("hsncode_s", catalogValMap.get("hsnCode"));
             catalogSolrObj.setField("show_default_b", true);
             String[] brands = (String[]) catalogValMap.get("brand");
+                }
+            }
             Map<Integer, Integer> avColorMap = (Map<Integer, Integer>) catalogValMap.get("avColor");
+            for (Integer warehouseId : ProfitMandiConstants.WAREHOUSE_MAP.keySet()) {
+                catalogSolrObj.setField("w" + warehouseId + "_i", 0);
+            }
             for (Entry<Integer, Integer> avColorEntry : avColorMap.entrySet()) {
                 int whId = avColorEntry.getKey();
                 int color = avColorEntry.getValue();
                 catalogSolrObj.setField("w" + whId + "_i", color);
+            }
+            List<Integer> similalarModels = null;
+            if ((Integer) catalogValMap.get("categoryId") == 3) {
+                float dp = (Float) catalogValMap.get("dp");
+                float starPrice = dp - 500f;
+                float endPrice = dp + 1000f;
+                similalarModels = new ArrayList<>();
+                for (Entry<Float, List<Integer>> floatListEntry : priceModelMap.entrySet()) {
+                    float modelPrice = floatListEntry.getKey();
+                    if (modelPrice >= starPrice && modelPrice <= endPrice) {
+                        List<Integer> instock = floatListEntry.getValue().stream().filter(x->catalogMap.get(x).get("stockColor") != null && (Integer) (catalogMap.get(x).get("stockColor"))>0).collect(Collectors.toList());
+                        logger.info("/Entry before - {}, after - {}", floatListEntry.getValue().size(), instock.size());
+                        similalarModels.addAll(instock);
+                    }
+                    if (modelPrice > endPrice) break;
+                }
+                similalarModels.remove(Integer.valueOf(catalogId));
+            }
+            catalogSolrObj.setField("similarModels_ii", similalarModels == null ? Collections.EMPTY_LIST : similalarModels);
             catalogSolrObj.setField("active_b", active);
             logger.info("catalogSolrObj {}", catalogSolrObj.get("rank_i"));
             catalogSolrObjs.add(catalogSolrObj);
+        }
         logger.info("catalogSolrObjs {}", catalogSolrObjs);
         String solrPath = "http://" + solrUrl + ":8984/solr/demo";
         SolrClient solr = new HttpSolrClient.Builder(solrPath).build();

Subversion Repositories SmartDukaan

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