work towards volumes using db layer

2 files changed, 22 insertions, 1 deletions

diff --git a/nova/db/api.py b/nova/db/api.py
index e4d79d16f..edc3b7bdc 100644
--- a/nova/db/api.py
+++ b/nova/db/api.py
@@ -195,7 +195,6 @@ def instance_get_by_name(context, name):
     return _impl.instance_get_by_project(context, name)
 
 
-
 def instance_get_by_project(context, project_id):
     """Get all instance belonging to a project."""
     return _impl.instance_get_by_project(context, project_id)
@@ -356,6 +355,16 @@ def volume_get(context, volume_id):
     return _impl.volume_get(context, volume_id)
 
 
+def volume_get_all(context):
+    """Get all volumes."""
+    return _impl.volume_get_all(context)
+
+
+def volume_get_by_project(context, project_id):
+    """Get all volumes belonging to a project."""
+    return _impl.volume_get_by_project(context, project_id)
+
+
 def volume_get_shelf_and_blade(context, volume_id):
     """Get the shelf and blade allocated to the volume."""
     return _impl.volume_get_shelf_and_blade(context, volume_id)
diff --git a/nova/db/sqlalchemy/api.py b/nova/db/sqlalchemy/api.py
index 2c5434b8f..2ce54a1d7 100644
--- a/nova/db/sqlalchemy/api.py
+++ b/nova/db/sqlalchemy/api.py
@@ -441,6 +441,18 @@ def volume_get(context, volume_id):
     return models.Volume.find(volume_id)
 
 
+def volume_get_all(context):
+    return models.Volume.all()
+
+
+def volume_get_by_project(context, project_id):
+    session = models.NovaBase.get_session()
+    query = session.query(models.Volume)
+    results = query.filter_by(project_id=project_id).all()
+    session.commit()
+    return results
+
+
 def volume_get_shelf_and_blade(context, volume_id):
     volume_ref = volume_get(context, volume_id)
     export_device = volume_ref.export_device