From 2986d31e0486080f998fabd1d3f46ad44bf79b89 Mon Sep 17 00:00:00 2001
From: prachi <prachi@cloud.com>
Date: Wed, 2 May 2012 14:52:22 -0700
Subject: [PATCH] Adding back code that S3 merge overwrote

---
 .../src/com/cloud/bridge/persist/EntityDao.java  | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/awsapi/src/com/cloud/bridge/persist/EntityDao.java b/awsapi/src/com/cloud/bridge/persist/EntityDao.java
index be846a3167b..d9bac7bc1b1 100644
--- a/awsapi/src/com/cloud/bridge/persist/EntityDao.java
+++ b/awsapi/src/com/cloud/bridge/persist/EntityDao.java
@@ -69,29 +69,29 @@ public class EntityDao<T> {
 	
 	@SuppressWarnings("unchecked")
 	public T get(Serializable id) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		return (T)session.get(clazz, id);
 	}
 	
 	public T save(T entity) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		session.saveOrUpdate(entity);
 		return entity;
 	}
 	
 	public T update(T entity) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		session.saveOrUpdate(entity);
 		return entity;
 	}
 	
 	public void delete(T entity) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		session.delete(entity);
 	}
 	
 	public T queryEntity(String hql, Object[] params) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		Query query = session.createQuery(hql);
 		query.setMaxResults(1);
 		QueryHelper.bindParameters(query, params);
@@ -99,7 +99,7 @@ public class EntityDao<T> {
 	}
 	
 	public List<T> queryEntities(String hql, Object[] params) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		Query query = session.createQuery(hql);
 		QueryHelper.bindParameters(query, params);
 		
@@ -107,7 +107,7 @@ public class EntityDao<T> {
 	}
 	
 	public List<T> queryEntities(String hql, int offset, int limit, Object[] params) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		Query query = session.createQuery(hql);
 		QueryHelper.bindParameters(query, params);
 		query.setFirstResult(offset);
@@ -116,7 +116,7 @@ public class EntityDao<T> {
 	}
 	
 	public int executeUpdate(String hql, Object[] params) {
-		Session session = PersistContext.getSession();
+		Session session = PersistContext.getSession(isCloudStackSession);
 		Query query = session.createQuery(hql);
 		QueryHelper.bindParameters(query, params);