db_router.py revision 9a535c9f3144690cf85d88a51e07b98c21454159 
1# Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5"""Django database Router
6
7Django gets configured with three database connections in frontend/settings.py.
8- The default database
9    - This database should be used for most things.
10    - For the master, this is the global database.
11    - For shards, this this is the shard-local database.
12- The global database
13    - For the master, this is the same database as default, which is the global
14      database.
15    - For the shards, this is the global database (the same as for the master).
16- The readonly connection
17    - This should be the same database as the global database, but it should
18      use an account on the database that only has readonly permissions.
19- The server database
20    - This is the database stores information about all servers in the Autotest
21      instance. Each instance, master or shard should have its own server
22      database is use_server_db is enabled in global config.
23
24The reason shards need two distinct databases for different objects is, that
25the tko parser should always write to the global database. Otherwise test
26results wouldn't be synced back to the master and would not be accessible in one
27place.
28
29Therefore this class will route all queries for tables starts with `server`
30prefix to server database, route all queries for tables that involve
31`tko_`-prefixed tables to the global database. For all others this router will
32not give a hint, which means the default database will be used.
33"""
34
35class Router(object):
36    """
37    Decide if an object should be written to the default or to the global db.
38
39    This is an implementaton of Django's multi-database router interface:
40    https://docs.djangoproject.com/en/1.5/topics/db/multi-db/
41    """
42
43    def _should_be_in_server_db(self, model):
44        """Return True if the model should be stored in the server db.
45
46        @param model: Model to decide for.
47
48        @return: True if querying the model requires server database.
49        """
50        return model._meta.db_table.startswith('server')
51
52
53    def _should_be_in_global(self, model):
54        """Returns True if the model should be stored in the global db.
55
56        @param model: Model to decide for.
57
58        @return: True if querying the model requires global database.
59        """
60        return model._meta.db_table.startswith('tko_')
61
62
63    def db_for_read(self, model, **hints):
64        """Return the database for a reading access.
65
66        @param model: Model to decide for.
67        @param hints: Optional arguments to determine which database for read.
68
69        @returns: 'server' for all server models. 'global' for all tko models,
70                  None otherwise. None means the router doesn't have an opinion.
71        """
72        if self._should_be_in_server_db(model):
73            return 'server'
74        if self._should_be_in_global(model):
75            return 'global'
76        return None
77
78
79    def db_for_write(self, model, **hints):
80        """Return the database for a writing access.
81
82        @param model: Model to decide for.
83        @param hints: Optional arguments to determine which database for write.
84
85        @returns: 'server' for all server models. 'global' for all tko models,
86                  None otherwise. None means the router doesn't have an opinion.
87        """
88        if self._should_be_in_server_db(model):
89            return 'server'
90        if self._should_be_in_global(model):
91            return 'global'
92        return None
93
94
95    def allow_relation(self, obj1, obj2, **hints):
96        """
97        Allow relations only if either both are in tko_ tables or none is.
98
99        @param obj1: First object involved in the relation.
100        @param obj2: Second object involved in the relation.
101        @param hints: Optional arguments to determine if relation is allowed.
102
103        @returns False, if the relation should be prohibited,
104                 None, if the router doesn't have an opinion.
105        """
106        if (not self._should_be_in_server_db(type(obj1)) ==
107            self._should_be_in_server_db(type(obj2))):
108            return False
109        if (not self._should_be_in_global(type(obj1)) ==
110            self._should_be_in_global(type(obj2))):
111            return False
112        return None
113