mongodb_url = os.environ["MONGODB_URL"]
target_database = os.environ["TARGET_DATABASE"]
prometheus_config_file = os.environ["PROMETHEUS_CONFIG_FILE"]
+prometheus_base_config_file = os.environ["PROMETHEUS_BASE_CONFIG_FILE"]
prometheus_url = os.environ["PROMETHEUS_URL"]
def save_successful_jobs(client, jobs):
for job in jobs:
client[target_database].prometheus_jobs.update_one(
- {"_id": ObjectId(job["_id"]["$oid"])},
- {"$set": {"is_active": True}}
+ {"_id": ObjectId(job["_id"]["$oid"])}, {"$set": {"is_active": True}}
)
def generate_prometheus_config(prometheus_jobs, config_file_path):
- config_file = open(config_file_path, encoding="utf-8", mode="r")
- config_file_contents = config_file.read()
- config_file.close()
-
- config_file_yaml = yaml.load(config_file_contents, yaml.FullLoader)
+ with open(config_file_path, encoding="utf-8", mode="r") as config_file:
+ config_file_yaml = yaml.safe_load(config_file)
if config_file_yaml is None:
config_file_yaml = {}
-
- if len(prometheus_jobs) == 0:
+ if "scrape_configs" not in config_file_yaml:
config_file_yaml["scrape_configs"] = []
- return config_file_yaml
-
- config_file_yaml["scrape_configs"] = []
for prometheus_job in prometheus_jobs:
cleaned_up_job = clean_up_job(prometheus_job)
stored_jobs = get_jobs(client)
print(f"Jobs detected : {len(stored_jobs):d}")
generated_prometheus_config = generate_prometheus_config(
- stored_jobs, prometheus_config_file
+ stored_jobs, prometheus_base_config_file
)
print(f"Writing new config file to {prometheus_config_file}")
config_file = open(prometheus_config_file, "w")
config_file.truncate(0)
- config_file.write(yaml.dump(generated_prometheus_config))
+ print(yaml.safe_dump(generated_prometheus_config))
+ config_file.write(yaml.safe_dump(generated_prometheus_config))
config_file.close()
print("New config written, updating prometheus")
update_resp = await reload_prometheus_config(prometheus_url)
- is_valid = await validate_configuration(
- prometheus_url, generated_prometheus_config
- )
+ is_valid = await validate_configuration(prometheus_url, generated_prometheus_config)
if update_resp and is_valid:
print("Prometheus config update successful")
save_successful_jobs(client, stored_jobs)
async def main():
client = pymongo.MongoClient(mongodb_url)
- print("Connected to MongoDB!")
+ print("Created MongoClient to connect to MongoDB!")
- try:
- print("Refreshing prometheus config file for first time")
- await main_task(client)
- except Exception as error:
- print("Error in first configuration attempt!")
- print(error)
+ # Initial loop. First refresh of prometheus config file
+ first_refresh_completed = False
+ tries = 1
+ while tries <= 3:
+ try:
+ print("Refreshing prometheus config file for first time")
+ await main_task(client)
+ first_refresh_completed = True
+ except Exception as error:
+ print(f"Error in configuration attempt! Number of tries: {tries}/3")
+ print(error)
+ time.sleep(5)
+ tries += 1
+ if not first_refresh_completed:
+ print("Not possible to refresh prometheus config file for first time")
+ return
+ # Main loop
while True:
try:
# Needs mongodb in replica mode as this feature relies in OpLog
print("Listening to changes in prometheus jobs collection")
for change in change_stream:
print("Change detected, updating prometheus config")
+ print(f"{change}")
await main_task(client)
print()
except Exception as error: