[Integration][New Relic] - Add support for service level

integrations/newrelic/.port/resources/blueprints.json

@@ -133,5 +133,59 @@
     "mirrorProperties": {},
     "calculationProperties": {},
     "relations": {}
+  },
+  {
+    "identifier": "newRelicServiceLevel",
+    "description": "This blueprint represents a New Relic Service Level",
+    "title": "New Relic Service Level",
+    "icon": "NewRelic",
+    "schema": {
+      "properties": {
+        "description": {
+          "title": "Description",
+          "type": "string"
+        },
+        "targetThreshold": {
+          "icon": "DefaultProperty",
+          "title": "Target Threshold",
+          "type": "number"
+        },
+        "createdAt": {
+          "title": "Created At",
+          "type": "string",
+          "format": "date-time"
+        },
+        "updatedAt": {
+          "title": "Updated At",
+          "type": "string",
+          "format": "date-time"
+        },
+        "createdBy": {
+          "title": "Creator",
+          "type": "string",
+          "format": "user"
+        },
+        "serviceLevelIndicator": {
+          "type": "number",
+          "title": "Service Level Indicator"
+        },
+        "tags": {
+          "type": "object",
+          "title": "Tags"
+        }
+      },
+      "required": []
+    },
+    "mirrorProperties": {},
+    "calculationProperties": {},
+    "aggregationProperties": {},
+    "relations": {
+      "newRelicService": {
+        "title": "New Relic service",
+        "target": "newRelicService",
+        "required": false,
+        "many": false
+      }
+    }
   }
 ]

integrations/newrelic/.port/resources/port-app-config.yaml

@@ -71,3 +71,22 @@ resources:
             activatedAt: .activatedAt
           relations:
             newRelicService: ".__APPLICATION.entity_guids + .__SERVICE.entity_guids"
+  - kind: newRelicServiceLevel
+    selector:
+      query: 'true'
+    port:
+      entity:
+        mappings:
+          blueprint: '"newRelicServiceLevel"'
+          identifier: .serviceLevel.indicators[0].id
+          title: .serviceLevel.indicators[0].name
+          properties:
+            description: .serviceLevel.indicators[0].description
+            targetThreshold: .serviceLevel.indicators[0].objectives[0].target
+            createdAt: if .serviceLevel.indicators[0].createdAt != null then (.serviceLevel.indicators[0].createdAt | tonumber / 1000 | todate) else null end
+            updatedAt: .serviceLevel.indicators[0].updatedAt
+            createdBy: .serviceLevel.indicators[0].createdBy.email
+            serviceLevelIndicator: .__SLI.SLI
+            tags: .tags
+          relations:
+            newRelicService: .serviceLevel.indicators[0].guid

integrations/newrelic/.port/spec.yaml

@@ -7,6 +7,7 @@ features:
     resources:
       - kind: newRelicService
       - kind: newRelicAlert
+      - kind: newRelicServiceLevel
 configurations:
   - name: newRelicAPIKey
     required: true

integrations/newrelic/CHANGELOG.md

@@ -7,6 +7,13 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 <!-- towncrier release notes start -->
 
+# Port_Ocean 0.1.62 (2024-07-24)
+
+### Improvements
+
+- Added support for service level indicators and objectives
+
+
 # Port_Ocean 0.1.61 (2024-07-24)
 
 ### Improvements

integrations/newrelic/newrelic_integration/core/query_templates/service_levels.py

@@ -0,0 +1,63 @@
+GET_SLI_BY_NRQL_QUERY = """
+{
+  actor {
+    account(id: {{ account_id }}) {
+      nrql(query: "{{ nrql_query }}") {
+  results
+      }
+    }
+  }
+}
+"""
+
+LIST_SLOS_QUERY = """
+{
+  actor {
+    entitySearch(query: "type ='SERVICE_LEVEL'") {
+      count
+      query
+      results{{ next_cursor_request }} {
+        entities {
+          serviceLevel {
+            indicators {
+              resultQueries {
+                indicator {
+                  nrql
+                }
+              }
+              id
+              name
+              description
+              createdBy {
+                email
+              }
+              guid
+              updatedAt
+              createdAt
+              updatedBy {
+                email
+              }
+              objectives {
+                description
+                target
+                name
+                timeWindow {
+                  rolling {
+                    count
+                    unit
+                  }
+                }
+              }
+            }
+          }
+          tags {
+            key
+            values
+          }
+        }
+        nextCursor
+      }
+    }
+  }
+}
+"""

integrations/newrelic/newrelic_integration/core/service_levels.py

@@ -0,0 +1,80 @@
+from typing import Any, AsyncIterable, Tuple, Optional
+import httpx
+from port_ocean.context.ocean import ocean
+from newrelic_integration.core.query_templates.service_levels import (
+    LIST_SLOS_QUERY,
+    GET_SLI_BY_NRQL_QUERY,
+)
+from newrelic_integration.core.utils import send_graph_api_request
+from newrelic_integration.utils import (
+    render_query,
+)
+from newrelic_integration.core.paging import send_paginated_graph_api_request
+
+SLI_OBJECT = "__SLI"
+
+
+class ServiceLevelsHandler:
+    def __init__(self, http_client: httpx.AsyncClient):
+        self.http_client = http_client
+
+    async def get_service_level_indicator_value(
+        self, http_client: httpx.AsyncClient, nrql: str
+    ) -> dict[Any, Any]:
+        query = await render_query(
+            GET_SLI_BY_NRQL_QUERY,
+            nrql_query=nrql,
+            account_id=ocean.integration_config.get("new_relic_account_id"),
+        )
+        response = await send_graph_api_request(
+            http_client, query, request_type="get_service_level_indicator_value"
+        )
+        service_levels = (
+            response.get("data", {})
+            .get("actor", {})
+            .get("account", {})
+            .get("nrql", {})
+            .get("results", [])
+        )
+        if service_levels:
+            return service_levels[0]
+        return {}
+
+    async def list_service_levels(self) -> AsyncIterable[dict[str, Any]]:
+
+        async for service_level in send_paginated_graph_api_request(
+            self.http_client,
+            LIST_SLOS_QUERY,
+            request_type="list_service_levels",
+            extract_data=self._extract_service_levels,
+        ):
+            nrql = (
+                service_level.get("serviceLevel", {})
+                .get("indicators", [])[0]
+                .get("resultQueries", {})
+                .get("indicator", {})
+                .get("nrql")
+            )
+            service_level[SLI_OBJECT] = await self.get_service_level_indicator_value(
+                self.http_client, nrql
+            )
+            self._format_tags(service_level)
+            yield service_level
+
+    @staticmethod
+    async def _extract_service_levels(
+        response: dict[Any, Any]
+    ) -> Tuple[Optional[str], list[dict[Any, Any]]]:
+        """Extract service levels from the response. used by send_paginated_graph_api_request"""
+        results = (
+            response.get("data", {})
+            .get("actor", {})
+            .get("entitySearch", {})
+            .get("results", {})
+        )
+        return results.get("nextCursor"), results.get("entities", [])
+
+    @staticmethod
+    def _format_tags(entity: dict[Any, Any]) -> dict[Any, Any]:
+        entity["tags"] = {tag["key"]: tag["values"] for tag in entity.get("tags", [])}
+        return entity

integrations/newrelic/newrelic_integration/ocean.py

@@ -5,6 +5,8 @@
 
 from newrelic_integration.core.entities import EntitiesHandler
 from newrelic_integration.core.issues import IssuesHandler, IssueState, IssueEvent
+from newrelic_integration.core.service_levels import ServiceLevelsHandler
+
 from newrelic_integration.utils import (
     get_port_resource_configuration_by_newrelic_entity_type,
     get_port_resource_configuration_by_port_kind,
@@ -63,6 +65,16 @@ async def resync_issues(kind: str) -> ASYNC_GENERATOR_RESYNC_TYPE:
             yield await IssuesHandler(http_client).list_issues()
 
 
+@ocean.on_resync(kind="newRelicServiceLevel")
+async def resync_service_levels(kind: str) -> ASYNC_GENERATOR_RESYNC_TYPE:
+    with logger.contextualize(resource_kind=kind):
+        async with httpx.AsyncClient() as http_client:
+            async for service_levels in ServiceLevelsHandler(
+                http_client
+            ).list_service_levels():
+                yield [service_levels]
+
+
 @ocean.router.post("/events")
 async def handle_issues_events(issue: IssueEvent) -> dict[str, bool]:
     with logger.contextualize(issue_id=issue.id, issue_state=issue.state):

integrations/newrelic/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "newrelic"
-version = "0.1.61"
+version = "0.1.62"
 description = "New Relic Integration"
 authors = ["Tom Tankilevitch <[email protected]>"]