natas_hw
/
cassandra
mirror of https://github.com/apache/cassandra
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cassandra-18837 Tab complete datacenter values in cqlsh

This commit is contained in:
pranavchaurasia18 2025-04-24 14:42:37 +05:30
parent 17cb89208c
commit 4f0a996934
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pylib/cqlshlib/cql3handling.py
View File

@ -461,6 +461,7 @@ def ks_prop_val_completer(ctxt, cass):
def ks_prop_val_mapkey_completer(ctxt, cass):
session = cass.session if hasattr(cass, 'session') else cass
optname = ctxt.get_binding('propname')[-1]
if optname != 'replication':
return ()
@ -475,7 +476,24 @@ def ks_prop_val_mapkey_completer(ctxt, cass):
if repclass == 'SimpleStrategy':
opts = {'replication_factor'}
elif repclass == 'NetworkTopologyStrategy':
return [Hint('<dc_name>')]
try:
resultLocal = session.execute("SELECT data_center FROM system.local")
row = resultLocal.one()
local_dc = row['data_center'] if row else None
resultPeer = session.execute("SELECT data_center FROM system.peers")
peer_dc = [row['data_center'] for row in resultPeer if row['data_center']]
dummy_dcs = ['dc_dummy1', 'dc_dummy2', 'dc_dummy3']
dc_list = sorted(set([dc for dc in [local_dc] + peer_dc + dummy_dcs if dc]))
# dc_list = sorted(set([dc for dc in [local_dc] + peer_dc if dc]))
return list(map(escape_value, dc_list)) if dc_list else [Hint('<dc_name_not_available>')]
except Exception as e:
print(f"Query failed: {e}")
return [Hint('<dc_name_not_available>')]
return list(map(escape_value, opts.difference(keysseen)))