Statistics
| Branch: | Tag: | Revision:

root / snf-astakos-app / astakos / im / managers.py @ 7deaaa5f

History | View | Annotate | Download (4.4 kB)

1
# Copyright 2012 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or without
4
# modification, are permitted provided that the following conditions
5
# are met:
6
#
7
#   1. Redistributions of source code must retain the above copyright
8
#      notice, this list of conditions and the following disclaimer.
9
#
10
#  2. Redistributions in binary form must reproduce the above copyright
11
#     notice, this list of conditions and the following disclaimer in the
12
#     documentation and/or other materials provided with the distribution.
13
#
14
# THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
15
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17
# ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
18
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24
# SUCH DAMAGE.
25
#
26
# The views and conclusions contained in the software and documentation are
27
# those of the authors and should not be interpreted as representing official
28
# policies, either expressed or implied, of GRNET S.A.
29

    
30
from django.db import connections
31
from django.db.models import Manager
32
from django.db.models.query import QuerySet
33

    
34

    
35
class ForUpdateManager(Manager):
36
    """ Model manager implementing SELECT .. FOR UPDATE statement
37

38
        This manager implements select_for_update() method in order to use
39
        row-level locking in the database and guarantee exclusive access, since
40
        this method is only implemented in Django>=1.4.
41

42
        Non-blocking reads are not implemented, and each query including a row
43
        that is locked by another transaction will block until the lock is
44
        released. Also care must be taken in order to avoid deadlocks or retry
45
        transactions that abort due to deadlocks.
46

47
        Example:
48
            networks = Network.objects.select_for_update().filter(public=True)
49

50
    """
51

    
52
    def __init__(self, *args, **kwargs):
53
        super(ForUpdateManager, self).__init__(*args, **kwargs)
54
        self._select_for_update = False
55

    
56
    def filter(self, *args, **kwargs):
57
        query = self.get_query_set().filter(*args, **kwargs)
58
        if self._select_for_update:
59
            self._select_for_update = False
60
            return for_update(query)
61
        else:
62
            return query
63

    
64
    def get(self, *args, **kwargs):
65
        if not self._select_for_update:
66
            return self.get_query_set().get(*args, **kwargs)
67

    
68
        query = self.filter(*args, **kwargs)
69
        query = list(query)
70
        num = len(query)
71
        if num == 1:
72
            return query[0]
73
        if not num:
74
            raise self.model.DoesNotExist(
75
                    "%s matching query does not exist. "
76
                    "Lookup parameters were %s" %
77
                    (self.model._meta.object_name, kwargs))
78
        raise self.model.MultipleObjectsReturned(
79
            "get() returned more than one %s -- it returned %s! "
80
            "Lookup parameters were %s" %
81
            (self.model._meta.object_name, num, kwargs))
82

    
83
    def select_for_update(self, *args, **kwargs):
84
        self._select_for_update = True
85
        return self
86

    
87

    
88
def for_update(query):
89
    """ Rewrite query using SELECT .. FOR UPDATE.
90

91
    """
92
    if 'sqlite' in connections[query.db].settings_dict['ENGINE'].lower():
93
        # SQLite  does not support FOR UPDATE
94
        return query
95
    sql, params = query.query.get_compiler(query.db).as_sql()
96
    return query.model._default_manager.raw(sql.rstrip() + ' FOR UPDATE',
97
                                            params)
98

    
99

    
100
class ProtectedDeleteManager(ForUpdateManager):
101
    """ Manager for protecting Backend deletion.
102

103
        Call Backend delete() method in order to prevent deletion
104
        of Backends that host non-deleted VirtualMachines.
105

106
    """
107

    
108
    def get_query_set(self):
109
        return BackendQuerySet(self.model, using=self._db)
110

    
111

    
112
class BackendQuerySet(QuerySet):
113
    def delete(self):
114
        for backend in self._clone():
115
            backend.delete()