Subversion Repositories SmartDukaan

    /**

     * Updates a single catalog in Solr based on catalogId.

     * Used for real-time updates when TagListing changes.

     *

     * @param catalogId The catalog ID to update

     */

    public void updateSingleCatalog(int catalogId) throws Exception {

        logger.info("Updating single catalog in Solr: catalogId={}", catalogId);

        List<status> statuses = Arrays.asList(status.ACTIVE, status.PAUSED_BY_RISK, status.PARTIALLY_ACTIVE);

        // Get all TagListings for this catalog

        List<ItemTagModel> itemTagModels = tagListingRepository.getItemTagByCatalogIdAndStatus(catalogId, statuses);

        if (itemTagModels.isEmpty()) {

            logger.warn("No active TagListings found for catalogId={}, removing from Solr", catalogId);

            String solrPath = "http://" + solrUrl + ":8984/solr/demo";

            SolrClient solr = new HttpSolrClient.Builder(solrPath).build();

            solr.deleteById("catalog" + catalogId);

            solr.commit();

            return;

        }

        // Get availability for items in this catalog

        Map<String, Map<String, Map<String, Integer>>> availabilityItemMap = getItemMap();

        // Get ranking info

        List<TagRanking> tagRankingList = tagRankingRepository.getAllTagRanking();

        List<Integer> rankingList = new ArrayList<>();

        Map<Integer, String> featureMap = new HashMap<>();

        for (TagRanking tagRanking : tagRankingList) {

            rankingList.add(tagRanking.getCatalogItemId());

            featureMap.put(tagRanking.getCatalogItemId(), tagRanking.getFeature());

        }

        // Get labels

        Map<Integer, List<String>> catalogLabelMap = this.getLabels();

        // Get EOL info

        List<Integer> eolWithOutStock = catalogRepository.findAllWithEOLWithOutStock();

        List<Integer> eolWithBadQty = catalogRepository.findAllWithNoGoodStock();

        Set<Integer> eolSet = new HashSet<>(eolWithOutStock);

        eolSet.addAll(eolWithBadQty);

        // Build catalog document

        com.spice.profitmandi.dao.entity.catalog.Item firstItem = itemTagModels.get(0).getItem();

        TagListing firstTagListing = itemTagModels.get(0).getTagListing();

        Map<Integer, Integer> avColorMap = new HashMap<>();

        List<SolrInputDocument> itemObjs = new ArrayList<>();

        boolean active = false;

        float mop = 0;

        float dp = 0;

        int stockColor = 0;

        for (ItemTagModel itemTagModel : itemTagModels) {

            TagListing tagListing = itemTagModel.getTagListing();

            com.spice.profitmandi.dao.entity.catalog.Item item = itemTagModel.getItem();

            active = active || tagListing.isActive();

            // Build item document

            SolrInputDocument itemObj = new SolrInputDocument();

            itemObj.setField("id", "itemtag-" + item.getId() + "-" + tagListing.getTagId());

            itemObj.setField("color_s", item.getColor().replace("f_", ""));

            itemObj.setField("itemId_i", item.getId());

            itemObj.setField("tagId_i", tagListing.getTagId());

            itemObj.setField("mrp_f", tagListing.getMrp());

            itemObj.setField("mop_f", tagListing.getMop());

            itemObj.setField("sellingPrice_f", tagListing.getSellingPrice());

            itemObj.setField("active_b", tagListing.isActive());

            itemObj.setField("hot_deal_b", tagListing.isHotDeals());

            mop = tagListing.getMop();

            dp = tagListing.getSellingPrice();

            itemObjs.add(itemObj);

            // Calculate availability color

            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))) {

                        avColorMap.put(Integer.parseInt(warehouseId), 0);

                    }

                    int color = avColorMap.get(Integer.parseInt(warehouseId));

                    if (avMap.get("netAvailability") > 0) {

                        color = 2;

                    } else if (avMap.get("netPo") > 0) {

                        color = Math.max(color, 1);

                    } else {

                        color = Math.max(color, 0);

                    }

                    avColorMap.put(Integer.parseInt(warehouseId), color);

                    if (color > itemStockColor) {

                        itemStockColor = color;

                    }

                }

                if (itemStockColor > stockColor) {

                    stockColor = itemStockColor;

                }

            }

        }

        // Build catalog Solr document

        SolrInputDocument catalogSolrObj = new SolrInputDocument();

        catalogSolrObj.setField("id", "catalog" + catalogId);

        catalogSolrObj.setField("rank_i", rankingList.indexOf(catalogId));

        catalogSolrObj.setField("title_s", String.join(" ", Arrays.asList(

                firstItem.getBrand(), firstItem.getModelName(), firstItem.getModelNumber())

                .stream().filter(s -> s != null && !s.isEmpty()).collect(Collectors.toList())));

        catalogSolrObj.addChildDocuments(itemObjs);

        catalogSolrObj.setField("child_b", itemObjs.size() > 0);

        catalogSolrObj.setField("catalogId_i", catalogId);

        // Super catalog mapping

        List<SuperCatalogModel> superCatalogMappingList = superCatalogMappingRepository.selectJoinedData();

        List<SuperCatalogMapping> superCatalogs = superCatalogMappingRepository.selectAll();

        Optional<SuperCatalogModel> superCatalogMapping = findMappingByCatalogId(superCatalogMappingList, catalogId, itemTagModels);

        if (superCatalogMapping.isPresent()) {

            int superCatalogId = superCatalogMapping.get().getSuperCatalogId();

            String superCatalogMappingVariants = findMappingVariantsByCatalogId(superCatalogs, superCatalogId, itemTagModels);

            catalogSolrObj.setField("superCatalog_i", superCatalogId);

            catalogSolrObj.setField("superCatalog_s", superCatalogMapping.get().getSuperCatalogName());

            catalogSolrObj.setField("superCatalogVariants_s", superCatalogMappingVariants);

        } else {

            catalogSolrObj.setField("superCatalog_i", 0);

            catalogSolrObj.setField("superCatalog_s", catalogSolrObj.getFieldValue("title_s"));

            catalogSolrObj.setField("superCatalogVariants_s", "[]");

        }

        // Image URL

        try {

            String imageUrl = contentMongoClient.getEntityById(catalogId).getDefaultImageUrl();

            if (imageUrl != null) {

                catalogSolrObj.setField("imageUrl_s", imageUrl.replace("saholic", "smartdukaan"));

            } else {

                catalogSolrObj.setField("imageUrl_s",

                        "https://images.smartdukaan.com/uploads/campaigns/" + catalogId + ".jpg");

            }

        } catch (Exception e) {

            catalogSolrObj.setField("imageUrl_s",

                    "https://images.smartdukaan.com/uploads/campaigns/" + catalogId + ".jpg");

        }

        // Feature

        if (featureMap.containsKey(catalogId) && featureMap.get(catalogId) != null) {

            catalogSolrObj.setField("feature_s", featureMap.get(catalogId));

        } else {

            catalogSolrObj.setField("feature_s", "");

        }

        // Brand

        String[] brands = new String[]{firstItem.getBrand()};

        if (firstItem.getBrand().equals("Huawei") || firstItem.getBrand().equals("Honor")) {

            brands = new String[]{"Huawei", "Honor"};

        }

        if (firstItem.getBrand().equals("Mi") || firstItem.getBrand().equals("Xiaomi")

                || firstItem.getBrand().equals("Redmi")) {

            brands = new String[]{"Mi", "Xiaomi", "Redmi"};

        }

        catalogSolrObj.setField("brand_ss", brands);

        // Category

        catalogSolrObj.setField("categoryId_i", CATEGORY_MASTER.contains(firstItem.getCategoryId())

                ? (firstItem.getCategoryId() == 10006 ? 3 : firstItem.getCategoryId()) : 6);

        catalogSolrObj.setField("subCategoryId_i", firstItem.getCategoryId());

        // Timestamps and labels

        catalogSolrObj.setField("create_s", firstTagListing.getCreatedTimestamp()

                .atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());

        if (catalogLabelMap.containsKey(catalogId) && catalogLabelMap.get(catalogId) != null) {

            catalogSolrObj.setField("label_ss", catalogLabelMap.get(catalogId));

        }

        catalogSolrObj.setField("mop_f", mop);

        catalogSolrObj.setField("dp_f", dp);

        catalogSolrObj.setField("hsncode_s", firstItem.getHsnCode());

        catalogSolrObj.setField("show_default_b", true);

        for (String brand : brands) {

            if (brand.equals("FOC") || brand.equals("Live Demo")) {

                catalogSolrObj.setField("show_default_b", false);

            }

        }

        // Warehouse availability

        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);

        }

        // Similar models - simplified for single update

        catalogSolrObj.setField("similarModels_ii", Collections.EMPTY_LIST);

        catalogSolrObj.setField("active_b", active);

        catalogSolrObj.setField("eol_no_stock_b", eolSet.contains(catalogId));

        // Push to Solr

        String solrPath = "http://" + solrUrl + ":8984/solr/demo";

        SolrClient solr = new HttpSolrClient.Builder(solrPath).build();

        // Delete existing and add updated document

        solr.deleteById("catalog" + catalogId);

        solr.add(catalogSolrObj);

        solr.commit();

        logger.info("Successfully updated catalogId={} in Solr", catalogId);

    }