com.zutubi.pulse.master.model.persistence.hibernate.HibernateAgentDailyStatisticsDao.java Source code

Introduction

Here is the source code for com.zutubi.pulse.master.model.persistence.hibernate.HibernateAgentDailyStatisticsDao.java

Source

/* Copyright 2017 Zutubi Pty Ltd
 *
 * 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.zutubi.pulse.master.model.persistence.hibernate;

import com.zutubi.pulse.master.model.AgentDailyStatistics;
import com.zutubi.pulse.master.model.persistence.AgentDailyStatisticsDao;
import com.zutubi.util.logging.Logger;
import org.hibernate.HibernateException;
import org.hibernate.Query;
import org.hibernate.Session;
import org.springframework.orm.hibernate4.HibernateCallback;

import java.sql.SQLException;
import java.util.List;
import java.util.Set;

/**
 * Hibernate-specific implementation of {@link AgentDailyStatisticsDao}.
 */
@SuppressWarnings({ "unchecked" })
public class HibernateAgentDailyStatisticsDao extends HibernateEntityDao<AgentDailyStatistics>
        implements AgentDailyStatisticsDao {
    private static final Logger LOG = Logger.getLogger(HibernateAgentDailyStatisticsDao.class);

    public Class<AgentDailyStatistics> persistentClass() {
        return AgentDailyStatistics.class;
    }

    public List<AgentDailyStatistics> findByAgent(final long agentId) {
        return getHibernateTemplate().execute(new HibernateCallback<List<AgentDailyStatistics>>() {
            public List<AgentDailyStatistics> doInHibernate(Session session) throws HibernateException {
                Query queryObject = session.createQuery("from AgentDailyStatistics where agentId = :agentId");
                queryObject.setLong("agentId", agentId);
                return queryObject.list();
            }
        });
    }

    public AgentDailyStatistics findByAgentAndDaySafe(final long agentId, final long dayStamp) {
        List<AgentDailyStatistics> results = getHibernateTemplate()
                .execute(new HibernateCallback<List<AgentDailyStatistics>>() {
                    public List<AgentDailyStatistics> doInHibernate(Session session) throws HibernateException {
                        Query queryObject = session.createQuery(
                                "from AgentDailyStatistics where agentId = :agentId and dayStamp = :dayStamp");
                        queryObject.setLong("agentId", agentId);
                        queryObject.setLong("dayStamp", dayStamp);
                        return queryObject.list();
                    }
                });

        if (results.size() == 0) {
            return null;
        } else if (results.size() == 1) {
            return results.get(0);
        } else {
            LOG.warning("Expected unique result for agent id '" + agentId + "' and day '" + dayStamp + "', but got "
                    + results.size() + " results.  Resetting statistics for this day.");
            for (AgentDailyStatistics stats : results) {
                delete(stats);
            }

            return null;
        }
    }

    public int deleteByDayStampBefore(final long dayStamp) {
        return getHibernateTemplate().execute(new HibernateCallback<Integer>() {
            public Integer doInHibernate(Session session) throws HibernateException {
                Query queryObject = session
                        .createQuery("delete from AgentDailyStatistics where dayStamp < :dayStamp");
                queryObject.setLong("dayStamp", dayStamp);
                return queryObject.executeUpdate();
            }
        });
    }

    public int deleteByAgentNotIn(final Set<Long> agentIds) {
        return getHibernateTemplate().execute(new HibernateCallback<Integer>() {
            public Integer doInHibernate(Session session) throws HibernateException {
                Query queryObject = session
                        .createQuery("delete from AgentDailyStatistics where agentId not in (:agentIds)");
                queryObject.setParameterList("agentIds", agentIds);
                return queryObject.executeUpdate();
            }
        });
    }
}