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();*/
}
}