Athou_commafeed/src/main/java/com/commafeed/backend/services/FeedUpdateService.java

package com.commafeed.backend.services;

import java.util.Date;
import java.util.List;

import javax.ejb.Stateless;
import javax.inject.Inject;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;

import com.commafeed.backend.MetricsBean;
import com.commafeed.backend.cache.CacheService;
import com.commafeed.backend.dao.FeedEntryDAO;
import com.commafeed.backend.dao.FeedEntryDAO.EntryWithFeed;
import com.commafeed.backend.dao.FeedEntryStatusDAO;
import com.commafeed.backend.dao.FeedSubscriptionDAO;
import com.commafeed.backend.feeds.FeedUtils;
import com.commafeed.backend.model.Feed;
import com.commafeed.backend.model.FeedEntry;
import com.commafeed.backend.model.FeedEntryContent;
import com.commafeed.backend.model.FeedFeedEntry;
import com.commafeed.backend.model.FeedSubscription;
import com.commafeed.backend.model.User;
import com.google.common.collect.Lists;

@Stateless
public class FeedUpdateService {
	
	@PersistenceContext
	protected EntityManager em;

	@Inject
	FeedSubscriptionDAO feedSubscriptionDAO;

	@Inject
	FeedEntryDAO feedEntryDAO;

	@Inject
	FeedEntryStatusDAO feedEntryStatusDAO;

	@Inject
	MetricsBean metricsBean;

	@Inject
	CacheService cache;

	public void updateEntry(Feed feed, FeedEntry entry,
			List<FeedSubscription> subscriptions) {

		EntryWithFeed existing = feedEntryDAO.findExisting(entry.getGuid(),
				entry.getUrl(), feed.getId());

		FeedEntry update = null;
		FeedFeedEntry ffe = null;
		if (existing == null) {
			entry.setAuthor(FeedUtils.truncate(FeedUtils.handleContent(
					entry.getAuthor(), feed.getLink(), true), 128));
			FeedEntryContent content = entry.getContent();
			content.setTitle(FeedUtils.truncate(FeedUtils.handleContent(
					content.getTitle(), feed.getLink(), true), 2048));
			content.setContent(FeedUtils.handleContent(content.getContent(),
					feed.getLink(), false));

			entry.setInserted(new Date());
			ffe = new FeedFeedEntry(feed, entry);

			update = entry;
		} else if (existing.ffe == null) {
			ffe = new FeedFeedEntry(feed, existing.entry);
			update = existing.entry;
		}

		if (update != null) {
			List<User> users = Lists.newArrayList();
			for (FeedSubscription sub : subscriptions) {
				users.add(sub.getUser());
			}
			cache.invalidateUserData(users.toArray(new User[0]));
			feedEntryDAO.saveOrUpdate(update);
			em.persist(ffe);
		}
	}
}