com.gwtplatform.crawlerservice.server.service.ObjectifyDao Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gwtp-crawler-service Show documentation
There is a newer version: 1.6
/**
 * Copyright 2011 ArcBees Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package com.gwtplatform.crawlerservice.server.service;

import java.lang.reflect.ParameterizedType;
import java.util.List;
import java.util.Map;

import com.googlecode.objectify.Key;
import com.googlecode.objectify.Objectify;
import com.googlecode.objectify.cmd.LoadType;
import com.googlecode.objectify.cmd.Query;
import com.gwtplatform.crawlerservice.server.objectify.OfyService;

/**
 * Generic DAO for use with Objectify.
 *
 * @param  The type of the entity handled by this DAO.
 */
public abstract class ObjectifyDao {
    protected final Class clazz;

    private Objectify lazyOfy;

    @SuppressWarnings("unchecked")
    public ObjectifyDao() {
        clazz = (Class) ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments()[0];
    }

    public LoadType load() {
        return ofy().load().type(clazz);
    }

    public Key put(T entity) {
        return ofy().save().entity(entity).now();
    }

    public Map, T> putAll(Iterable entities) {
        return ofy().save().entities(entities).now();
    }

    public void delete(T entity) {
        ofy().delete().entity(entity);
    }

    public void deleteKey(Key entityKey) {
        ofy().delete().entity(entityKey);
    }

    public void deleteAll(Iterable entities) {
        ofy().delete().entities(entities);
    }

    public void deleteKeys(Iterable> keys) {
        ofy().delete().keys(keys);
    }

    public T get(Long id) {
        return load().id(id).now();
    }

    public T get(Key key) {
        return ofy().load().key(key).now();
    }

    public Map, T> get(Iterable> keys) {
        return ofy().load().keys(keys);
    }

    public List listAll(int start, int length) {
        Query q = load().offset(start).limit(length);
        return q.list();
    }

    public int countAll() {
        return load().count();
    }

    public List listByProperty(String propName, Object propValue) {
        Query q = load();
        q.filter(propName, propValue);
        return q.list();
    }

    public List> listKeysByProperty(String propName, Object propValue) {
        Query q = load();
        q.filter(propName, propValue);
        return q.keys().list();
    }

    public Key getKey(Long id) {
        return Key.create(clazz, id);
    }

    public List listChildren(Object parent) {
        return load().ancestor(parent).list();
    }

    public List> listChildKeys(Object parent) {
        return load().ancestor(parent).keys().list();
    }

    protected Objectify ofy() {
        if (lazyOfy == null) {
            lazyOfy = OfyService.ofy();
        }
        return lazyOfy;
    }
}