d0zingcat/redis-keys-statistics
1
0
Fork 0
mirror of https://github.com/d0zingcat/redis-keys-statistics.git synced 2026-05-13 15:09:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2 from d0zingcat/enh/usage_samples

Browse Source 
add samples args
This commit is contained in:
d0zingcat
2024-07-24 12:20:24 +08:00
committed by GitHub
parent a0d951e1d1 a24812a60a
commit 14321d0635
2 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rks/main.py
View File

@@ -12,6 +12,8 @@ def main():
parser.add_argument('--port', required=True, help='Redis port')
parser.add_argument('--username', default=None, help='Redis username')
parser.add_argument('--password', default=None, help='Redis password')
parser.add_argument('--samples', default=5,
help='Redis memory usage samples')
parser.add_argument('--cluster', action='store_true',
help='Enable cluster mode')
parser.add_argument('--batch_size', type=int, default=1000,
@@ -91,7 +93,7 @@ def main():
if db.startswith('db'):
db_num = int(db[2:])
get_redis_keys(r, args.batch_size,
db_num, args.pretty_format)
db_num, args.pretty_format, args.samples)
r.close()

16
rks/redis_utils.py
View File

@@ -4,7 +4,7 @@ from .display import analyze_redis_keys
from .analysis import update_topk_heap, update_statistics
def get_redis_keys(r, batch_size, db_num, use_pretty):
def get_redis_keys(r, batch_size, db_num, use_pretty, samples):
r.execute_command('SELECT', db_num)
@@ -25,7 +25,7 @@ def get_redis_keys(r, batch_size, db_num, use_pretty):
for i, key in ipairs(scanResult[2]) do
local key_type = redis.call('TYPE', key)['ok']
local memory = redis.call('MEMORY', 'USAGE', key)
local memory = redis.call('MEMORY', 'USAGE', key, 'SAMPLES', SAMPLES_PLACEHOLDER)
local ttl = redis.call('TTL', key)
table.insert(result, {key, key_type, memory, ttl})
@@ -35,6 +35,7 @@ def get_redis_keys(r, batch_size, db_num, use_pretty):
"""
script = script.replace("BATCH_SIZE_PLACEHOLDER", str(batch_size))
script = script.replace("SAMPLES_PLACEHOLDER", str(samples))
cursor = b'0'
while True:
@@ -78,7 +79,8 @@ def get_redis_cluster_keys(rc, batch_size, replica_only, use_pretty):
masters = []
for node in nodes:
if 'master' in node['flags']:
master_dict = {'master': node['id'], 'slots': node['slots'], 'slaves': []}
master_dict = {'master': node['id'],
'slots': node['slots'], 'slaves': []}
masters.append(master_dict.copy())
for node in nodes:
@@ -93,9 +95,11 @@ def get_redis_cluster_keys(rc, batch_size, replica_only, use_pretty):
for master in masters:
if slave_flag is False:
node = next((node for node in nodes if node['id'] == master['master']), None)
node = next(
(node for node in nodes if node['id'] == master['master']), None)
else:
node = next((node for node in nodes if node['id'] == master['slaves'][0]), None)
node = next(
(node for node in nodes if node['id'] == master['slaves'][0]), None)
r = redis.Redis(host=node['host'], port=node['port'])
r.execute_command('READONLY')
@@ -144,4 +148,4 @@ def get_redis_cluster_keys(rc, batch_size, replica_only, use_pretty):
r.close()
return analyze_redis_keys(min_heap, prefix_statistics_map, total_key_size, total_key_count, key_count_by_type, 0, use_pretty)
return analyze_redis_keys(min_heap, prefix_statistics_map, total_key_size, total_key_count, key_count_by_type, 0, use_pretty)