package com.visualbuilder.session;

import java.util.Collection;

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

import com.visualbuilder.entity.Book;

@Stateless
public class Library implements LibraryLocal {
	
	@PersistenceContext
	protected EntityManager em;
	
	public void addBook(String title, String author) {
		
		Book book = new Book(title, author);
		em.persist(book);
		
	}

	public void addBook(Book book) {
		
		em.persist(book);
		
	}

	public void deleteBook(Integer id) {
		
		Book book = em.find(Book.class, new Integer(id));
		em.remove(book);
	}

	public Collection<Book> getBooks() {
		
		Collection<Book> results = em.createQuery("from Book b").getResultList();
		
		return results;
	}

}