##// END OF EJS Templates
caches: log properly how we cleanup cache keys
super-admin -
r4850:cbf6ce71 default
parent child Browse files
Show More
@@ -1,53 +1,54 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2015-2020 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20
21 21 import os
22 22 import atexit
23 23 import logging
24 24 import signal
25 25
26 26 log = logging.getLogger(__name__)
27 27
28 28 cache_keys_by_pid = set()
29 29
30 30
31 31 def free_cache_keys(*args):
32 32 ssh_cmd = os.environ.get('RC_CMD_SSH_WRAPPER')
33 33 if ssh_cmd:
34 34 return
35 35
36 36 from rhodecode.model.db import Session, CacheKey
37 log.info('Clearing %s cache keys', len(cache_keys_by_pid))
38 37
39 38 if cache_keys_by_pid:
40 39 try:
41 40 for cache_proc in set(cache_keys_by_pid):
42 41 like_expression = '{}%'.format(cache_proc)
43 CacheKey.query().filter(
44 CacheKey.cache_key.like(like_expression)).delete(synchronize_session='fetch')
42 qry = CacheKey.query().filter(CacheKey.cache_key.like(like_expression))
43 count = qry.count()
44 log.info('Clearing %s: %s cache keys, total: %s', cache_proc, len(cache_keys_by_pid), count)
45 qry.delete(synchronize_session='fetch')
45 46 cache_keys_by_pid.remove(cache_proc)
46 47 Session().commit()
47 48 except Exception:
48 49 log.exception('Failed to clear keys, exiting gracefully')
49 50
50 51
51 52 atexit.register(free_cache_keys)
52 53 signal.signal(signal.SIGTERM, free_cache_keys)
53 54 signal.signal(signal.SIGINT, free_cache_keys)
General Comments 0
You need to be logged in to leave comments. Login now