Example #1
0
 def test_nesting(self):
     set_thread_readwrite_db()
     assert is_thread_readwrite()
     with readonly_db():
         assert not is_thread_readwrite()
         with readwrite_db():
             assert is_thread_readwrite()
         assert not is_thread_readwrite()
     assert is_thread_readwrite()
Example #2
0
 def test_readwrite(self):
     self.assertFalse(was_db_written())
     set_thread_readwrite_db()
     master = 'my_master_db'
     with override_settings(MASTER_DATABASE=master):
         router = DBRouter()
         out = router.db_for_read(Submission)
         self.assertEqual(master, out)
         out = router.db_for_write(Submission)
         self.assertEqual(master, out)
     self.assertTrue(was_db_written())
Example #3
0
def update_recent_events():
    """
    Compute recent events and other statistics and cache them.
    """
    logger.debug("update_recent_events: started")

    try:
        # No middleware on tasks, so this won't get set otherwise.
        # Tell the DB router this thread only needs to read the DB, not write.
        # We run this task frequently, so it's worthwhile.
        set_thread_readonly_db()

        votes = Vote.objects.select_related(
            "voter", "voter__user", "submission").filter(
            submission__approved=True, submission__duplicate_of__isnull=True).exclude(
            voter=F('submission__voter')).order_by("-id")[:10]
        submissions = Submission.objects.select_related(
            "voter", "voter__user").filter(
            approved=True, duplicate_of__isnull=True).order_by("-id")[:10]

        entries = list(votes) + list(submissions)
        entries = sorted(entries, key=lambda x: x.created_at, reverse=True)[:10]

        cache.set(RECENT_EVENTS_CACHE_ENTRY, entries, 24*3600)

        number_of_votes = Vote.objects.count()
        cache.set(NUMBER_OF_VOTES_CACHE_ENTRY, number_of_votes, 24*3600)

        logger.debug("There are %d entries" % len(entries))
        logger.debug("There are %d votes" % number_of_votes)
    except:
        # This task runs too frequently to report a recurring problem every time
        # it happens.
        global exception_logged
        if not exception_logged:
            logger.exception("Unexpected exception in update_recent_events")
            exception_logged = True
    finally:
        # Be a good citizen and reset the worker's thread to the default state
        set_thread_readwrite_db()
Example #4
0
 def test_with_readonly_db_when_readwrite(self):
     set_thread_readwrite_db()
     assert is_thread_readwrite()
     with readonly_db():
         assert not is_thread_readwrite()
     assert is_thread_readwrite()