RestaurantDao.java

package com.springone.myrestaurants.dao;

import java.util.ArrayList;
import java.util.List;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.stereotype.Repository;
import org.springframework.web.client.RestTemplate;

import com.springone.myrestaurants.config.CouchDbConfig;
import com.springone.myrestaurants.domain.Restaurant;
import com.springone.myrestaurants.web.CouchDbMappingJacksonHttpMessageConverter;

@Repository
public class RestaurantDao {

	//@PersistenceContext
	//private EntityManager entityManager;

	private RestTemplate restTemplate;

	public RestaurantDao() {
		restTemplate = new RestTemplate();

		List<HttpMessageConverter<?>> converters = new ArrayList<HttpMessageConverter<?>>();
		//Converter specific to processing CouchDB view queries.
		converters.add(new CouchDbMappingJacksonHttpMessageConverter());
		restTemplate.setMessageConverters(converters);
	}


	public Restaurant findRestaurant(String id) {
		if (id == null)
			return null;
		List<Restaurant> rests = findAllRestaurants();
		for (Restaurant restaurant : rests) {
			if (restaurant.getId().equals(id)) {
				return restaurant;
			}
		}
		return null;
		//return entityManager.find(Restaurant.class, id);
	}

	@SuppressWarnings("unchecked")
	public List<Restaurant> findAllRestaurants() {
		
		return (List<Restaurant>) restTemplate.getForObject(CouchDbConfig.URL + "_design/demo/_view/all", Restaurant.class);
		//return entityManager.createQuery("select o from Restaurant o").getResultList();
	}

	@SuppressWarnings("unchecked")
	public List<Restaurant> findRestaurantEntries(int firstResult,
			int maxResults) {
		
		// TODO respect bounds
		return findAllRestaurants();
		/*
		return entityManager.createQuery("select o from Restaurant o")
				.setFirstResult(firstResult).setMaxResults(maxResults)
				.getResultList();*/
	}

	public long countRestaurants() {
		
		//TODO optimize with the creation of common views for counts.
		return ((List<Restaurant>) restTemplate.getForObject(CouchDbConfig.URL + "_design/demo/_view/all", Restaurant.class)).size();
		/*
		return ((Number) entityManager.createQuery(
				"select count(o) from Restaurant o").getSingleResult())
				.longValue();*/
	}

}