BillCacheEvictionService.java

package gov.nysenate.openleg.service.bill.data;

import com.google.common.eventbus.EventBus;
import com.google.common.eventbus.Subscribe;
import gov.nysenate.openleg.model.agenda.Agenda;
import gov.nysenate.openleg.model.bill.BillId;
import gov.nysenate.openleg.model.calendar.Calendar;
import gov.nysenate.openleg.service.agenda.event.AgendaUpdateEvent;
import gov.nysenate.openleg.service.agenda.event.BulkAgendaUpdateEvent;
import gov.nysenate.openleg.service.calendar.event.BulkCalendarUpdateEvent;
import gov.nysenate.openleg.service.calendar.event.CalendarUpdateEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;

/**
 * Evicts bills from the cached bill data service cache based on events
 */
@Service
public class BillCacheEvictionService {
    Logger logger = LoggerFactory.getLogger(BillCacheEvictionService.class);

    @Autowired private EventBus eventBus;
    @Autowired private CachedBillDataService billDataService;

    @PostConstruct
    public void init() {
        eventBus.register(this);
    }

    @Subscribe
    public void handleCalendarUpdate(CalendarUpdateEvent calendarUpdateEvent) {
        evictCalendarBills(calendarUpdateEvent.getCalendar());
    }

    @Subscribe
    public void handleBulkCalendarUpdate(BulkCalendarUpdateEvent bulkCalendarUpdateEvent) {
        bulkCalendarUpdateEvent.getCalendars().forEach(this::evictCalendarBills);
    }

    @Subscribe
    public void handleAgendaUpdate(AgendaUpdateEvent agendaUpdateEvent) {
        evictAgendaBills(agendaUpdateEvent.getAgenda());
    }

    @Subscribe
    public void handleBulkAgendaUpdate(BulkAgendaUpdateEvent bulkAgendaUpdateEvent) {
        bulkAgendaUpdateEvent.getAgendas().forEach(this::evictAgendaBills);
    }

    private void evictCalendarBills(Calendar calendar) {
        calendar.getSupplementalMap().values().stream()
                .flatMap(calSup -> calSup.getAllEntries().stream())
                .map(calSupEntry -> BillId.getBaseId(calSupEntry.getBillId()))
                .forEach(billDataService::evictContent);
    }

    private void evictAgendaBills(Agenda agenda) {
        agenda.getAgendaInfoAddenda().values().stream()
                .flatMap(addendum -> addendum.getCommitteeInfoMap().values().stream())
                .flatMap(commInfo -> commInfo.getItems().stream())
                .map(infCommItem -> BillId.getBaseId(infCommItem.getBillId()))
                .forEach(billDataService::evictContent);
        agenda.getAgendaVoteAddenda().values().stream()
                .flatMap(voteAdd -> voteAdd.getCommitteeVoteMap().values().stream())
                .flatMap(voteComm -> voteComm.getVotedBills().keySet().stream())
                .map(BillId::getBaseId)
                .forEach(billDataService::evictContent);
    }
}