WebSVN – SmartDukaan – Diff – /trunk/profitmandi-cron/src/main/java/com/smartdukaan/cron/scheduled/ScheduledTasks.java

+	}
 	public void getBestSeller() throws ProfitMandiBusinessException {
 		WebListing webListing = webListingRepository.selectByUrl("partner-best-sellers");
-		List<WebProductListing> webProductListings = webProductListingRepository
-				.selectAllByWebListingId(webListing.getId());
 		List<Integer> catalogItemIds = orderRepository.selectAllOrderWeight(LocalDateTime.now().minusDays(30),
 				LocalDateTime.now(), OrderStatus.DELIVERY_SUCCESS);
+		Map<Integer, List<Item>> catalogItemsMap = itemRepository.selectAllByCatalogIds(new HashSet<>(catalogItemIds))
+				.stream().collect(Collectors.groupingBy(x -> x.getCatalogItemId()));
+		Map<String, Integer> brandCount = new HashMap<>();
 		LOGGER.info("catalogItemIds {}", catalogItemIds);
 		if (!catalogItemIds.isEmpty()) {
 			webProductListingRepository.deleteByEqual(webListing.getId());
 			int count = 0;
 			for (Integer catalogItemId : catalogItemIds) {
+				List<Item> items = catalogItemsMap.get(catalogItemId);
+				String brand = items.get(0).getBrand();
+				int brandItemCount = 0;
+				if (brandCount.containsKey(items.get(0).getBrand())) {
+					brandItemCount = brandCount.get(brand);
+				}
+				brandCount.put(brand, brandItemCount + 1);
+				if (brandCount.get(brand) > 3) {
+					continue;
+				}
 				WebProductListing webProductListing = new WebProductListing();
 				webProductListing.setEntityId(catalogItemId);
 				webProductListing.setWebListingId(webListing.getId());
 				count = count + 1;

Subversion Repositories SmartDukaan