1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#!/usr/bin/env python3
import os
import glob
from ydb_wrapper import YDBWrapper
def get_monitoring_queries(queries_dir):
"""Get list of all SQL files from monitoring queries directory"""
# Get absolute path to the script directory
script_dir = os.path.dirname(os.path.abspath(__file__))
queries_path = os.path.join(script_dir, queries_dir)
# Find all .sql files
sql_files = glob.glob(os.path.join(queries_path, "*.sql"))
if not sql_files:
print(f"Warning: No SQL files found in {queries_path}")
return []
# Sort for predictable execution order
sql_files.sort()
return sql_files
def execute_monitoring_query(ydb_wrapper, query_file_path):
"""Execute a single monitoring query"""
# Get filename without extension for query_name
query_name = os.path.splitext(os.path.basename(query_file_path))[0]
print(f"\n{'='*80}")
print(f"Executing monitoring query: {query_name}")
print(f"File: {query_file_path}")
print(f"{'='*80}\n")
try:
# Read SQL query from file
with open(query_file_path, 'r', encoding='utf-8') as f:
query = f.read()
# Check that query is not empty
if not query.strip():
print(f"Warning: Query file {query_file_path} is empty, skipping")
return False
# Execute query with query_name parameter
results = ydb_wrapper.execute_scan_query(query, query_name=query_name)
# Log result
row_count = len(results) if results else 0
print(f"✅ Query '{query_name}' completed successfully: {row_count} rows returned")
return True
except Exception as e:
print(f"❌ Error executing query '{query_name}': {e}")
return False
def main():
script_name = os.path.basename(__file__)
queries_dir = "monitoring_queries"
print(f"Starting monitoring queries executor...")
print(f"Script: {script_name}")
print(f"Queries directory: {queries_dir}\n")
with YDBWrapper() as ydb_wrapper:
# Check credentials
if not ydb_wrapper.check_credentials():
return 1
# Get list of monitoring queries
query_files = get_monitoring_queries(queries_dir)
if not query_files:
print("No monitoring queries to execute")
return 0
print(f"Found {len(query_files)} monitoring queries to execute:\n")
for i, query_file in enumerate(query_files, 1):
query_name = os.path.splitext(os.path.basename(query_file))[0]
print(f" {i}. {query_name}")
print()
# Execute all queries
success_count = 0
failed_count = 0
for query_file in query_files:
if execute_monitoring_query(ydb_wrapper, query_file):
success_count += 1
else:
failed_count += 1
# Final statistics
print(f"\n{'='*80}")
print(f"Monitoring queries execution completed")
print(f"Total: {len(query_files)}, Success: {success_count}, Failed: {failed_count}")
print(f"{'='*80}\n")
# Return exit code: 0 if all successful, 1 if there were errors
return 0 if failed_count == 0 else 1
if __name__ == "__main__":
exit(main())
|
