org.energyos.espi.common.repositories.jpa.UsagePointRepositoryImpl.java Source code

Java tutorial

Introduction

Here is the source code for org.energyos.espi.common.repositories.jpa.UsagePointRepositoryImpl.java

Source

/*
 * Copyright 2013, 2014, 2015 EnergyOS.org
 *
 *    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 org.energyos.espi.common.repositories.jpa;

import java.util.GregorianCalendar;
import java.util.List;
import java.util.UUID;

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

import org.energyos.espi.common.domain.IdentifiedObject;
import org.energyos.espi.common.domain.RetailCustomer;
import org.energyos.espi.common.domain.ServiceCategory;
import org.energyos.espi.common.domain.Subscription;
import org.energyos.espi.common.domain.UsagePoint;
import org.energyos.espi.common.repositories.UsagePointRepository;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;

@Repository
@Transactional(rollbackFor = { javax.xml.bind.JAXBException.class }, noRollbackFor = {
        javax.persistence.NoResultException.class, org.springframework.dao.EmptyResultDataAccessException.class })
public class UsagePointRepositoryImpl implements UsagePointRepository {

    @PersistenceContext
    protected EntityManager em;

    public void setEntityManager(EntityManager em) {
        this.em = em;
    }

    @SuppressWarnings("unchecked")
    public List<UsagePoint> findAllByRetailCustomerId(Long id) {
        return (List<UsagePoint>) this.em.createNamedQuery(UsagePoint.QUERY_FIND_ALL_BY_RETAIL_CUSTOMER_ID)
                .setParameter("retailCustomerId", id).getResultList();
    }

    @Override
    public UsagePoint findById(Long id) {
        return (UsagePoint) this.em.createNamedQuery(UsagePoint.QUERY_FIND_BY_ID).setParameter("id", id)
                .getSingleResult();
    }

    @Transactional(rollbackFor = { javax.xml.bind.JAXBException.class }, noRollbackFor = {
            javax.persistence.NoResultException.class,
            org.springframework.dao.EmptyResultDataAccessException.class })
    @Override
    public void persist(UsagePoint up) {
        this.em.persist(up);
    }

    @Override
    @Transactional(rollbackFor = { javax.xml.bind.JAXBException.class }, noRollbackFor = {
            javax.persistence.NoResultException.class,
            org.springframework.dao.EmptyResultDataAccessException.class })
    public void associateByUUID(RetailCustomer retailCustomer, UUID uuid) {
        try {
            UsagePoint existingUsagePoint = findByUUID(uuid);
            existingUsagePoint.setRetailCustomer(retailCustomer);
            persist(existingUsagePoint);
        } catch (NoResultException e) {
            UsagePoint usagePoint = new UsagePoint();
            usagePoint.setUUID(uuid);
            usagePoint.setServiceCategory(new ServiceCategory(ServiceCategory.ELECTRICITY_SERVICE));
            usagePoint.setRetailCustomer(retailCustomer);
            persist(usagePoint);
        }
    }

    @Transactional(rollbackFor = { javax.xml.bind.JAXBException.class }, noRollbackFor = {
            javax.persistence.NoResultException.class,
            org.springframework.dao.EmptyResultDataAccessException.class })
    @Override
    public void createOrReplaceByUUID(UsagePoint usagePoint) {
        try {
            UsagePoint existingUsagePoint = findByUUID(usagePoint.getUUID());
            usagePoint.setId(existingUsagePoint.getId());
            if (usagePoint.getRetailCustomer() == null) {
                usagePoint.setRetailCustomer(existingUsagePoint.getRetailCustomer());
            }

            if (existingUsagePoint.getMeterReadings() != null) {
                usagePoint.setMeterReadings(existingUsagePoint.getMeterReadings());
            }

            if (existingUsagePoint.getLocalTimeParameters() != null) {
                usagePoint.setLocalTimeParameters(existingUsagePoint.getLocalTimeParameters());
            }

            if (existingUsagePoint.getRelatedLinks() != null) {
                usagePoint.setRelatedLinks(existingUsagePoint.getRelatedLinks());
            }

            if (existingUsagePoint.getSelfLink() != null) {
                usagePoint.setSelfLink(existingUsagePoint.getSelfLink());
            }

            if (existingUsagePoint.getUpLink() != null) {
                usagePoint.setUpLink(existingUsagePoint.getUpLink());
            }
            usagePoint.setPublished(existingUsagePoint.getPublished());
            usagePoint.setUpdated(existingUsagePoint.getUpdated());

            em.merge(usagePoint);
        } catch (NoResultException e) {
            usagePoint.setPublished(new GregorianCalendar());
            usagePoint.setUpdated(new GregorianCalendar());
            persist(usagePoint);
        }
    }

    @SuppressWarnings("unchecked")
    @Override
    public List<UsagePoint> findAllUpdatedFor(Subscription subscription) {
        return (List<UsagePoint>) this.em.createNamedQuery(UsagePoint.QUERY_FIND_ALL_UPDATED_FOR)
                .setParameter("lastUpdate", subscription.getLastUpdate()).getResultList();
    }

    @Transactional(rollbackFor = { javax.xml.bind.JAXBException.class }, noRollbackFor = {
            javax.persistence.NoResultException.class,
            org.springframework.dao.EmptyResultDataAccessException.class })
    @Override
    public void deleteById(Long id) {
        UsagePoint r = findById(id);
        em.remove(em.contains(r) ? r : em.merge(r));
    }

    @Override
    public IdentifiedObject findByRelatedHref(String href) {
        return (UsagePoint) this.em.createNamedQuery(UsagePoint.QUERY_FIND_BY_RELATED_HREF)
                .setParameter("href", href).getSingleResult();
    }

    @SuppressWarnings("unchecked")
    @Override
    public List<Long> findAllIdsForRetailCustomer(Long retailCustomerId) {
        return (List<Long>) this.em.createNamedQuery(UsagePoint.QUERY_FIND_ALL_IDS_FOR_RETAIL_CUSTOMER)
                .setParameter("retailCustomerId", retailCustomerId).getResultList();
    }

    @Override
    public UsagePoint findByURI(String uri) {
        return (UsagePoint) em.createNamedQuery(UsagePoint.QUERY_FIND_BY_URI).setParameter("uri", uri)
                .getSingleResult();
    }

    @Override
    public UsagePoint findByUUID(UUID uuid) {
        return (UsagePoint) this.em.createNamedQuery(UsagePoint.QUERY_FIND_BY_UUID)
                .setParameter("uuid", uuid.toString().toUpperCase()).getSingleResult();
    }

    @SuppressWarnings("unchecked")
    @Override
    public List<Long> findAllIds() {
        return (List<Long>) this.em.createNamedQuery(UsagePoint.QUERY_FIND_ALL_IDS).getResultList();
    }

}