Merge pull request #395 from fabric-testbed/rel1.7

Rel1.7 - changes

Author: kthare10

.gitignore

@@ -150,3 +150,8 @@ fabric_cf/actor/test/*.graphml
 secrets/kafkacat2-ca1-signed.pem
 secrets/kafkacat1-ca1-signed.pem
 secrets/...
+neo4j/RENC.graphml
+neo4j/Network-dev.graphml
+.DS_Store
+*.log*
+*.avsc

Dockerfile-auth

@@ -1,7 +1,7 @@
 FROM python:3.11.0
 MAINTAINER Komal Thareja<[email protected]>
 
-ARG HANDLERS_VER=1.6.3
+ARG HANDLERS_VER=1.7.1
 
 RUN mkdir -p /usr/src/app
 WORKDIR /usr/src/app
@@ -11,6 +11,7 @@ EXPOSE 11000
 
 RUN apt-get update
 RUN apt-get install cron -y
+RUN apt-get install sshpass -y
 
 COPY docker-entrypoint.sh /usr/src/app/
 COPY fabric_cf /usr/src/app/fabric_cf

fabric_cf/__init__.py

@@ -1,2 +1,2 @@
-__version__ = "1.6.2"
+__version__ = "1.7.0"
 __VERSION__ = __version__

fabric_cf/actor/core/apis/abc_actor_container.py

@@ -152,3 +152,12 @@ def remove_actor_database(self, *, actor_name: str):
         Remove Actor Database
         @params actor_name: actor name
         """
+
+    @abstractmethod
+    def get_actor(self) -> ABCActorMixin:
+        """
+        Return Actor
+
+        @return Actor
+        """
+        pass

fabric_cf/actor/core/apis/abc_actor_management_object.py

@@ -26,9 +26,9 @@
 from __future__ import annotations
 
 from abc import abstractmethod
+from datetime import datetime
 from typing import TYPE_CHECKING, Tuple, Dict, List
 
-from fabric_mb.message_bus.messages.poa_avro import PoaAvro
 from fabric_mb.message_bus.messages.result_avro import ResultAvro
 from fabric_mb.message_bus.messages.result_delegation_avro import ResultDelegationAvro
 from fabric_mb.message_bus.messages.result_poa_avro import ResultPoaAvro
@@ -38,6 +38,7 @@
 from fabric_mb.message_bus.messages.result_slice_avro import ResultSliceAvro
 from fabric_mb.message_bus.messages.result_string_avro import ResultStringAvro
 from fabric_mb.message_bus.messages.slice_avro import SliceAvro
+from fim.user import GraphFormat
 
 from fabric_cf.actor.core.apis.abc_management_object import ABCManagementObject
 from fabric_cf.actor.core.container.maintenance import Site
@@ -237,7 +238,8 @@ def get_sites(self, *, caller: AuthToken, site: str) -> ResultSitesAvro:
     def get_reservations(self, *, caller: AuthToken, states: List[int] = None,
                          slice_id: ID = None, rid: ID = None, oidc_claim_sub: str = None,
                          email: str = None, rid_list: List[str] = None, type: str = None,
-                         site: str = None, node_id: str = None) -> ResultReservationAvro:
+                         site: str = None, node_id: str = None,
+                         host: str = None, ip_subnet: str = None) -> ResultReservationAvro:
         """
         Get Reservations
         @param states states
@@ -251,12 +253,16 @@ def get_reservations(self, *, caller: AuthToken, states: List[int] = None,
         @param node_id node id
         Obtains all reservations with error information in case of failure
         @param caller caller
+        @param host host
+        @param ip_subnet ip subnet
+
         @return returns list of the reservations
         """
 
     def get_slices(self, *, slice_id: ID, caller: AuthToken, slice_name: str = None, email: str = None,
                    states: List[int] = None, project: str = None, limit: int = None,
-                   offset: int = None, user_id: str = None) -> ResultSliceAvro:
+                   offset: int = None, user_id: str = None, search: str = None,
+                   exact_match: bool = False) -> ResultSliceAvro:
         """
         Obtains all slices.
         @param slice_id slice id
@@ -268,13 +274,73 @@ def get_slices(self, *, slice_id: ID, caller: AuthToken, slice_name: str = None,
         @param offset offset
         @param caller caller
         @param user_id user_id
+        @param search: search term applied
+        @param exact_match: Exact Match for Search term
         @return returns list of slices
         """
 
+    @abstractmethod
+    def increment_metrics(self, *, project_id: str, oidc_sub: str, slice_count: int = 1) -> bool:
+        """
+        Add or update metrics
+
+        @param project_id project id
+        @param oidc_sub oidc sub
+        @param slice_count slice_count
+
+        @return true or false
+
+        @throws Exception in case of error
+        """
+
+    @abstractmethod
+    def get_metrics(self, *, project_id: str, oidc_sub: str, excluded_projects: List[str] = None) -> list:
+        """
+        Get metrics
+
+        @param project_id project id
+        @param oidc_sub oidc sub
+        @param excluded_projects excluded_projects
+
+        @return list of metric information
+
+        @throws Exception in case of error
+        """
+
+    def get_slice_count(self, *, caller: AuthToken, email: str = None, states: List[int] = None,
+                        project: str = None, user_id: str = None, excluded_projects: List[str] = None) -> int:
+        """
+        Obtains Slice count matching the filter criteria.
+
+        @param email email
+        @param project project id
+        @param states slice states
+        @param caller caller
+        @param user_id user_id
+        @param excluded_projects excluded_projects
+        @return returns number of slices
+        """
+
     def remove_slice(self, *, slice_id: ID, caller: AuthToken) -> ResultAvro:
         """
         Removes the specified slice
         @param slice_id slice id
         @param caller caller
         @return true for success; false otherwise
+        """
+
+    def build_broker_query_model(self, level_0_broker_query_model: str, level: int,
+                                 graph_format: GraphFormat = GraphFormat.GRAPHML,
+                                 start: datetime = None, end: datetime = None, includes: str = None,
+                                 excludes: str = None) -> str:
+        """
+        Build the BQM Model using current usage
+        @param level_0_broker_query_model Capacity Model
+        @param level: level of details
+        @param graph_format: Graph Format
+        @param start: start time
+        @param end: end time
+        @param includes: comma separated lists of sites to include
+        @param excludes: comma separated lists of sites to exclude
+        @return BQM
         """

fabric_cf/actor/core/apis/abc_actor_mixin.py

@@ -38,6 +38,7 @@
 from fabric_cf.actor.core.apis.abc_tick import ABCTick
 
 if TYPE_CHECKING:
+    from fabric_cf.actor.core.kernel.slice_state_machine import SliceState
     from fabric_cf.actor.core.apis.abc_actor_event import ABCActorEvent
     from fabric_cf.actor.core.apis.abc_actor_proxy import ABCActorProxy
     from fabric_cf.actor.core.apis.abc_base_plugin import ABCBasePlugin
@@ -431,12 +432,13 @@ def register_slice(self, *, slice_object: ABCSlice):
         """
 
     @abstractmethod
-    def modify_slice(self, *, slice_object: ABCSlice):
+    def modify_slice(self, *, slice_object: ABCSlice, new_state: SliceState):
         """
         Modify the slice registered with the actor. Moves the slice into Modifying State
 
         Args:
             slice_object: slice_object
+            new_state: new_state
         Raises:
             Exception in case of error
         """

fabric_cf/actor/core/apis/abc_client_actor_management_object.py

@@ -128,15 +128,22 @@ def update_broker(self, *, broker: ProxyAvro, caller: AuthToken) -> ResultAvro:
 
     @abstractmethod
     def get_broker_query_model(self, *, broker: ID, caller: AuthToken, id_token: str,
-                               level: int, graph_format: GraphFormat) -> ResultBrokerQueryModelAvro:
+                               level: int, graph_format: GraphFormat, start: datetime = None,
+                               end: datetime = None, includes: str = None,
+                               excludes: str = None) -> ResultBrokerQueryModelAvro:
         """
         Get Pool Info
         @param broker : broker ID
         @param caller: caller
         @param id_token: str
         @param level: level of details
         @param graph_format: Graph Format
-        @return pool information
+        @param start: start time
+        @param end: end time
+        @param includes: comma separated lists of sites to include
+        @param excludes: comma separated lists of sites to exclude
+
+        @return resource information
         """
 
     @abstractmethod

fabric_cf/actor/core/apis/abc_database.py

@@ -159,9 +159,9 @@ def update_slice(self, *, slice_object: ABCSlice):
 
     @abstractmethod
     def get_reservations(self, *, slice_id: ID = None, graph_node_id: str = None, project_id: str = None,
-                         email: str = None, oidc_sub: str = None, rid: ID = None,
-                         states: list[int] = None, site: str = None,
-                         rsv_type: list[str] = None) -> List[ABCReservationMixin]:
+                         email: str = None, oidc_sub: str = None, rid: ID = None, states: list[int] = None,
+                         site: str = None, rsv_type: list[str] = None, start: datetime = None,
+                         end: datetime = None, ip_subnet: str = None, host: str = None) -> List[ABCReservationMixin]:
         """
         Retrieves the reservations.
 
@@ -172,13 +172,22 @@ def get_reservations(self, *, slice_id: ID = None, graph_node_id: str = None, pr
 
     @abstractmethod
     def get_components(self, *, node_id: str, states: list[int], rsv_type: list[str], component: str = None,
-                       bdf: str = None) -> Dict[str, List[str]]:
+                       bdf: str = None, start: datetime = None, end: datetime = None,
+                       excludes: List[str] = None) -> Dict[str, List[str]]:
         """
-        Retrieves the components.
+        Returns components matching the search criteria
+        @param node_id: Worker Node ID to which components belong
+        @param states: list of states used to find reservations
+        @param rsv_type: type of reservations
+        @param component: component name
+        @param bdf: Component's PCI address
+        @param start: start time
+        @param end: end time
+        @param excludes: Excludes the list of reservations
 
-        @return list of components
+        NOTE# For P4 switches; node_id=node+renc-p4-sw  component=ip+192.168.11.8 bdf=p1
 
-        @throws Exception in case of error
+        @return Dictionary with component name as the key and value as list of associated PCI addresses in use.
         """
 
     @abstractmethod
@@ -194,7 +203,8 @@ def get_client_reservations(self, *, slice_id: ID = None) -> List[ABCReservation
     @abstractmethod
     def get_slices(self, *, slice_id: ID = None, slice_name: str = None, project_id: str = None, email: str = None,
                    states: list[int] = None, oidc_sub: str = None, slc_type: List[SliceTypes] = None,
-                   limit: int = None, offset: int = None, lease_end: datetime = None) -> List[ABCSlice] or None:
+                   limit: int = None, offset: int = None, lease_end: datetime = None,
+                   search: str = None, exact_match: bool = False) -> List[ABCSlice] or None:
         """
         Retrieves the specified slices.
 
@@ -208,12 +218,58 @@ def get_slices(self, *, slice_id: ID = None, slice_name: str = None, project_id:
         @param limit limit
         @param offset offset
         @param lease_end lease_end
+        @param search: search term applied
+        @param exact_match: Exact Match for Search term
 
         @return list of slices
 
         @throws Exception in case of error
         """
 
+    @abstractmethod
+    def get_slice_count(self, *, project_id: str = None, email: str = None, states: list[int] = None,
+                        oidc_sub: str = None, slc_type: List[SliceTypes] = None,
+                        excluded_projects: List[str] = None) -> int:
+        """
+        Retrieves the slices count.
+
+        @param project_id project id
+        @param email email
+        @param states states
+        @param oidc_sub oidc sub
+        @param slc_type slice type
+        @param excluded_projects excluded_projects
+
+        @return number of slices matching the filter criteria
+
+        @throws Exception in case of error
+        """
+
+    @abstractmethod
+    def increment_metrics(self, *, project_id: str, oidc_sub: str, slice_count: int = 1) -> bool:
+        """
+        Add or Update Metrics
+
+        @param project_id project id
+        @param oidc_sub oidc sub
+        @param slice_count slice_count
+
+        @return true or false
+
+        @throws Exception in case of error
+        """
+
+    @abstractmethod
+    def get_metrics(self, *, project_id: str, oidc_sub: str, excluded_projects: List[str] = None) -> list:
+        """
+        Get Metrics
+        @param project_id: project id
+        @param oidc_sub: user id
+        @param excluded_projects: list of project ids to exclude
+
+        @return list of metrics
+        """
+
     @abstractmethod
     def initialize(self):
         """