Subversion Repositories SmartDukaan

import com.spice.profitmandi.common.enumuration.NotificationFormat;

import com.spice.profitmandi.common.enumuration.ScheduleStatus;

import com.spice.profitmandi.common.model.NotificationTargetCriteria;

import org.springframework.transaction.annotation.Propagation;

import org.springframework.transaction.annotation.Transactional;

    NotificationScheduleRepository notificationScheduleRepository;

    @Autowired

        NotificationTargetCriteria criteria = new NotificationTargetCriteria(

                sendNotificationModel.getUserIds(),

                sendNotificationModel.getFofoIds(),

                sendNotificationModel.getStateIds(),

                sendNotificationModel.getRegionIds());

        NotificationCampaign nc = buildAndPersistCampaign(sendNotificationModel, criteria);

        List<LocalDateTime> schedule = sendNotificationModel.getScheduledTimestamps();

        if (schedule != null && !schedule.isEmpty()) {

            LocalDateTime now = LocalDateTime.now();

            for (LocalDateTime ts : schedule) {

                NotificationSchedule ns = new NotificationSchedule();

                ns.setCampaignId(nc.getId());

                ns.setScheduledTimestamp(ts);

                ns.setStatus(ScheduleStatus.SCHEDULED);

                ns.setCreatedTimestamp(now);

                notificationScheduleRepository.persist(ns);

            }

            return;

        }

        Set<Integer> userIds = resolveTargetUserIds(criteria);

        if (userIds.isEmpty()) {

            LOGGER.info("Failed to send notification to any retailer with this model - {}", sendNotificationModel);

            return;

        }

        fanOutToUsers(nc.getId(), userIds, nc.getFormat());

    }

    private NotificationCampaign buildAndPersistCampaign(SendNotificationModel model, NotificationTargetCriteria criteria) {

        // Pop-up / Story fields — silently no-op for PUSH since model getters return null.

        scp.setButtonLabel(model.getButtonLabel());

        scp.setShowCta(model.getShowCta());

        scp.setShowClose(model.getShowClose());

        scp.setAutoCloseSeconds(model.getAutoCloseSeconds());

        scp.setFrequencyCap(model.getFrequencyCap());

        scp.setLiveFrom(model.getLiveFrom());

        scp.setLiveUntil(model.getLiveUntil());

        scp.setStoryCards(model.getStoryCards());

        scp.setPostInCategory(model.getPostInCategory());

        scp.setPlayDurationSeconds(model.getPlayDurationSeconds());

        scp.setExpiresAfterPolicy(model.getExpiresAfterPolicy());

        scp.setLinkedEntities(model.getLinkedEntities());

        nc.setTargetCriteria(gson.toJson(criteria));

        nc.setFormat(model.getFormat() != null ? model.getFormat() : NotificationFormat.PUSH);

        nc.setCategory(model.getCategory());

        nc.setCampaignGroupId(model.getCampaignGroupId());

        return nc;

    }

    @Override

    public Set<Integer> resolveTargetUserIds(NotificationTargetCriteria criteria) throws ProfitMandiBusinessException {

                        .map(x -> x.getFofoId()).collect(Collectors.toList());

        return userIds;

    }

    private void fanOutToUsers(int campaignId, Set<Integer> userIds, NotificationFormat format) {

        // FCM/APNs delivery only for PUSH. Pop-ups and Stories are pull-based (mobile fetches on app open),

        // so we still write user_campaign rows above but skip pushnotifications.

        if (format == null || format == NotificationFormat.PUSH) {

    }

    @Override

    public void dispatchScheduled(int scheduleId) throws ProfitMandiBusinessException {

        NotificationSchedule ns = notificationScheduleRepository.selectById(scheduleId);

        if (ns == null) {

            LOGGER.warn("dispatchScheduled: schedule id {} not found", scheduleId);

            return;

        }

        if (ns.getStatus() != ScheduleStatus.SCHEDULED) {

            LOGGER.debug("dispatchScheduled: skip id {} in status {}", scheduleId, ns.getStatus());

            return;

        }

        // Atomic claim: only one worker transitions SCHEDULED -> DISPATCHED; others get 0 rows.

        int claimed = notificationScheduleRepository.claimForDispatch(scheduleId, LocalDateTime.now());

        if (claimed == 0) {

            LOGGER.debug("dispatchScheduled: id {} already claimed by another worker", scheduleId);

            return;

        }

        try {

            NotificationCampaign campaign = notificationCampaignRepository.selectById(ns.getCampaignId());

            if (campaign == null) {

                LOGGER.error("dispatchScheduled: campaign {} missing for schedule {}", ns.getCampaignId(), scheduleId);

                notificationScheduleRepository.markFailed(scheduleId);

                return;

            }

            NotificationTargetCriteria criteria = campaign.getTargetCriteria() == null

                    ? new NotificationTargetCriteria()

                    : gson.fromJson(campaign.getTargetCriteria(), NotificationTargetCriteria.class);

            Set<Integer> userIds = resolveTargetUserIds(criteria);

            if (userIds.isEmpty()) {

                LOGGER.info("dispatchScheduled: no users resolved for schedule {} (campaign {})",

                        scheduleId, campaign.getId());

                return;

            }

            fanOutToUsers(campaign.getId(), userIds, campaign.getFormat());

        } catch (RuntimeException e) {

            // Swallow: markFailed must commit, and cron shouldn't abort the batch on one bad row.

            LOGGER.error("dispatchScheduled: fan-out failed for schedule {}", scheduleId, e);

            notificationScheduleRepository.markFailed(scheduleId);

        }

    }

    @Override

    @Transactional(propagation = Propagation.REQUIRES_NEW, rollbackFor = Throwable.class)

    public int dispatchDueScheduled(int batchSize) throws ProfitMandiBusinessException {

        List<Integer> due = notificationScheduleRepository.selectDueScheduledIds(LocalDateTime.now(), batchSize);

        if (due.isEmpty()) {

            return 0;

        }

        LOGGER.info("dispatchDueScheduled: dispatching {} scheduled notification(s)", due.size());

        for (Integer id : due) {

            dispatchScheduled(id);

        }

        return due.size();

    }

    @Override

    public boolean cancelSchedule(int scheduleId) {

        return notificationScheduleRepository.cancelIfScheduled(scheduleId) == 1;

    }

    @Override

    public int resolveAudienceSize(NotificationTargetCriteria criteria) throws ProfitMandiBusinessException {

        if (criteria == null) return 0;

        return resolveTargetUserIds(criteria).size();

    }

    @Override

    public int addScheduleForCampaign(int campaignId, LocalDateTime scheduledAt) throws ProfitMandiBusinessException {

        NotificationCampaign nc = notificationCampaignRepository.selectById(campaignId);

        if (nc == null) {

            throw new ProfitMandiBusinessException("Campaign not found: ", campaignId, "- campaignId");

        }

        NotificationSchedule ns = new NotificationSchedule();

        ns.setCampaignId(campaignId);

        ns.setScheduledTimestamp(scheduledAt);

        ns.setStatus(ScheduleStatus.SCHEDULED);

        ns.setCreatedTimestamp(LocalDateTime.now());

        notificationScheduleRepository.persist(ns);

        LOGGER.info("addScheduleForCampaign: campaignId={} scheduledAt={} newScheduleId={}",

                campaignId, scheduledAt, ns.getId());

        return ns.getId();