com.intuit.tank.dao.JobQueueDao.java Source code

Introduction

Here is the source code for com.intuit.tank.dao.JobQueueDao.java

Source

/**
 * Copyright 2011 Intuit Inc. All Rights Reserved
 */
package com.intuit.tank.dao;

/*
 * #%L
 * Data Access
 * %%
 * Copyright (C) 2011 - 2015 Intuit Inc.
 * %%
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * #L%
 */

import java.util.Calendar;
import java.util.Date;
import java.util.HashSet;
import java.util.List;

import javax.annotation.Nonnull;
import javax.persistence.Query;

import org.apache.log4j.Logger;
import org.hibernate.LockOptions;

import com.intuit.tank.project.JobInstance;
import com.intuit.tank.project.JobQueue;

/**
 * ProductDao
 * 
 * @author dangleton
 * 
 */
public class JobQueueDao extends BaseDao<JobQueue> {
    private static final Logger LOG = Logger.getLogger(JobQueueDao.class);

    /**
     * @param entityClass
     */
    public JobQueueDao() {
        super();
        setReloadEntities(true);
    }

    /**
     * 
     * @param projectId
     * @return
     */
    public synchronized JobQueue findOrCreateForProjectId(@Nonnull int projectId) {
        String prefix = "x";
        JobQueue result = null;
        NamedParameter parameter = new NamedParameter(JobQueue.PROPERTY_PROJECT_ID, "pId", projectId);
        StringBuilder sb = new StringBuilder();
        sb.append(buildQlSelect(prefix)).append(startWhere())
                .append(buildWhereClause(Operation.EQUALS, prefix, parameter));
        List<JobQueue> resultList = super.listWithJQL(sb.toString(), parameter);
        if (resultList.size() > 0) {
            result = resultList.get(0);
        }
        if (resultList.size() > 1) {
            LOG.warn("Have " + resultList.size() + " queues for project " + projectId);
        }
        if (result == null) {
            result = new JobQueue(projectId);
            result = saveOrUpdate(result);
        } else {
            getHibernateSession().refresh(result, LockOptions.READ);
        }
        return result;
    }

    /**
     * 
     * @param projectId
     * @return
     */
    public List<JobQueue> getForProjectIds(@Nonnull List<Integer> projectIds) {
        String prefix = "x";
        NamedParameter parameter = new NamedParameter(JobQueue.PROPERTY_PROJECT_ID, "pId", projectIds);
        StringBuilder sb = new StringBuilder();
        sb.append(buildQlSelect(prefix)).append(startWhere())
                .append(buildWhereClause(Operation.IN, prefix, parameter));
        List<JobQueue> resultList = super.listWithJQL(sb.toString(), parameter);
        return resultList;
    }

    public List<JobQueue> findRecent() {
        String prefix = "x";
        Calendar c = Calendar.getInstance();
        c.add(Calendar.DAY_OF_YEAR, -5);
        NamedParameter parameter = new NamedParameter(JobQueue.PROPERTY_MODIFIED, "m", c.getTime());
        StringBuilder sb = new StringBuilder();
        sb.append(buildQlSelect(prefix)).append(startWhere())
                .append(buildWhereClause(Operation.GREATER_THAN, prefix, parameter));
        List<JobQueue> resultList = super.listWithJQL(sb.toString(), parameter);
        //        LOG.info("Retrieved " + resultList.size() + " from query " + sb.toString());
        return resultList;
    }

    public JobQueue findForJobId(Integer jobId) {
        JobQueue ret = null;
        try {
            String string = "select x.test_id from test_instance_jobs x where x.job_id = ?";
            Query q = getEntityManager().createNativeQuery(string);
            q.setParameter(1, jobId);
            Integer result = (Integer) q.getSingleResult();
            if (result != null) {
                ret = findById(result);
            }
        } catch (Exception e) {
            LOG.error("Error finding for Job ID: " + e, e);
        }
        return ret;
    }
}