/testing/library/src/com/google/appengine/library/JDOBookDataService.java

// Copyright 2008 Google Inc. All Rights Reserved.

package com.google.appengine.library;

import java.util.Collection;

import javax.jdo.JDOHelper;
import javax.jdo.PersistenceManager;
import javax.jdo.PersistenceManagerFactory;
import javax.jdo.Query;

/**
 * @author kjin@google.com (Kevin Jin)
 */
final class JDOBookDataService implements BookDataService {

  private static final PersistenceManagerFactory pmf =
      JDOHelper.getPersistenceManagerFactory("transactional");
  private final PersistenceManager pm = pmf.getPersistenceManager();
  private static final String FROM_CLAUSE = "SELECT FROM " + Book.class.getName();

  @SuppressWarnings("unchecked")
  public Iterable<Book> asIterable(String jpqlQuery) {
    return (Iterable<Book>) pm.newQuery("javax.jdo.query.JPQL", FROM_CLAUSE + jpqlQuery).execute();
  }

  @SuppressWarnings("unchecked")
  public Iterable<Book> asIterable(String jpqlQuery, int limit, int offset) {
    Query query = pm.newQuery("javax.jdo.query.JPQL", FROM_CLAUSE + jpqlQuery);
    query.setRange(offset, offset + limit);
    return (Iterable<Book>) query.execute();
  }

  @SuppressWarnings("unchecked")
  public int countEntities(String jpqlQuery) {
    return ((Collection) asIterable(jpqlQuery)).size();
  }

  public void delete(Book book) {
    pm.deletePersistent(book);
  }

  public void put(Book book) {
    pm.makePersistent(book);
  }

  public void close() {
    pm.close();
  }
}