diff --git a/backend/app/connectors/wazuh_indexer/routes/monitoring.py b/backend/app/connectors/wazuh_indexer/routes/monitoring.py index 6f62231d..f56caeb0 100644 --- a/backend/app/connectors/wazuh_indexer/routes/monitoring.py +++ b/backend/app/connectors/wazuh_indexer/routes/monitoring.py @@ -14,8 +14,10 @@ from app.connectors.wazuh_indexer.services.monitoring import cluster_healthcheck from app.connectors.wazuh_indexer.services.monitoring import indices_stats from app.connectors.wazuh_indexer.services.monitoring import node_allocation +from app.connectors.wazuh_indexer.services.monitoring import ( + output_shard_number_to_be_set_based_on_nodes, +) from app.connectors.wazuh_indexer.services.monitoring import shards -from app.connectors.wazuh_indexer.services.monitoring import output_shard_number_to_be_set_based_on_nodes wazuh_indexer_router = APIRouter() @@ -122,6 +124,7 @@ async def get_shards() -> Union[ShardsResponse, HTTPException]: else: raise HTTPException(status_code=500, detail="Failed to retrieve shards.") + @wazuh_indexer_router.get( "/output_shard_number_to_be_set_based_on_nodes", description="Fetch Wazuh Indexer output_shard_number_to_be_set_based_on_nodes", @@ -144,4 +147,3 @@ async def get_output_shard_number_to_be_set_based_on_nodes_route() -> int: return output_shard_number_to_be_set_based_on_nodes_response else: raise HTTPException(status_code=500, detail="Failed to retrieve output_shard_number_to_be_set_based_on_nodes.") - diff --git a/backend/app/integrations/carbonblack/services/provision.py b/backend/app/integrations/carbonblack/services/provision.py index 240d3987..a215bf95 100644 --- a/backend/app/integrations/carbonblack/services/provision.py +++ b/backend/app/integrations/carbonblack/services/provision.py @@ -12,6 +12,9 @@ from app.connectors.grafana.utils.universal import create_grafana_client from app.connectors.graylog.services.management import start_stream from app.connectors.graylog.utils.universal import send_post_request +from app.connectors.wazuh_indexer.services.monitoring import ( + output_shard_number_to_be_set_based_on_nodes, +) from app.customer_provisioning.schema.grafana import GrafanaDatasource from app.customer_provisioning.schema.grafana import GrafanaDataSourceCreationResponse from app.customer_provisioning.schema.graylog import GraylogIndexSetCreationResponse @@ -28,7 +31,6 @@ from app.integrations.routes import create_integration_meta from app.integrations.schema import CustomerIntegrationsMetaSchema from app.utils import get_connector_attribute -from app.connectors.wazuh_indexer.services.monitoring import output_shard_number_to_be_set_based_on_nodes ################## ! GRAYLOG ! ################## diff --git a/backend/app/integrations/huntress/services/provision.py b/backend/app/integrations/huntress/services/provision.py index e9f6b9c0..7c84f768 100644 --- a/backend/app/integrations/huntress/services/provision.py +++ b/backend/app/integrations/huntress/services/provision.py @@ -12,6 +12,9 @@ from app.connectors.grafana.utils.universal import create_grafana_client from app.connectors.graylog.services.management import start_stream from app.connectors.graylog.utils.universal import send_post_request +from app.connectors.wazuh_indexer.services.monitoring import ( + output_shard_number_to_be_set_based_on_nodes, +) from app.customer_provisioning.schema.grafana import GrafanaDatasource from app.customer_provisioning.schema.grafana import GrafanaDataSourceCreationResponse from app.customer_provisioning.schema.graylog import GraylogIndexSetCreationResponse @@ -28,7 +31,7 @@ from app.integrations.routes import create_integration_meta from app.integrations.schema import CustomerIntegrationsMetaSchema from app.utils import get_connector_attribute -from app.connectors.wazuh_indexer.services.monitoring import output_shard_number_to_be_set_based_on_nodes + ################## ! GRAYLOG ! ################## async def build_index_set_config(