/**
* Abiquo community edition
* cloud management application for hybrid clouds
* Copyright (C) 2008-2010 - Abiquo Holdings S.L.
*
* This application is free software; you can redistribute it and/or
* modify it under the terms of the GNU LESSER GENERAL PUBLIC
* LICENSE as published by the Free Software Foundation under
* version 3 of the License
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* LESSER GENERAL PUBLIC LICENSE v.3 for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
package com.abiquo.server.core.task;
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.Transaction;
import com.abiquo.server.core.task.enums.TaskOwnerType;
/**
* Repository to manage the redis-based DAOs {@link TaskDAO} and {@link JobDAO}.
*
* @author eruiz@abiquo.com
*/
@Component
public class AsyncTaskRep
{
@Autowired
protected JedisPool jedisPool;
@Autowired
protected TaskDAO taskDao;
@Autowired
protected JobDAO jobDao;
public Task save(Task task)
{
Jedis jedis = jedisPool.getResource();
Transaction transaction = jedis.multi();
boolean discard = true;
try
{
// Persist referenced Jobs
for (Job job : task.getJobs())
{
job.setParentTaskId(task.getTaskId());
jobDao.save(job, transaction);
}
// Persist task
taskDao.save(task, transaction);
transaction.exec();
discard = false;
}
finally
{
if (discard)
{
transaction.discard();
}
jedisPool.returnResource(jedis);
}
return task;
}
public Job save(Job job)
{
Jedis jedis = jedisPool.getResource();
Transaction transaction = jedis.multi();
boolean discard = true;
try
{
jobDao.save(job, transaction);
transaction.exec();
discard = false;
}
finally
{
if (discard)
{
transaction.discard();
}
jedisPool.returnResource(jedis);
}
return job;
}
public void delete(Task task)
{
Jedis jedis = jedisPool.getResource();
Transaction transaction = jedis.multi();
boolean discard = true;
try
{
// Delete referenced jobs
for (Job job : task.getJobs())
{
jobDao.delete(job, transaction);
}
// Delete task
taskDao.delete(task, transaction);
transaction.exec();
discard = false;
}
finally
{
if (discard)
{
transaction.discard();
}
jedisPool.returnResource(jedis);
}
}
public List<Task> findTasksByOwnerId(final TaskOwnerType type, final String ownerId)
{
Jedis jedis = jedisPool.getResource();
try
{
List<Task> tasks = taskDao.findByOwnerId(type, ownerId, jedis);
for (Task task : tasks)
{
if (task != null)
{
task.getJobs().addAll(
jobDao.findJobs(taskDao.getTaskJobsEntityKey(task.getIdAsString()), jedis));
}
}
return tasks;
}
finally
{
jedisPool.returnResource(jedis);
}
}
public Task findTask(final String taskId)
{
Jedis jedis = jedisPool.getResource();
try
{
Task task = taskDao.findById(taskId, jedis);
if (task != null)
{
task.getJobs().addAll(
jobDao.findJobs(taskDao.getTaskJobsEntityKey(task.getIdAsString()), jedis));
}
return task;
}
finally
{
jedisPool.returnResource(jedis);
}
}
public Task findTaskByJobId(final String jobId)
{
Jedis jedis = jedisPool.getResource();
try
{
Job job = findJob(jobId);
if (job == null)
{
return null;
}
return findTask(job.getParentTaskId());
}
finally
{
jedisPool.returnResource(jedis);
}
}
public Job findJob(final String jobId)
{
Jedis jedis = jedisPool.getResource();
try
{
return jobDao.findById(jobId, jedis);
}
finally
{
jedisPool.returnResource(jedis);
}
}
}