package com.commafeed.backend.feeds; import java.util.Calendar; import java.util.Date; import java.util.List; import java.util.Queue; import javax.annotation.PostConstruct; import javax.inject.Inject; import javax.inject.Singleton; import org.apache.commons.lang3.time.DateUtils; import com.commafeed.backend.MetricsBean; import com.commafeed.backend.dao.FeedDAO; import com.commafeed.backend.model.Feed; import com.commafeed.backend.services.ApplicationSettingsService; import com.google.common.collect.Queues; @Singleton public class FeedRefreshTaskGiver { @Inject FeedDAO feedDAO; @Inject ApplicationSettingsService applicationSettingsService; @Inject MetricsBean metricsBean; private int backgroundThreads; private Queue addQueue = Queues.newConcurrentLinkedQueue(); private Queue takeQueue = Queues.newConcurrentLinkedQueue(); private Queue giveBackQueue = Queues.newConcurrentLinkedQueue(); @PostConstruct public void init() { backgroundThreads = applicationSettingsService.get() .getBackgroundThreads(); } public void add(Feed feed) { Date now = Calendar.getInstance().getTime(); boolean heavyLoad = applicationSettingsService.get().isHeavyLoad(); Date threshold = DateUtils.addMinutes(now, heavyLoad ? -10 : -1); if (feed.getLastUpdated() == null || feed.getLastUpdated().before(threshold)) { feed.setEtagHeader(null); feed.setLastModifiedHeader(null); } addQueue.add(feed); } public synchronized Feed take() { Feed feed = takeQueue.poll(); if (feed == null) { int count = 5 * backgroundThreads; List feeds = feedDAO.findNextUpdatable(count); int size = addQueue.size(); for (int i = 0; i < size; i++) { feeds.add(addQueue.poll()); } for (Feed f : feeds) { takeQueue.add(f); f.setLastUpdated(Calendar.getInstance().getTime()); } size = giveBackQueue.size(); for (int i = 0; i < size; i++) { feeds.add(giveBackQueue.poll()); } feedDAO.update(feeds); feed = takeQueue.poll(); } if (feed != null) { metricsBean.feedRefreshed(); } return feed; } public void giveBack(Feed feed) { metricsBean.feedUpdated(); giveBackQueue.add(feed); } }